tests: frontend tests

Author: dmadisetti

frontend/src/core/codemirror/language/__tests__/extension.test.ts

@@ -163,36 +163,30 @@ describe("switchLanguage", () => {
     // Switch back to python
     switchLanguage(mockEditor, { language: "python", keepCodeAsIs: false });
     expect(mockEditor.state.doc.toString()).toMatchInlineSnapshot(`
-      "mo.md(
-          r"""
+      "mo.md(r"""
       print('Hello')
       print('Goodbye')
-      """
-      )"
+      """)"
     `);
 
     // Switch to sql
     switchLanguage(mockEditor, { language: "sql", keepCodeAsIs: false });
     expect(mockEditor.state.doc.toString()).toMatchInlineSnapshot(`
-      "mo.md(
-          r"""
+      "mo.md(r"""
       print('Hello')
       print('Goodbye')
-      """
-      )"
+      """)"
     `);
 
     // Switch back to python
     switchLanguage(mockEditor, { language: "python", keepCodeAsIs: false });
     expect(mockEditor.state.doc.toString()).toMatchInlineSnapshot(`
       "_df = mo.sql(
           f"""
-          mo.md(
-              r\\"""
+          mo.md(r\\"""
           print('Hello')
           print('Goodbye')
-          \\"""
-          )
+          \\""")
           """
       )"
     `);

frontend/src/core/codemirror/language/__tests__/markdown.test.ts

@@ -186,7 +186,7 @@ describe("MarkdownLanguageAdapter", () => {
 
     it("should preserve indentation in f-strings", () => {
       const pythonCode =
-        'mo.md(\n    f"""\n```python\n{some_variable}\n```\n"""\n)';
+        'mo.md(f"""\n```python\n{some_variable}\n```\n""")';
       const [innerCode, offset, metadata] = adapter.transformIn(pythonCode);
       expect(innerCode).toBe("```python\n{some_variable}\n```");
       expect(offset).toBe(15);
@@ -312,12 +312,10 @@ describe("MarkdownLanguageAdapter", () => {
       metadata.quotePrefix = "f";
       const [wrappedCode, offset] = adapter.transformOut(code, metadata);
       expect(wrappedCode).toMatchInlineSnapshot(`
-        "mo.md(
-            f"""
+        "mo.md(f"""
         # Title
         {some_variable}
-        """
-        )"
+        """)"
       `);
       expect(offset).toBe(17);
     });
@@ -327,12 +325,10 @@ describe("MarkdownLanguageAdapter", () => {
       metadata.quotePrefix = "rf";
       const [wrappedCode, offset] = adapter.transformOut(code, metadata);
       expect(wrappedCode).toMatchInlineSnapshot(`
-        "mo.md(
-            rf"""
+        "mo.md(rf"""
         # Title
         {some_variable}
-        """
-        )"
+        """)"
       `);
       expect(offset).toBe(18);
     });
@@ -369,13 +365,11 @@ describe("MarkdownLanguageAdapter", () => {
 
     it("should return true for complex nested markdown", () => {
       const pythonCode = String.raw`
-      mo.md(
-        rf"""
-        \`\`\`python
-        {pathlib.Path(__file__).read_text(encoding="utf-8")}
-        \`\`\`
-        """
-      )
+      mo.md(rf"""
+      \`\`\`python
+      {pathlib.Path(__file__).read_text(encoding="utf-8")}
+      \`\`\`
+      """)
       `;
       expect(adapter.isSupported(pythonCode)).toBe(true);
     });