@alloy-js/python 0.5.0-dev.0 → 0.5.0-dev.2

package/dist/test/sourcefiles.test.js

@@ -1,168 +1,168 @@
 import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
 import { Prose } from "@alloy-js/core";
-import { d } from "@alloy-js/core/testing";
 import { expect, it } from "vitest";
 import * as py from "../src/index.js";
 import { dataclassesModule } from "../src/index.js";
-import { toSourceText } from "./utils.js";
+import { TestOutput, TestOutputDirectory } from "./utils.js";
 
 /**
- * toSourceText wraps the children in a SourceFile component
- * and renders it to a string.
+ * TestOutputDirectory wraps content in an Output component without a default
+ * SourceFile, so SourceFile components with their own path/props can be used.
  */
 it("renders an empty source file", () => {
-  const result = toSourceText([]);
-  expect(result).toRenderTo(d`
-
+  expect(_$createComponent(TestOutput, {})).toRenderTo(`
 
   `);
 });
 it("correct formatting of source file", () => {
-  const result = toSourceText([_$createComponent(py.ClassDeclaration, {
-    name: "someClass",
+  expect(_$createComponent(TestOutput, {
     get children() {
-      return _$createComponent(py.StatementList, {
+      return [_$createComponent(py.ClassDeclaration, {
+        name: "someClass",
         get children() {
-          return [_$createComponent(py.FunctionDeclaration, {
-            name: "someMethod",
-            returnType: "str",
+          return _$createComponent(py.StatementList, {
             get children() {
-              return _$createComponent(py.StatementList, {
+              return [_$createComponent(py.FunctionDeclaration, {
+                name: "someMethod",
+                returnType: "str",
                 get children() {
-                  return [_$createComponent(py.VariableDeclaration, {
-                    name: "x",
-                    type: "int",
-                    get initializer() {
-                      return _$createComponent(py.Atom, {
-                        jsValue: 42
-                      });
-                    }
-                  }), _$createComponent(py.VariableDeclaration, {
-                    name: "y",
-                    type: "int",
-                    get initializer() {
-                      return _$createComponent(py.Atom, {
-                        jsValue: 42
-                      });
-                    }
-                  }), _$createComponent(py.FunctionCallExpression, {
-                    target: "foo",
-                    args: ["a", "b"]
-                  }), _$createComponent(py.MemberExpression, {
+                  return _$createComponent(py.StatementList, {
                     get children() {
-                      return [_$createComponent(py.MemberExpression.Part, {
-                        id: "a"
-                      }), _$createComponent(py.MemberExpression.Part, {
-                        id: "b"
-                      }), _$createComponent(py.MemberExpression.Part, {
-                        key: "special-prop"
+                      return [_$createComponent(py.VariableDeclaration, {
+                        name: "x",
+                        type: "int",
+                        get initializer() {
+                          return _$createComponent(py.Atom, {
+                            jsValue: 42
+                          });
+                        }
+                      }), _$createComponent(py.VariableDeclaration, {
+                        name: "y",
+                        type: "int",
+                        get initializer() {
+                          return _$createComponent(py.Atom, {
+                            jsValue: 42
+                          });
+                        }
+                      }), _$createComponent(py.FunctionCallExpression, {
+                        target: "foo",
+                        args: ["a", "b"]
+                      }), _$createComponent(py.MemberExpression, {
+                        get children() {
+                          return [_$createComponent(py.MemberExpression.Part, {
+                            id: "a"
+                          }), _$createComponent(py.MemberExpression.Part, {
+                            id: "b"
+                          }), _$createComponent(py.MemberExpression.Part, {
+                            key: "special-prop"
+                          })];
+                        }
+                      }), _$createComponent(py.VariableDeclaration, {
+                        name: "z",
+                        type: "int",
+                        get initializer() {
+                          return _$createComponent(py.Atom, {
+                            jsValue: 42
+                          });
+                        }
                       })];
                     }
-                  }), _$createComponent(py.VariableDeclaration, {
-                    name: "z",
-                    type: "int",
-                    get initializer() {
-                      return _$createComponent(py.Atom, {
-                        jsValue: 42
-                      });
-                    }
-                  })];
+                  });
                 }
-              });
-            }
-          }), _$createComponent(py.VariableDeclaration, {
-            name: "someVar",
-            type: "int",
-            get initializer() {
-              return _$createComponent(py.Atom, {
-                jsValue: 42
-              });
-            }
-          }), _$createComponent(py.FunctionDeclaration, {
-            name: "someOtherMethod",
-            returnType: "str"
-          }), _$createComponent(py.VariableDeclaration, {
-            name: "someOtherVar",
-            type: "int",
-            get initializer() {
-              return _$createComponent(py.Atom, {
-                jsValue: 42
-              });
+              }), _$createComponent(py.VariableDeclaration, {
+                name: "someVar",
+                type: "int",
+                get initializer() {
+                  return _$createComponent(py.Atom, {
+                    jsValue: 42
+                  });
+                }
+              }), _$createComponent(py.FunctionDeclaration, {
+                name: "someOtherMethod",
+                returnType: "str"
+              }), _$createComponent(py.VariableDeclaration, {
+                name: "someOtherVar",
+                type: "int",
+                get initializer() {
+                  return _$createComponent(py.Atom, {
+                    jsValue: 42
+                  });
+                }
+              })];
             }
-          })];
+          });
         }
-      });
-    }
-  }), _$createComponent(py.FunctionDeclaration, {
-    name: "someFunction",
-    get children() {
-      return _$createComponent(py.StatementList, {
+      }), _$createComponent(py.FunctionDeclaration, {
+        name: "someFunction",
         get children() {
-          return [_$createComponent(py.VariableDeclaration, {
-            name: "x",
-            type: "int",
-            get initializer() {
-              return _$createComponent(py.Atom, {
-                jsValue: 42
-              });
-            }
-          }), _$createComponent(py.VariableDeclaration, {
-            name: "y",
-            type: "int",
-            get initializer() {
-              return _$createComponent(py.Atom, {
-                jsValue: 42
-              });
-            }
-          }), _$createComponent(py.FunctionCallExpression, {
-            target: "foo",
-            args: ["a", "b"]
-          }), _$createComponent(py.MemberExpression, {
+          return _$createComponent(py.StatementList, {
             get children() {
-              return [_$createComponent(py.MemberExpression.Part, {
-                id: "a"
-              }), _$createComponent(py.MemberExpression.Part, {
-                id: "b"
-              }), _$createComponent(py.MemberExpression.Part, {
-                key: "special-prop"
+              return [_$createComponent(py.VariableDeclaration, {
+                name: "x",
+                type: "int",
+                get initializer() {
+                  return _$createComponent(py.Atom, {
+                    jsValue: 42
+                  });
+                }
+              }), _$createComponent(py.VariableDeclaration, {
+                name: "y",
+                type: "int",
+                get initializer() {
+                  return _$createComponent(py.Atom, {
+                    jsValue: 42
+                  });
+                }
+              }), _$createComponent(py.FunctionCallExpression, {
+                target: "foo",
+                args: ["a", "b"]
+              }), _$createComponent(py.MemberExpression, {
+                get children() {
+                  return [_$createComponent(py.MemberExpression.Part, {
+                    id: "a"
+                  }), _$createComponent(py.MemberExpression.Part, {
+                    id: "b"
+                  }), _$createComponent(py.MemberExpression.Part, {
+                    key: "special-prop"
+                  })];
+                }
+              }), _$createComponent(py.VariableDeclaration, {
+                name: "z",
+                type: "int",
+                get initializer() {
+                  return _$createComponent(py.Atom, {
+                    jsValue: 42
+                  });
+                }
               })];
             }
-          }), _$createComponent(py.VariableDeclaration, {
-            name: "z",
-            type: "int",
-            get initializer() {
-              return _$createComponent(py.Atom, {
-                jsValue: 42
+          });
+        }
+      }), _$createComponent(py.ClassDeclaration, {
+        name: "someOtherClass",
+        get children() {
+          return _$createComponent(py.StatementList, {
+            get children() {
+              return _$createComponent(py.FunctionDeclaration, {
+                name: "someMethod",
+                returnType: "str"
               });
             }
-          })];
+          });
         }
-      });
-    }
-  }), _$createComponent(py.ClassDeclaration, {
-    name: "someOtherClass",
-    get children() {
-      return _$createComponent(py.StatementList, {
+      }), _$createComponent(py.MemberExpression, {
         get children() {
-          return _$createComponent(py.FunctionDeclaration, {
-            name: "someMethod",
-            returnType: "str"
-          });
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "a"
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "b"
+          }), _$createComponent(py.MemberExpression.Part, {
+            key: "special-prop"
+          })];
         }
-      });
-    }
-  }), _$createComponent(py.MemberExpression, {
-    get children() {
-      return [_$createComponent(py.MemberExpression.Part, {
-        id: "a"
-      }), _$createComponent(py.MemberExpression.Part, {
-        id: "b"
-      }), _$createComponent(py.MemberExpression.Part, {
-        key: "special-prop"
       })];
     }
-  })]);
-  expect(result).toRenderTo(d`
+  })).toRenderTo(`
     class SomeClass:
         def some_method() -> str:
             x: int = 42
@@ -193,7 +193,6 @@ it("correct formatting of source file", () => {
 
 
     a.b["special-prop"]
-
   `);
 });
 it("renders module documentation correctly", () => {
@@ -215,23 +214,27 @@ it("renders module documentation correctly", () => {
     todo: ["Add caching functionality", "Improve error messages"],
     style: "google"
   });
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    doc: moduleDoc,
+  expect(_$createComponent(TestOutputDirectory, {
     get children() {
-      return [_$createComponent(py.VariableDeclaration, {
-        name: "DEFAULT_TIMEOUT",
-        initializer: 30
-      }), _$createComponent(py.VariableDeclaration, {
-        name: "MAX_RETRIES",
-        initializer: 3
-      }), _$createComponent(py.FunctionDeclaration, {
-        name: "process_data",
-        children: "pass"
-      })];
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        doc: moduleDoc,
+        get children() {
+          return [_$createComponent(py.VariableDeclaration, {
+            name: "DEFAULT_TIMEOUT",
+            initializer: 30
+          }), _$createComponent(py.VariableDeclaration, {
+            name: "MAX_RETRIES",
+            initializer: 3
+          }), _$createComponent(py.FunctionDeclaration, {
+            name: "process_data",
+            children: "pass"
+          })];
+        }
+      });
     }
-  });
-  expect(toSourceText(content)).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     """
     This module provides utility functions for data processing. It includes
     functions for validation, transformation, and analysis.
@@ -253,54 +256,67 @@ it("renders module documentation correctly", () => {
     def process_data():
         pass
 
-
-  `);
+  `
+  });
 });
 it("renders source file without documentation correctly", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
+  expect(_$createComponent(TestOutputDirectory, {
     get children() {
-      return _$createComponent(py.FunctionDeclaration, {
-        name: "hello_world",
-        children: "print(\"Hello, World!\")"
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        get children() {
+          return _$createComponent(py.FunctionDeclaration, {
+            name: "hello_world",
+            children: "print(\"Hello, World!\")"
+          });
+        }
       });
     }
-  });
-  expect(toSourceText(content)).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     def hello_world():
         print("Hello, World!")
 
-
-  `);
+  `
+  });
 });
 it("nothing before top-level definition", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
+  expect(_$createComponent(TestOutputDirectory, {
     get children() {
-      return _$createComponent(py.FunctionDeclaration, {
-        name: "hello",
-        children: "pass"
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        get children() {
+          return _$createComponent(py.FunctionDeclaration, {
+            name: "hello",
+            children: "pass"
+          });
+        }
       });
     }
-  });
-  expect(toSourceText(content)).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     def hello():
         pass
 
-
-  `);
+  `
+  });
 });
 it("nothing before non-definition", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
+  expect(_$createComponent(TestOutputDirectory, {
     get children() {
-      return _$createComponent(py.VariableDeclaration, {
-        name: "x",
-        initializer: 42
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        get children() {
+          return _$createComponent(py.VariableDeclaration, {
+            name: "x",
+            initializer: 42
+          });
+        }
       });
     }
+  })).toRenderTo({
+    "test.py": "x = 42"
   });
-  expect(toSourceText(content)).toRenderTo("x = 42");
 });
 it("only doc before definition", () => {
   const moduleDoc = _$createComponent(py.ModuleDoc, {
@@ -310,17 +326,21 @@ it("only doc before definition", () => {
       })];
     }
   });
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    doc: moduleDoc,
+  expect(_$createComponent(TestOutputDirectory, {
     get children() {
-      return _$createComponent(py.FunctionDeclaration, {
-        name: "hello",
-        children: "pass"
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        doc: moduleDoc,
+        get children() {
+          return _$createComponent(py.FunctionDeclaration, {
+            name: "hello",
+            children: "pass"
+          });
+        }
       });
     }
-  });
-  expect(toSourceText(content)).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     """
     Module description.
     """
@@ -329,8 +349,8 @@ it("only doc before definition", () => {
     def hello():
         pass
 
-
-  `);
+  `
+  });
 });
 it("only doc before non-definition", () => {
   const moduleDoc = _$createComponent(py.ModuleDoc, {
@@ -340,127 +360,149 @@ it("only doc before non-definition", () => {
       })];
     }
   });
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    doc: moduleDoc,
+  expect(_$createComponent(TestOutputDirectory, {
     get children() {
-      return _$createComponent(py.VariableDeclaration, {
-        name: "x",
-        initializer: 42
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        doc: moduleDoc,
+        get children() {
+          return _$createComponent(py.VariableDeclaration, {
+            name: "x",
+            initializer: 42
+          });
+        }
       });
     }
-  });
-  expect(toSourceText(content)).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     """
     Module description.
     """
 
-    x = 42`);
+    x = 42`
+  });
 });
 it("only header before definition", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    header: "#!/usr/bin/env python3",
+  expect(_$createComponent(TestOutputDirectory, {
     get children() {
-      return _$createComponent(py.FunctionDeclaration, {
-        name: "hello",
-        children: "pass"
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        header: "#!/usr/bin/env python3",
+        get children() {
+          return _$createComponent(py.FunctionDeclaration, {
+            name: "hello",
+            children: "pass"
+          });
+        }
       });
     }
-  });
-
-  // 2 blank lines before definition (PEP 8)
-  expect(toSourceText(content)).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     #!/usr/bin/env python3
 
 
     def hello():
         pass
 
-
-  `);
+  `
+  });
 });
 it("only header before non-definition", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    header: "#!/usr/bin/env python3",
+  expect(_$createComponent(TestOutputDirectory, {
     get children() {
-      return _$createComponent(py.VariableDeclaration, {
-        name: "x",
-        initializer: 42
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        header: "#!/usr/bin/env python3",
+        get children() {
+          return _$createComponent(py.VariableDeclaration, {
+            name: "x",
+            initializer: 42
+          });
+        }
       });
     }
-  });
-
-  // 1 blank line for non-definition
-  expect(toSourceText(content)).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     #!/usr/bin/env python3
 
-    x = 42`);
+    x = 42`
+  });
 });
 it("only futureImports before definition", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    get futureImports() {
-      return [_$createComponent(py.FutureStatement, {
-        feature: "annotations"
-      })];
-    },
+  expect(_$createComponent(TestOutputDirectory, {
     get children() {
-      return _$createComponent(py.FunctionDeclaration, {
-        name: "hello",
-        children: "pass"
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        get futureImports() {
+          return [_$createComponent(py.FutureStatement, {
+            feature: "annotations"
+          })];
+        },
+        get children() {
+          return _$createComponent(py.FunctionDeclaration, {
+            name: "hello",
+            children: "pass"
+          });
+        }
       });
     }
-  });
-  expect(toSourceText(content)).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     from __future__ import annotations
 
 
     def hello():
         pass
 
-
-  `);
+  `
+  });
 });
 it("only futureImports before non-definition", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    get futureImports() {
-      return [_$createComponent(py.FutureStatement, {
-        feature: "annotations"
-      })];
-    },
+  expect(_$createComponent(TestOutputDirectory, {
     get children() {
-      return _$createComponent(py.VariableDeclaration, {
-        name: "x",
-        initializer: 42
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        get futureImports() {
+          return [_$createComponent(py.FutureStatement, {
+            feature: "annotations"
+          })];
+        },
+        get children() {
+          return _$createComponent(py.VariableDeclaration, {
+            name: "x",
+            initializer: 42
+          });
+        }
       });
     }
-  });
-  expect(toSourceText(content)).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     from __future__ import annotations
 
-    x = 42`);
+    x = 42`
+  });
 });
 it("only imports before definition", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
+  expect(_$createComponent(TestOutputDirectory, {
+    externals: [dataclassesModule],
     get children() {
-      return _$createComponent(py.DataclassDeclaration, {
-        name: "User",
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
         get children() {
-          return _$createComponent(py.VariableDeclaration, {
-            name: "name",
-            type: "str"
+          return _$createComponent(py.DataclassDeclaration, {
+            name: "User",
+            get children() {
+              return _$createComponent(py.VariableDeclaration, {
+                name: "name",
+                type: "str"
+              });
+            }
           });
         }
       });
     }
-  });
-  expect(toSourceText(content, {
-    externals: [dataclassesModule]
-  })).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     from dataclasses import dataclass
 
 
@@ -468,31 +510,35 @@ it("only imports before definition", () => {
     class User:
         name: str = None
 
-
-  `);
+  `
+  });
 });
 it("only imports before non-definition", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
+  expect(_$createComponent(TestOutputDirectory, {
+    externals: [dataclassesModule],
     get children() {
-      return _$createComponent(py.VariableDeclaration, {
-        name: "x",
-        get initializer() {
-          return _$createComponent(py.Reference, {
-            get refkey() {
-              return dataclassesModule["."].dataclass;
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        get children() {
+          return _$createComponent(py.VariableDeclaration, {
+            name: "x",
+            get initializer() {
+              return _$createComponent(py.Reference, {
+                get refkey() {
+                  return dataclassesModule["."].dataclass;
+                }
+              });
             }
           });
         }
       });
     }
-  });
-  expect(toSourceText(content, {
-    externals: [dataclassesModule]
-  })).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     from dataclasses import dataclass
 
-    x = dataclass`);
+    x = dataclass`
+  });
 });
 it("doc + futureImports before definition", () => {
   const moduleDoc = _$createComponent(py.ModuleDoc, {
@@ -502,22 +548,26 @@ it("doc + futureImports before definition", () => {
       })];
     }
   });
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    doc: moduleDoc,
-    get futureImports() {
-      return [_$createComponent(py.FutureStatement, {
-        feature: "annotations"
-      })];
-    },
+  expect(_$createComponent(TestOutputDirectory, {
     get children() {
-      return _$createComponent(py.FunctionDeclaration, {
-        name: "hello",
-        children: "pass"
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        doc: moduleDoc,
+        get futureImports() {
+          return [_$createComponent(py.FutureStatement, {
+            feature: "annotations"
+          })];
+        },
+        get children() {
+          return _$createComponent(py.FunctionDeclaration, {
+            name: "hello",
+            children: "pass"
+          });
+        }
       });
     }
-  });
-  expect(toSourceText(content)).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     """
     Module description.
     """
@@ -528,8 +578,8 @@ it("doc + futureImports before definition", () => {
     def hello():
         pass
 
-
-  `);
+  `
+  });
 });
 it("doc + futureImports before non-definition", () => {
   const moduleDoc = _$createComponent(py.ModuleDoc, {
@@ -539,29 +589,34 @@ it("doc + futureImports before non-definition", () => {
       })];
     }
   });
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    doc: moduleDoc,
-    get futureImports() {
-      return [_$createComponent(py.FutureStatement, {
-        feature: "annotations"
-      })];
-    },
+  expect(_$createComponent(TestOutputDirectory, {
     get children() {
-      return _$createComponent(py.VariableDeclaration, {
-        name: "x",
-        initializer: 42
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        doc: moduleDoc,
+        get futureImports() {
+          return [_$createComponent(py.FutureStatement, {
+            feature: "annotations"
+          })];
+        },
+        get children() {
+          return _$createComponent(py.VariableDeclaration, {
+            name: "x",
+            initializer: 42
+          });
+        }
       });
     }
-  });
-  expect(toSourceText(content)).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     """
     Module description.
     """
 
     from __future__ import annotations
 
-    x = 42`);
+    x = 42`
+  });
 });
 it("doc + imports before definition", () => {
   const moduleDoc = _$createComponent(py.ModuleDoc, {
@@ -571,24 +626,27 @@ it("doc + imports before definition", () => {
       })];
     }
   });
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    doc: moduleDoc,
+  expect(_$createComponent(TestOutputDirectory, {
+    externals: [dataclassesModule],
     get children() {
-      return _$createComponent(py.DataclassDeclaration, {
-        name: "User",
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        doc: moduleDoc,
         get children() {
-          return _$createComponent(py.VariableDeclaration, {
-            name: "name",
-            type: "str"
+          return _$createComponent(py.DataclassDeclaration, {
+            name: "User",
+            get children() {
+              return _$createComponent(py.VariableDeclaration, {
+                name: "name",
+                type: "str"
+              });
+            }
           });
         }
       });
     }
-  });
-  expect(toSourceText(content, {
-    externals: [dataclassesModule]
-  })).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     """
     Module description.
     """
@@ -600,8 +658,8 @@ it("doc + imports before definition", () => {
     class User:
         name: str = None
 
-
-  `);
+  `
+  });
 });
 it("doc + imports before non-definition", () => {
   const moduleDoc = _$createComponent(py.ModuleDoc, {
@@ -611,56 +669,63 @@ it("doc + imports before non-definition", () => {
       })];
     }
   });
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    doc: moduleDoc,
+  expect(_$createComponent(TestOutputDirectory, {
+    externals: [dataclassesModule],
     get children() {
-      return _$createComponent(py.VariableDeclaration, {
-        name: "x",
-        get initializer() {
-          return _$createComponent(py.Reference, {
-            get refkey() {
-              return dataclassesModule["."].dataclass;
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        doc: moduleDoc,
+        get children() {
+          return _$createComponent(py.VariableDeclaration, {
+            name: "x",
+            get initializer() {
+              return _$createComponent(py.Reference, {
+                get refkey() {
+                  return dataclassesModule["."].dataclass;
+                }
+              });
             }
           });
         }
       });
     }
-  });
-  expect(toSourceText(content, {
-    externals: [dataclassesModule]
-  })).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     """
     Module description.
     """
 
     from dataclasses import dataclass
 
-    x = dataclass`);
+    x = dataclass`
+  });
 });
 it("futureImports + imports before definition", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    get futureImports() {
-      return [_$createComponent(py.FutureStatement, {
-        feature: "annotations"
-      })];
-    },
+  expect(_$createComponent(TestOutputDirectory, {
+    externals: [dataclassesModule],
     get children() {
-      return _$createComponent(py.DataclassDeclaration, {
-        name: "User",
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        get futureImports() {
+          return [_$createComponent(py.FutureStatement, {
+            feature: "annotations"
+          })];
+        },
         get children() {
-          return _$createComponent(py.VariableDeclaration, {
-            name: "name",
-            type: "str"
+          return _$createComponent(py.DataclassDeclaration, {
+            name: "User",
+            get children() {
+              return _$createComponent(py.VariableDeclaration, {
+                name: "name",
+                type: "str"
+              });
+            }
           });
         }
       });
     }
-  });
-  expect(toSourceText(content, {
-    externals: [dataclassesModule]
-  })).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     from __future__ import annotations
 
     from dataclasses import dataclass
@@ -670,38 +735,42 @@ it("futureImports + imports before definition", () => {
     class User:
         name: str = None
 
-
-  `);
+  `
+  });
 });
 it("futureImports + imports before non-definition", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    get futureImports() {
-      return [_$createComponent(py.FutureStatement, {
-        feature: "annotations"
-      })];
-    },
+  expect(_$createComponent(TestOutputDirectory, {
+    externals: [dataclassesModule],
     get children() {
-      return _$createComponent(py.VariableDeclaration, {
-        name: "x",
-        get initializer() {
-          return _$createComponent(py.Reference, {
-            get refkey() {
-              return dataclassesModule["."].dataclass;
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        get futureImports() {
+          return [_$createComponent(py.FutureStatement, {
+            feature: "annotations"
+          })];
+        },
+        get children() {
+          return _$createComponent(py.VariableDeclaration, {
+            name: "x",
+            get initializer() {
+              return _$createComponent(py.Reference, {
+                get refkey() {
+                  return dataclassesModule["."].dataclass;
+                }
+              });
             }
           });
         }
       });
     }
-  });
-  expect(toSourceText(content, {
-    externals: [dataclassesModule]
-  })).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     from __future__ import annotations
 
     from dataclasses import dataclass
 
-    x = dataclass`);
+    x = dataclass`
+  });
 });
 it("doc + futureImports + imports before definition", () => {
   const moduleDoc = _$createComponent(py.ModuleDoc, {
@@ -711,29 +780,32 @@ it("doc + futureImports + imports before definition", () => {
       })];
     }
   });
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    doc: moduleDoc,
-    get futureImports() {
-      return [_$createComponent(py.FutureStatement, {
-        feature: "annotations"
-      })];
-    },
+  expect(_$createComponent(TestOutputDirectory, {
+    externals: [dataclassesModule],
     get children() {
-      return _$createComponent(py.DataclassDeclaration, {
-        name: "User",
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        doc: moduleDoc,
+        get futureImports() {
+          return [_$createComponent(py.FutureStatement, {
+            feature: "annotations"
+          })];
+        },
         get children() {
-          return _$createComponent(py.VariableDeclaration, {
-            name: "name",
-            type: "str"
+          return _$createComponent(py.DataclassDeclaration, {
+            name: "User",
+            get children() {
+              return _$createComponent(py.VariableDeclaration, {
+                name: "name",
+                type: "str"
+              });
+            }
           });
         }
       });
     }
-  });
-  expect(toSourceText(content, {
-    externals: [dataclassesModule]
-  })).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     """
     Module description.
     """
@@ -747,8 +819,8 @@ it("doc + futureImports + imports before definition", () => {
     class User:
         name: str = None
 
-
-  `);
+  `
+  });
 });
 it("doc + futureImports + imports before non-definition", () => {
   const moduleDoc = _$createComponent(py.ModuleDoc, {
@@ -758,30 +830,33 @@ it("doc + futureImports + imports before non-definition", () => {
       })];
     }
   });
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    doc: moduleDoc,
-    get futureImports() {
-      return [_$createComponent(py.FutureStatement, {
-        feature: "annotations"
-      })];
-    },
+  expect(_$createComponent(TestOutputDirectory, {
+    externals: [dataclassesModule],
     get children() {
-      return _$createComponent(py.VariableDeclaration, {
-        name: "x",
-        get initializer() {
-          return _$createComponent(py.Reference, {
-            get refkey() {
-              return dataclassesModule["."].dataclass;
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        doc: moduleDoc,
+        get futureImports() {
+          return [_$createComponent(py.FutureStatement, {
+            feature: "annotations"
+          })];
+        },
+        get children() {
+          return _$createComponent(py.VariableDeclaration, {
+            name: "x",
+            get initializer() {
+              return _$createComponent(py.Reference, {
+                get refkey() {
+                  return dataclassesModule["."].dataclass;
+                }
+              });
             }
           });
         }
       });
     }
-  });
-  expect(toSourceText(content, {
-    externals: [dataclassesModule]
-  })).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     """
     Module description.
     """
@@ -790,7 +865,8 @@ it("doc + futureImports + imports before non-definition", () => {
 
     from dataclasses import dataclass
 
-    x = dataclass`);
+    x = dataclass`
+  });
 });
 it("only doc in file (no children)", () => {
   const moduleDoc = _$createComponent(py.ModuleDoc, {
@@ -800,42 +876,52 @@ it("only doc in file (no children)", () => {
       })];
     }
   });
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    doc: moduleDoc
-  });
-  expect(toSourceText(content)).toRenderTo(d`
+  expect(_$createComponent(TestOutputDirectory, {
+    get children() {
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        doc: moduleDoc
+      });
+    }
+  })).toRenderTo({
+    "test.py": `
     """
     Module description.
     """
 
-
-  `);
+  `
+  });
 });
 it("only header in file (no children)", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    header: "#!/usr/bin/env python3"
-  });
-  expect(toSourceText(content)).toRenderTo(d`
+  expect(_$createComponent(TestOutputDirectory, {
+    get children() {
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        header: "#!/usr/bin/env python3"
+      });
+    }
+  })).toRenderTo({
+    "test.py": `
     #!/usr/bin/env python3
 
-
-  `);
+  `
+  });
 });
 it("only futureImports in file (no children)", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    get futureImports() {
-      return [_$createComponent(py.FutureStatement, {
-        feature: "annotations"
-      })];
+  expect(_$createComponent(TestOutputDirectory, {
+    get children() {
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        get futureImports() {
+          return [_$createComponent(py.FutureStatement, {
+            feature: "annotations"
+          })];
+        }
+      });
     }
+  })).toRenderTo({
+    "test.py": "from __future__ import annotations"
   });
-  expect(toSourceText(content)).toRenderTo(d`
-    from __future__ import annotations
-
-  `);
 });
 it("doc + futureImports in file (no children)", () => {
   const moduleDoc = _$createComponent(py.ModuleDoc, {
@@ -845,64 +931,75 @@ it("doc + futureImports in file (no children)", () => {
       })];
     }
   });
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    doc: moduleDoc,
-    get futureImports() {
-      return [_$createComponent(py.FutureStatement, {
-        feature: "annotations"
-      })];
+  expect(_$createComponent(TestOutputDirectory, {
+    get children() {
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        doc: moduleDoc,
+        get futureImports() {
+          return [_$createComponent(py.FutureStatement, {
+            feature: "annotations"
+          })];
+        }
+      });
     }
-  });
-  expect(toSourceText(content)).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     """
     Module description.
     """
 
     from __future__ import annotations
-
-  `);
+  `
+  });
 });
 
 // headerComment tests
 it("only headerComment before definition", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    headerComment: "Copyright 2024 My Company",
+  expect(_$createComponent(TestOutputDirectory, {
     get children() {
-      return _$createComponent(py.FunctionDeclaration, {
-        name: "hello",
-        children: "pass"
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        headerComment: "Copyright 2024 My Company",
+        get children() {
+          return _$createComponent(py.FunctionDeclaration, {
+            name: "hello",
+            children: "pass"
+          });
+        }
       });
     }
-  });
-
-  // 2 blank lines before definition (PEP 8)
-  expect(toSourceText(content)).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     # Copyright 2024 My Company
 
 
     def hello():
         pass
 
-
-  `);
+  `
+  });
 });
 it("only headerComment before non-definition", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    headerComment: "Copyright 2024 My Company",
+  expect(_$createComponent(TestOutputDirectory, {
     get children() {
-      return _$createComponent(py.VariableDeclaration, {
-        name: "x",
-        initializer: 42
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        headerComment: "Copyright 2024 My Company",
+        get children() {
+          return _$createComponent(py.VariableDeclaration, {
+            name: "x",
+            initializer: 42
+          });
+        }
       });
     }
-  });
-  expect(toSourceText(content)).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     # Copyright 2024 My Company
 
-    x = 42`);
+    x = 42`
+  });
 });
 it("headerComment + doc before definition", () => {
   const moduleDoc = _$createComponent(py.ModuleDoc, {
@@ -912,20 +1009,22 @@ it("headerComment + doc before definition", () => {
       })];
     }
   });
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    headerComment: "Copyright 2024 My Company",
-    doc: moduleDoc,
+  expect(_$createComponent(TestOutputDirectory, {
     get children() {
-      return _$createComponent(py.FunctionDeclaration, {
-        name: "hello",
-        children: "pass"
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        headerComment: "Copyright 2024 My Company",
+        doc: moduleDoc,
+        get children() {
+          return _$createComponent(py.FunctionDeclaration, {
+            name: "hello",
+            children: "pass"
+          });
+        }
       });
     }
-  });
-
-  // headerComment and doc adjacent, then PEP 8 spacing before definition
-  expect(toSourceText(content)).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     # Copyright 2024 My Company
     """
     Module description.
@@ -935,28 +1034,30 @@ it("headerComment + doc before definition", () => {
     def hello():
         pass
 
-
-  `);
+  `
+  });
 });
 it("headerComment + futureImports before definition", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    headerComment: "Copyright 2024 My Company",
-    get futureImports() {
-      return [_$createComponent(py.FutureStatement, {
-        feature: "annotations"
-      })];
-    },
+  expect(_$createComponent(TestOutputDirectory, {
     get children() {
-      return _$createComponent(py.FunctionDeclaration, {
-        name: "hello",
-        children: "pass"
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        headerComment: "Copyright 2024 My Company",
+        get futureImports() {
+          return [_$createComponent(py.FutureStatement, {
+            feature: "annotations"
+          })];
+        },
+        get children() {
+          return _$createComponent(py.FunctionDeclaration, {
+            name: "hello",
+            children: "pass"
+          });
+        }
       });
     }
-  });
-
-  // headerComment and futureImports adjacent, then PEP 8 spacing
-  expect(toSourceText(content)).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     # Copyright 2024 My Company
 
     from __future__ import annotations
@@ -965,28 +1066,31 @@ it("headerComment + futureImports before definition", () => {
     def hello():
         pass
 
-
-  `);
+  `
+  });
 });
 it("headerComment + imports before definition", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    headerComment: "Copyright 2024 My Company",
+  expect(_$createComponent(TestOutputDirectory, {
+    externals: [dataclassesModule],
     get children() {
-      return _$createComponent(py.DataclassDeclaration, {
-        name: "User",
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        headerComment: "Copyright 2024 My Company",
         get children() {
-          return _$createComponent(py.VariableDeclaration, {
-            name: "name",
-            type: "str"
+          return _$createComponent(py.DataclassDeclaration, {
+            name: "User",
+            get children() {
+              return _$createComponent(py.VariableDeclaration, {
+                name: "name",
+                type: "str"
+              });
+            }
           });
         }
       });
     }
-  });
-  expect(toSourceText(content, {
-    externals: [dataclassesModule]
-  })).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     # Copyright 2024 My Company
 
     from dataclasses import dataclass
@@ -996,8 +1100,8 @@ it("headerComment + imports before definition", () => {
     class User:
         name: str = None
 
-
-  `);
+  `
+  });
 });
 it("headerComment + doc + futureImports before definition", () => {
   const moduleDoc = _$createComponent(py.ModuleDoc, {
@@ -1007,23 +1111,27 @@ it("headerComment + doc + futureImports before definition", () => {
       })];
     }
   });
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    headerComment: "Copyright 2024 My Company",
-    doc: moduleDoc,
-    get futureImports() {
-      return [_$createComponent(py.FutureStatement, {
-        feature: "annotations"
-      })];
-    },
+  expect(_$createComponent(TestOutputDirectory, {
     get children() {
-      return _$createComponent(py.FunctionDeclaration, {
-        name: "hello",
-        children: "pass"
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        headerComment: "Copyright 2024 My Company",
+        doc: moduleDoc,
+        get futureImports() {
+          return [_$createComponent(py.FutureStatement, {
+            feature: "annotations"
+          })];
+        },
+        get children() {
+          return _$createComponent(py.FunctionDeclaration, {
+            name: "hello",
+            children: "pass"
+          });
+        }
       });
     }
-  });
-  expect(toSourceText(content)).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     # Copyright 2024 My Company
     """
     Module description.
@@ -1035,8 +1143,8 @@ it("headerComment + doc + futureImports before definition", () => {
     def hello():
         pass
 
-
-  `);
+  `
+  });
 });
 it("headerComment + doc + futureImports + imports before definition", () => {
   const moduleDoc = _$createComponent(py.ModuleDoc, {
@@ -1046,30 +1154,33 @@ it("headerComment + doc + futureImports + imports before definition", () => {
       })];
     }
   });
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    headerComment: "Copyright 2024 My Company",
-    doc: moduleDoc,
-    get futureImports() {
-      return [_$createComponent(py.FutureStatement, {
-        feature: "annotations"
-      })];
-    },
+  expect(_$createComponent(TestOutputDirectory, {
+    externals: [dataclassesModule],
     get children() {
-      return _$createComponent(py.DataclassDeclaration, {
-        name: "User",
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        headerComment: "Copyright 2024 My Company",
+        doc: moduleDoc,
+        get futureImports() {
+          return [_$createComponent(py.FutureStatement, {
+            feature: "annotations"
+          })];
+        },
         get children() {
-          return _$createComponent(py.VariableDeclaration, {
-            name: "name",
-            type: "str"
+          return _$createComponent(py.DataclassDeclaration, {
+            name: "User",
+            get children() {
+              return _$createComponent(py.VariableDeclaration, {
+                name: "name",
+                type: "str"
+              });
+            }
           });
         }
       });
     }
-  });
-  expect(toSourceText(content, {
-    externals: [dataclassesModule]
-  })).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     # Copyright 2024 My Company
     """
     Module description.
@@ -1084,34 +1195,38 @@ it("headerComment + doc + futureImports + imports before definition", () => {
     class User:
         name: str = None
 
-
-  `);
+  `
+  });
 });
 it("only headerComment in file (no children)", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    headerComment: "Copyright 2024 My Company"
+  expect(_$createComponent(TestOutputDirectory, {
+    get children() {
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        headerComment: "Copyright 2024 My Company"
+      });
+    }
+  })).toRenderTo({
+    "test.py": "# Copyright 2024 My Company"
   });
