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

package/dist/dev/test/dataclassdeclarations.test.js

@@ -1,14 +1,14 @@
 import { createComponent as _$createComponent, createIntrinsic as _$createIntrinsic } from "@alloy-js/core/jsx-runtime";
-import { Prose, namekey, refkey } from "@alloy-js/core";
-import { d } from "@alloy-js/core/testing";
+import { Prose, namekey, refkey, render } from "@alloy-js/core";
 import { describe, expect, it } from "vitest";
 import { dataclassesModule } from "../src/builtins/python.js";
 import * as py from "../src/index.js";
-import { assertFileContents, toSourceText, toSourceTextMultiple } from "./utils.js";
+import { TestOutput, TestOutputDirectory } from "./utils.js";
 describe("DataclassDeclaration", () => {
   it("stacks user decorators above @dataclass", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
@@ -16,18 +16,15 @@ describe("DataclassDeclaration", () => {
           decorators: ["@final"]
         }, {
           fileName: import.meta.url,
-          lineNumber: 17,
-          columnNumber: 11
+          lineNumber: 11,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 16,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 10,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -36,7 +33,6 @@ describe("DataclassDeclaration", () => {
         class User:
             pass
 
-
       `);
   });
   it("Creates a dataclass with a class doc", () => {
@@ -46,35 +42,33 @@ describe("DataclassDeclaration", () => {
           children: "Represents a user."
         }, {
           fileName: import.meta.url,
-          lineNumber: 40,
+          lineNumber: 29,
           columnNumber: 34
         })];
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 40,
+      lineNumber: 29,
       columnNumber: 7
     });
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
           doc: doc
         }, {
           fileName: import.meta.url,
-          lineNumber: 45,
-          columnNumber: 11
+          lineNumber: 33,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 44,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 32,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -86,12 +80,12 @@ describe("DataclassDeclaration", () => {
 
             pass
 
-
       `);
   });
   it("Creates a dataclass with fields and defaults", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
@@ -103,8 +97,8 @@ describe("DataclassDeclaration", () => {
               type: "int"
             }, {
               fileName: import.meta.url,
-              lineNumber: 74,
-              columnNumber: 13
+              lineNumber: 56,
+              columnNumber: 11
             }), _$createComponent(py.VariableDeclaration, {
               instanceVariable: true,
               get name() {
@@ -118,8 +112,8 @@ describe("DataclassDeclaration", () => {
               omitNone: true
             }, {
               fileName: import.meta.url,
-              lineNumber: 80,
-              columnNumber: 13
+              lineNumber: 62,
+              columnNumber: 11
             }), _$createComponent(py.VariableDeclaration, {
               instanceVariable: true,
               name: "name",
@@ -127,24 +121,21 @@ describe("DataclassDeclaration", () => {
               initializer: "Anonymous"
             }, {
               fileName: import.meta.url,
-              lineNumber: 86,
-              columnNumber: 13
+              lineNumber: 68,
+              columnNumber: 11
             })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 73,
-          columnNumber: 11
+          lineNumber: 55,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 72,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 54,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
         from typing import TYPE_CHECKING
 
@@ -158,12 +149,12 @@ describe("DataclassDeclaration", () => {
             _: KW_ONLY
             name: str = "Anonymous"
 
-
       `);
   });
   it("Creates a dataclass with keyword arguments", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
@@ -178,24 +169,21 @@ describe("DataclassDeclaration", () => {
               type: "int"
             }, {
               fileName: import.meta.url,
-              lineNumber: 123,
-              columnNumber: 13
+              lineNumber: 99,
+              columnNumber: 11
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 122,
-          columnNumber: 11
+          lineNumber: 98,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 121,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 97,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -203,12 +191,12 @@ describe("DataclassDeclaration", () => {
         class User:
             id: int
 
-
       `);
   });
   it("Creates a dataclass with all keyword arguments", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
@@ -224,18 +212,15 @@ describe("DataclassDeclaration", () => {
           weakrefSlot: false
         }, {
           fileName: import.meta.url,
-          lineNumber: 153,
-          columnNumber: 11
+          lineNumber: 123,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 152,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 122,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -243,33 +228,32 @@ describe("DataclassDeclaration", () => {
         class User:
             pass
 
-
       `);
   });
   it("Throws error when weakref_slot=True without slots=True", () => {
-    expect(() => toSourceText([_$createComponent(py.SourceFile, {
+    expect(() => render(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
           weakrefSlot: true
         }, {
           fileName: import.meta.url,
-          lineNumber: 190,
-          columnNumber: 13
+          lineNumber: 154,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 189,
-      columnNumber: 11
-    })], {
-      externals: [dataclassesModule]
-    })).toThrowError(/weakref_slot=True requires slots=True in @dataclass decorator/);
+      lineNumber: 153,
+      columnNumber: 9
+    }))).toThrowError(/weakref_slot=True requires slots=True in @dataclass decorator/);
   });
   it("Allows weakref_slot=True when slots=True", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
@@ -277,18 +261,15 @@ describe("DataclassDeclaration", () => {
           weakrefSlot: true
         }, {
           fileName: import.meta.url,
-          lineNumber: 204,
-          columnNumber: 11
+          lineNumber: 165,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 203,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 164,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -296,12 +277,12 @@ describe("DataclassDeclaration", () => {
         class User:
             pass
 
-
       `);
   });
   it("Throws error when order=True and eq=False", () => {
-    expect(() => toSourceText([_$createComponent(py.SourceFile, {
+    expect(() => render(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
@@ -309,39 +290,35 @@ describe("DataclassDeclaration", () => {
           eq: false
         }, {
           fileName: import.meta.url,
-          lineNumber: 228,
-          columnNumber: 13
+          lineNumber: 184,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 227,
-      columnNumber: 11
-    })], {
-      externals: [dataclassesModule]
-    })).toThrowError(/order=True requires eq=True/);
+      lineNumber: 183,
+      columnNumber: 9
+    }))).toThrowError(/order=True requires eq=True/);
   });
   it("Creates a dataclass with order=True and no conflicting methods", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
           order: true
         }, {
           fileName: import.meta.url,
-          lineNumber: 240,
-          columnNumber: 11
+          lineNumber: 193,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 239,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 192,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -349,12 +326,12 @@ describe("DataclassDeclaration", () => {
         class User:
             pass
 
-
       `);
   });
   it("Throws error when order=True and class defines __lt__", () => {
-    expect(() => toSourceText([_$createComponent(py.SourceFile, {
+    expect(() => render(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
@@ -364,27 +341,26 @@ describe("DataclassDeclaration", () => {
               name: "__lt__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 265,
-              columnNumber: 15
+              lineNumber: 213,
+              columnNumber: 13
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 264,
-          columnNumber: 13
+          lineNumber: 212,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 263,
-      columnNumber: 11
-    })], {
-      externals: [dataclassesModule]
-    })).toThrowError(/Cannot specify order=True when the class already defines __lt__\(\)/);
+      lineNumber: 211,
+      columnNumber: 9
+    }))).toThrowError(/Cannot specify order=True when the class already defines __lt__\(\)/);
   });
   it("Throws error when order=True and class defines __le__", () => {
-    expect(() => toSourceText([_$createComponent(py.SourceFile, {
+    expect(() => render(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
@@ -394,27 +370,26 @@ describe("DataclassDeclaration", () => {
               name: "__le__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 282,
-              columnNumber: 15
+              lineNumber: 227,
+              columnNumber: 13
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 281,
-          columnNumber: 13
+          lineNumber: 226,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 280,
-      columnNumber: 11
-    })], {
-      externals: [dataclassesModule]
-    })).toThrowError(/Cannot specify order=True when the class already defines __le__\(\)/);
+      lineNumber: 225,
+      columnNumber: 9
+    }))).toThrowError(/Cannot specify order=True when the class already defines __le__\(\)/);
   });
   it("Throws error when order=True and class defines __gt__", () => {
-    expect(() => toSourceText([_$createComponent(py.SourceFile, {
+    expect(() => render(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
@@ -424,27 +399,26 @@ describe("DataclassDeclaration", () => {
               name: "__gt__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 299,
-              columnNumber: 15
+              lineNumber: 241,
+              columnNumber: 13
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 298,
-          columnNumber: 13
+          lineNumber: 240,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 297,
-      columnNumber: 11
-    })], {
-      externals: [dataclassesModule]
-    })).toThrowError(/Cannot specify order=True when the class already defines __gt__\(\)/);
+      lineNumber: 239,
+      columnNumber: 9
+    }))).toThrowError(/Cannot specify order=True when the class already defines __gt__\(\)/);
   });
   it("Throws error when order=True and class defines __ge__", () => {
-    expect(() => toSourceText([_$createComponent(py.SourceFile, {
+    expect(() => render(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
@@ -454,23 +428,21 @@ describe("DataclassDeclaration", () => {
               name: "__ge__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 316,
-              columnNumber: 15
+              lineNumber: 255,
+              columnNumber: 13
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 315,
-          columnNumber: 13
+          lineNumber: 254,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 314,
-      columnNumber: 11
-    })], {
-      externals: [dataclassesModule]
-    })).toThrowError(/Cannot specify order=True when the class already defines __ge__\(\)/);
+      lineNumber: 253,
+      columnNumber: 9
+    }))).toThrowError(/Cannot specify order=True when the class already defines __ge__\(\)/);
   });
   it("Throws error when order=True and a wrapper defines __lt__", () => {
     function Wrapper() {
@@ -478,12 +450,13 @@ describe("DataclassDeclaration", () => {
         name: "__lt__"
       }, {
         fileName: import.meta.url,
-        lineNumber: 329,
+        lineNumber: 266,
         columnNumber: 14
       });
     }
-    expect(() => toSourceText([_$createComponent(py.SourceFile, {
+    expect(() => render(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
@@ -491,27 +464,26 @@ describe("DataclassDeclaration", () => {
           get children() {
             return _$createComponent(Wrapper, {}, {
               fileName: import.meta.url,
-              lineNumber: 336,
-              columnNumber: 15
+              lineNumber: 272,
+              columnNumber: 13
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 335,
-          columnNumber: 13
+          lineNumber: 271,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 334,
-      columnNumber: 11
-    })], {
-      externals: [dataclassesModule]
-    })).toThrowError(/Cannot specify order=True when the class already defines __lt__\(\)/);
+      lineNumber: 270,
+      columnNumber: 9
+    }))).toThrowError(/Cannot specify order=True when the class already defines __lt__\(\)/);
   });
   it("Throws error when unsafe_hash=True and class defines __hash__", () => {
-    expect(() => toSourceText([_$createComponent(py.SourceFile, {
+    expect(() => render(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
@@ -521,27 +493,26 @@ describe("DataclassDeclaration", () => {
               name: "__hash__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 353,
-              columnNumber: 15
+              lineNumber: 286,
+              columnNumber: 13
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 352,
-          columnNumber: 13
+          lineNumber: 285,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 351,
-      columnNumber: 11
-    })], {
-      externals: [dataclassesModule]
-    })).toThrowError(/Cannot specify unsafe_hash=True when the class already defines __hash__\(\)/);
+      lineNumber: 284,
+      columnNumber: 9
+    }))).toThrowError(/Cannot specify unsafe_hash=True when the class already defines __hash__\(\)/);
   });
   it("Throws error when frozen=True and class defines __setattr__", () => {
-    expect(() => toSourceText([_$createComponent(py.SourceFile, {
+    expect(() => render(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
@@ -551,27 +522,26 @@ describe("DataclassDeclaration", () => {
               name: "__setattr__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 370,
-              columnNumber: 15
+              lineNumber: 300,
+              columnNumber: 13
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 369,
-          columnNumber: 13
+          lineNumber: 299,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 368,
-      columnNumber: 11
-    })], {
-      externals: [dataclassesModule]
-    })).toThrowError(/Cannot specify frozen=True when the class already defines __setattr__\(\)/);
+      lineNumber: 298,
+      columnNumber: 9
+    }))).toThrowError(/Cannot specify frozen=True when the class already defines __setattr__\(\)/);
   });
   it("Throws errorwhen frozen=True and class defines __delattr__", () => {
-    expect(() => toSourceText([_$createComponent(py.SourceFile, {
+    expect(() => render(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
@@ -581,27 +551,26 @@ describe("DataclassDeclaration", () => {
               name: "__delattr__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 387,
-              columnNumber: 15
+              lineNumber: 314,
+              columnNumber: 13
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 386,
-          columnNumber: 13
+          lineNumber: 313,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 385,
-      columnNumber: 11
-    })], {
-      externals: [dataclassesModule]
-    })).toThrowError(/Cannot specify frozen=True when the class already defines __delattr__\(\)/);
+      lineNumber: 312,
+      columnNumber: 9
+    }))).toThrowError(/Cannot specify frozen=True when the class already defines __delattr__\(\)/);
   });
   it("Throws error when slots=True and class defines __slots__", () => {
-    expect(() => toSourceText([_$createComponent(py.SourceFile, {
+    expect(() => render(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
@@ -611,27 +580,26 @@ describe("DataclassDeclaration", () => {
               name: "__slots__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 404,
-              columnNumber: 15
+              lineNumber: 328,
+              columnNumber: 13
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 403,
-          columnNumber: 13
+          lineNumber: 327,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 402,
-      columnNumber: 11
-    })], {
-      externals: [dataclassesModule]
-    })).toThrowError(/Cannot specify slots=True when the class already defines __slots__\(\)/);
+      lineNumber: 326,
+      columnNumber: 9
+    }))).toThrowError(/Cannot specify slots=True when the class already defines __slots__\(\)/);
   });
   it("Creates a dataclass with kw_only=True on decorator (sentinel not used)", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
@@ -644,24 +612,21 @@ describe("DataclassDeclaration", () => {
               type: "int"
             }, {
               fileName: import.meta.url,
-              lineNumber: 420,
-              columnNumber: 13
+              lineNumber: 341,
+              columnNumber: 11
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 419,
-          columnNumber: 11
+          lineNumber: 340,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 418,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 339,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -669,30 +634,27 @@ describe("DataclassDeclaration", () => {
         class User:
             id: int
 
-
       `);
   });
   it("Creates a dataclass with base classes", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
           bases: ["Base"]
         }, {
           fileName: import.meta.url,
-          lineNumber: 449,
-          columnNumber: 11
+          lineNumber: 365,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 448,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 364,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -700,12 +662,12 @@ describe("DataclassDeclaration", () => {
         class User(Base):
             pass
 
-
       `);
   });
   it("Throws error when more than one KW_ONLY sentinel is present", () => {
-    expect(() => toSourceText([_$createComponent(py.SourceFile, {
+    expect(() => render(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
@@ -723,8 +685,8 @@ describe("DataclassDeclaration", () => {
               omitNone: true
             }, {
               fileName: import.meta.url,
-              lineNumber: 475,
-              columnNumber: 15
+              lineNumber: 385,
+              columnNumber: 13
             }), _$createComponent(py.VariableDeclaration, {
               instanceVariable: true,
               get name() {
@@ -738,27 +700,26 @@ describe("DataclassDeclaration", () => {
               omitNone: true
             }, {
               fileName: import.meta.url,
-              lineNumber: 481,
-              columnNumber: 15
+              lineNumber: 391,
+              columnNumber: 13
             })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 474,
-          columnNumber: 13
+          lineNumber: 384,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 473,
-      columnNumber: 11
-    })], {
-      externals: [dataclassesModule]
-    })).toThrowError(/Only one KW_ONLY sentinel is allowed per dataclass body/);
+      lineNumber: 383,
+      columnNumber: 9
+    }))).toThrowError(/Only one KW_ONLY sentinel is allowed per dataclass body/);
   });
   it("Will raise arg validation errors first over member conflicts", () => {
-    expect(() => toSourceText([_$createComponent(py.SourceFile, {
+    expect(() => render(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
@@ -769,27 +730,26 @@ describe("DataclassDeclaration", () => {
               name: "__lt__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 501,
-              columnNumber: 15
+              lineNumber: 408,
+              columnNumber: 13
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 500,
-          columnNumber: 13
+          lineNumber: 407,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 499,
-      columnNumber: 11
-    })], {
-      externals: [dataclassesModule]
-    })).toThrowError(/order=True requires eq=True/);
+      lineNumber: 406,
+      columnNumber: 9
+    }))).toThrowError(/order=True requires eq=True/);
   });
   it("Does not raise errors for member conflict checks without the equivalent kwargs", () => {
-    expect(() => toSourceText([_$createComponent(py.SourceFile, {
+    expect(() => render(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
@@ -798,69 +758,65 @@ describe("DataclassDeclaration", () => {
               name: "__lt__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 516,
-              columnNumber: 15
+              lineNumber: 420,
+              columnNumber: 13
             }), _$createComponent(py.DunderMethodDeclaration, {
               name: "__slots__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 517,
-              columnNumber: 15
+              lineNumber: 421,
+              columnNumber: 13
             }), _$createComponent(py.DunderMethodDeclaration, {
               name: "__hash__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 518,
-              columnNumber: 15
+              lineNumber: 422,
+              columnNumber: 13
             }), _$createComponent(py.DunderMethodDeclaration, {
               name: "__setattr__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 519,
-              columnNumber: 15
+              lineNumber: 423,
+              columnNumber: 13
             }), _$createComponent(py.DunderMethodDeclaration, {
               name: "__delattr__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 520,
-              columnNumber: 15
+              lineNumber: 424,
+              columnNumber: 13
             })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 515,
-          columnNumber: 13
+          lineNumber: 419,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 514,
-      columnNumber: 11
-    })], {
-      externals: [dataclassesModule]
-    })).not.toThrow();
+      lineNumber: 418,
+      columnNumber: 9
+    }))).not.toThrow();
   });
   it("Allows unsafe_hash=True when no __hash__ is defined", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
           unsafeHash: true
         }, {
           fileName: import.meta.url,
-          lineNumber: 533,
-          columnNumber: 11
+          lineNumber: 434,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 532,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 433,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -868,7 +824,6 @@ describe("DataclassDeclaration", () => {
         class User:
             pass
 
-
       `);
   });
   it("Counts KW_ONLY sentinels through wrappers (symbol-level)", () => {
@@ -886,12 +841,13 @@ describe("DataclassDeclaration", () => {
         omitNone: true
       }, {
         fileName: import.meta.url,
-        lineNumber: 555,
+        lineNumber: 452,
         columnNumber: 9
       });
     }
-    expect(() => toSourceText([_$createComponent(py.SourceFile, {
+    expect(() => render(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
@@ -909,49 +865,45 @@ describe("DataclassDeclaration", () => {
               omitNone: true
             }, {
               fileName: import.meta.url,
-              lineNumber: 568,
-              columnNumber: 15
+              lineNumber: 464,
+              columnNumber: 13
             }), _$createComponent(Wrapper, {}, {
               fileName: import.meta.url,
-              lineNumber: 574,
-              columnNumber: 15
+              lineNumber: 470,
+              columnNumber: 13
             })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 567,
-          columnNumber: 13
+          lineNumber: 463,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 566,
-      columnNumber: 11
-    })], {
-      externals: [dataclassesModule]
-    })).toThrowError(/Only one KW_ONLY sentinel is allowed per dataclass body/);
+      lineNumber: 462,
+      columnNumber: 9
+    }))).toThrowError(/Only one KW_ONLY sentinel is allowed per dataclass body/);
   });
   it("Allows frozen=True when no conflicting dunders exist", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
           frozen: true
         }, {
           fileName: import.meta.url,
-          lineNumber: 587,
-          columnNumber: 11
+          lineNumber: 480,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 586,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 479,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -959,30 +911,27 @@ describe("DataclassDeclaration", () => {
         class User:
             pass
 
-
       `);
   });
   it("Allows slots=True when no __slots__ is defined", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "user.py",
+      externals: [dataclassesModule],
       get children() {
         return _$createComponent(py.DataclassDeclaration, {
           name: "User",
           slots: true
         }, {
           fileName: import.meta.url,
-          lineNumber: 610,
-          columnNumber: 11
+          lineNumber: 498,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 609,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 497,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -990,111 +939,116 @@ describe("DataclassDeclaration", () => {
         class User:
             pass
 
-
       `);
   });
   it("Forwards refkey prop for symbol resolution in type references", () => {
     const userRefkey = refkey();
-    const res = toSourceTextMultiple([_$createComponent(py.SourceFile, {
-      path: "models.py",
+    expect(_$createComponent(TestOutputDirectory, {
+      externals: [dataclassesModule],
       get children() {
-        return _$createComponent(py.DataclassDeclaration, {
-          name: "User",
-          refkey: userRefkey,
+        return [_$createComponent(py.SourceFile, {
+          path: "models.py",
           get children() {
-            return [_$createComponent(py.VariableDeclaration, {
-              instanceVariable: true,
-              omitNone: true,
-              name: "id",
-              type: "int"
-            }, {
-              fileName: import.meta.url,
-              lineNumber: 635,
-              columnNumber: 13
-            }), _$createComponent(py.VariableDeclaration, {
-              instanceVariable: true,
-              omitNone: true,
-              name: "name",
-              type: "str"
+            return _$createComponent(py.DataclassDeclaration, {
+              name: "User",
+              refkey: userRefkey,
+              get children() {
+                return [_$createComponent(py.VariableDeclaration, {
+                  instanceVariable: true,
+                  omitNone: true,
+                  name: "id",
+                  type: "int"
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 519,
+                  columnNumber: 13
+                }), _$createComponent(py.VariableDeclaration, {
+                  instanceVariable: true,
+                  omitNone: true,
+                  name: "name",
+                  type: "str"
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 525,
+                  columnNumber: 13
+                })];
+              }
             }, {
               fileName: import.meta.url,
-              lineNumber: 641,
-              columnNumber: 13
-            })];
+              lineNumber: 518,
+              columnNumber: 11
+            });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 634,
-          columnNumber: 11
-        });
-      }
-    }, {
-      fileName: import.meta.url,
-      lineNumber: 633,
-      columnNumber: 9
-    }), _$createComponent(py.SourceFile, {
-      path: "services.py",
-      get children() {
-        return _$createComponent(py.FunctionDeclaration, {
-          name: "get_user",
-          returnType: userRefkey,
+          lineNumber: 517,
+          columnNumber: 9
+        }), _$createComponent(py.SourceFile, {
+          path: "services.py",
           get children() {
-            return [_$createComponent(py.VariableDeclaration, {
-              name: "user",
-              type: userRefkey,
-              get initializer() {
-                return _$createComponent(py.ClassInstantiation, {
-                  target: userRefkey,
-                  args: ["1", '"Alice"']
+            return _$createComponent(py.FunctionDeclaration, {
+              name: "get_user",
+              returnType: userRefkey,
+              get children() {
+                return [_$createComponent(py.VariableDeclaration, {
+                  name: "user",
+                  type: userRefkey,
+                  get initializer() {
+                    return _$createComponent(py.ClassInstantiation, {
+                      target: userRefkey,
+                      args: ["1", '"Alice"']
+                    }, {
+                      fileName: import.meta.url,
+                      lineNumber: 539,
+                      columnNumber: 17
+                    });
+                  }
                 }, {
                   fileName: import.meta.url,
-                  lineNumber: 655,
-                  columnNumber: 17
-                });
+                  lineNumber: 535,
+                  columnNumber: 13
+                }), _$createIntrinsic("hbr", {}, {
+                  fileName: import.meta.url,
+                  lineNumber: 545,
+                  columnNumber: 13
+                }), "return user"];
               }
             }, {
               fileName: import.meta.url,
-              lineNumber: 651,
-              columnNumber: 13
-            }), _$createIntrinsic("hbr", {}, {
-              fileName: import.meta.url,
-              lineNumber: 661,
-              columnNumber: 13
-            }), "return user"];
+              lineNumber: 534,
+              columnNumber: 11
+            });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 650,
-          columnNumber: 11
-        });
+          lineNumber: 533,
+          columnNumber: 9
+        })];
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 649,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    assertFileContents(res, {
+      lineNumber: 516,
+      columnNumber: 7
+    })).toRenderTo({
       "models.py": `
-        from dataclasses import dataclass
+          from dataclasses import dataclass
 
 
-        @dataclass
-        class User:
-            id: int
-            name: str
+          @dataclass
+          class User:
+              id: int
+              name: str
 
-        `,
+          `,
       "services.py": `
-        from models import User
+          from models import User
 
 
-        def get_user() -> User:
-            user: User = User(1, "Alice")
-            return user
+          def get_user() -> User:
+              user: User = User(1, "Alice")
+              return user
 
-        `
+          `
     });
   });
 });