format

marimo-team · Jan 6, 2025 · b69c92c · b69c92c
1 parent 97648b5
commit b69c92c
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/frontend/src/core/errors/__tests__/utils.test.ts b/frontend/src/core/errors/__tests__/utils.test.ts
@@ -7,6 +7,8 @@ describe("wrapInFunction", () => {
     const input = "1 + 2";
     const expected = `def _():
     return 1 + 2
+
+
 _()`;
     expect(wrapInFunction(input)).toBe(expected);
   });
@@ -17,6 +19,8 @@ _()`;
     const expected = `def _():
     return (1 +
     2)
+
+
 _()`;
     expect(wrapInFunction(input)).toBe(expected);
   });
@@ -31,6 +35,8 @@ _()`;
         bar(1, 2),
         baz(3, 4)
     )
+
+
 _()`;
     expect(wrapInFunction(input)).toBe(expected);
   });
@@ -44,6 +50,8 @@ _()`;
         x = 1
         y = 2
     return
+
+
 _()`;
     expect(wrapInFunction(input)).toBe(expected);
   });
@@ -62,6 +70,8 @@ y = "bar"
         x="x",
         y="y",
     ))
+
+
 _()`;
     expect(wrapInFunction(input)).toBe(expected);
   });
@@ -75,6 +85,8 @@ y = 2`;
 
     y = 2
     return
+
+
 _()`;
     expect(wrapInFunction(input)).toBe(expected);
   });
@@ -88,6 +100,8 @@ _()`;
         x = 1
         y = 2
     return
+
+
 _()`;
     expect(wrapInFunction(input)).toBe(expected);
   });

diff --git a/frontend/src/core/errors/utils.ts b/frontend/src/core/errors/utils.ts
@@ -28,6 +28,8 @@ export function wrapInFunction(code: string) {
       "def _():",
       ...indentLines(lines, indentation),
       `${indentation}return`,
+      "",
+      "",
       "_()",
     ].join("\n");
   }
@@ -42,6 +44,8 @@ export function wrapInFunction(code: string) {
     ...indentLines(linesBeforeLastStmt, indentation),
     `${indentation}return ${linesRest[0]}`,
     ...indentLines(linesRest.slice(1), indentation),
+    "",
+    "",
     "_()",
   ].join("\n");
 }