-  expect(toSourceText(content)).toRenderTo(d`
-    # Copyright 2024 My Company
-
-  `);
 });
 it("header + headerComment before definition", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    header: "#!/usr/bin/env python3",
-    headerComment: "Copyright 2024 My Company",
+  expect(_$createComponent(TestOutputDirectory, {
     get children() {
-      return _$createComponent(py.FunctionDeclaration, {
-        name: "hello",
-        children: "pass"
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        header: "#!/usr/bin/env python3",
+        headerComment: "Copyright 2024 My Company",
+        get children() {
+          return _$createComponent(py.FunctionDeclaration, {
+            name: "hello",
+            children: "pass"
+          });
+        }
       });
     }
-  });
-
-  // 2 blank lines before definition (PEP 8)
-  expect(toSourceText(content)).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     #!/usr/bin/env python3
     # Copyright 2024 My Company
 
@@ -1119,24 +1234,26 @@ it("header + headerComment before definition", () => {
     def hello():
         pass
 
-
-  `);
+  `
+  });
 });
 it("header + headerComment before class definition", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    header: "#!/usr/bin/env python3",
-    headerComment: "Copyright 2024 My Company",
+  expect(_$createComponent(TestOutputDirectory, {
     get children() {
-      return _$createComponent(py.ClassDeclaration, {
-        name: "MyClass",
-        children: "pass"
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        header: "#!/usr/bin/env python3",
+        headerComment: "Copyright 2024 My Company",
+        get children() {
+          return _$createComponent(py.ClassDeclaration, {
+            name: "MyClass",
+            children: "pass"
+          });
+        }
       });
     }
-  });
-
-  // 2 blank lines before definition (PEP 8)
-  expect(toSourceText(content)).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     #!/usr/bin/env python3
     # Copyright 2024 My Company
 
@@ -1144,27 +1261,30 @@ it("header + headerComment before class definition", () => {
     class MyClass:
         pass
 
-
-  `);
+  `
+  });
 });
 it("header + headerComment before non-definition", () => {
-  const content = _$createComponent(py.SourceFile, {
-    path: "test.py",
-    header: "#!/usr/bin/env python3",
-    headerComment: "Copyright 2024 My Company",
+  expect(_$createComponent(TestOutputDirectory, {
     get children() {
-      return _$createComponent(py.VariableDeclaration, {
-        name: "x",
-        initializer: 42
+      return _$createComponent(py.SourceFile, {
+        path: "test.py",
+        header: "#!/usr/bin/env python3",
+        headerComment: "Copyright 2024 My Company",
+        get children() {
+          return _$createComponent(py.VariableDeclaration, {
+            name: "x",
+            initializer: 42
+          });
+        }
       });
     }
-  });
-
-  // 1 blank line for non-definition
-  expect(toSourceText(content)).toRenderTo(d`
+  })).toRenderTo({
+    "test.py": `
     #!/usr/bin/env python3
     # Copyright 2024 My Company
 
-    x = 42`);
+    x = 42`
+  });
 });
 //# sourceMappingURL=sourcefiles.test.js.map