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

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

@@ -1,11 +1,40 @@
 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", () => {
+    expect(_$createComponent(TestOutput, {
+      path: "user.py",
+      externals: [dataclassesModule],
+      get children() {
+        return _$createComponent(py.DataclassDeclaration, {
+          name: "User",
+          frozen: true,
+          decorators: ["@final"]
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 11,
+          columnNumber: 9
+        });
+      }
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 10,
+      columnNumber: 7
+    })).toRenderTo(`
+        from dataclasses import dataclass
+
+
+        @final
+        @dataclass(frozen=True)
+        class User:
+            pass
+
+      `);
+  });
   it("Creates a dataclass with a class doc", () => {
     const doc = _$createComponent(py.ClassDoc, {
       get description() {
@@ -13,35 +42,33 @@ describe("DataclassDeclaration", () => {
           children: "Represents a user."
         }, {
           fileName: import.meta.url,
-          lineNumber: 15,
+          lineNumber: 29,
           columnNumber: 34
         })];
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 15,
+      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: 20,
-          columnNumber: 11
+          lineNumber: 33,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 19,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 32,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -53,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",
@@ -70,8 +97,8 @@ describe("DataclassDeclaration", () => {
               type: "int"
             }, {
               fileName: import.meta.url,
-              lineNumber: 49,
-              columnNumber: 13
+              lineNumber: 56,
+              columnNumber: 11
             }), _$createComponent(py.VariableDeclaration, {
               instanceVariable: true,
               get name() {
@@ -85,8 +112,8 @@ describe("DataclassDeclaration", () => {
               omitNone: true
             }, {
               fileName: import.meta.url,
-              lineNumber: 55,
-              columnNumber: 13
+              lineNumber: 62,
+              columnNumber: 11
             }), _$createComponent(py.VariableDeclaration, {
               instanceVariable: true,
               name: "name",
@@ -94,24 +121,21 @@ describe("DataclassDeclaration", () => {
               initializer: "Anonymous"
             }, {
               fileName: import.meta.url,
-              lineNumber: 61,
-              columnNumber: 13
+              lineNumber: 68,
+              columnNumber: 11
             })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 48,
-          columnNumber: 11
+          lineNumber: 55,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 47,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 54,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
         from typing import TYPE_CHECKING
 
@@ -125,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",
@@ -145,24 +169,21 @@ describe("DataclassDeclaration", () => {
               type: "int"
             }, {
               fileName: import.meta.url,
-              lineNumber: 98,
-              columnNumber: 13
+              lineNumber: 99,
+              columnNumber: 11
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 97,
-          columnNumber: 11
+          lineNumber: 98,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 96,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 97,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -170,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",
@@ -191,18 +212,15 @@ describe("DataclassDeclaration", () => {
           weakrefSlot: false
         }, {
           fileName: import.meta.url,
-          lineNumber: 128,
-          columnNumber: 11
+          lineNumber: 123,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 127,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 122,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -210,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: 165,
-          columnNumber: 13
+          lineNumber: 154,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 164,
-      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",
@@ -244,18 +261,15 @@ describe("DataclassDeclaration", () => {
           weakrefSlot: true
         }, {
           fileName: import.meta.url,
-          lineNumber: 179,
-          columnNumber: 11
+          lineNumber: 165,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 178,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 164,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -263,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",
@@ -276,39 +290,35 @@ describe("DataclassDeclaration", () => {
           eq: false
         }, {
           fileName: import.meta.url,
-          lineNumber: 203,
-          columnNumber: 13
+          lineNumber: 184,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 202,
-      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: 215,
-          columnNumber: 11
+          lineNumber: 193,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 214,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 192,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -316,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",
@@ -331,27 +341,26 @@ describe("DataclassDeclaration", () => {
               name: "__lt__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 240,
-              columnNumber: 15
+              lineNumber: 213,
+              columnNumber: 13
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 239,
-          columnNumber: 13
+          lineNumber: 212,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 238,
-      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",
@@ -361,27 +370,26 @@ describe("DataclassDeclaration", () => {
               name: "__le__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 257,
-              columnNumber: 15
+              lineNumber: 227,
+              columnNumber: 13
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 256,
-          columnNumber: 13
+          lineNumber: 226,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 255,
-      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",
@@ -391,27 +399,26 @@ describe("DataclassDeclaration", () => {
               name: "__gt__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 274,
-              columnNumber: 15
+              lineNumber: 241,
+              columnNumber: 13
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 273,
-          columnNumber: 13
+          lineNumber: 240,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 272,
-      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",
@@ -421,23 +428,21 @@ describe("DataclassDeclaration", () => {
               name: "__ge__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 291,
-              columnNumber: 15
+              lineNumber: 255,
+              columnNumber: 13
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 290,
-          columnNumber: 13
+          lineNumber: 254,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 289,
-      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() {
@@ -445,12 +450,13 @@ describe("DataclassDeclaration", () => {
         name: "__lt__"
       }, {
         fileName: import.meta.url,
-        lineNumber: 304,
+        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",
@@ -458,27 +464,26 @@ describe("DataclassDeclaration", () => {
           get children() {
             return _$createComponent(Wrapper, {}, {
               fileName: import.meta.url,
-              lineNumber: 311,
-              columnNumber: 15
+              lineNumber: 272,
+              columnNumber: 13
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 310,
-          columnNumber: 13
+          lineNumber: 271,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 309,
-      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",
@@ -488,27 +493,26 @@ describe("DataclassDeclaration", () => {
               name: "__hash__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 328,
-              columnNumber: 15
+              lineNumber: 286,
+              columnNumber: 13
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 327,
-          columnNumber: 13
+          lineNumber: 285,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 326,
-      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",
@@ -518,27 +522,26 @@ describe("DataclassDeclaration", () => {
               name: "__setattr__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 345,
-              columnNumber: 15
+              lineNumber: 300,
+              columnNumber: 13
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 344,
-          columnNumber: 13
+          lineNumber: 299,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 343,
-      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",
@@ -548,27 +551,26 @@ describe("DataclassDeclaration", () => {
               name: "__delattr__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 362,
-              columnNumber: 15
+              lineNumber: 314,
+              columnNumber: 13
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 361,
-          columnNumber: 13
+          lineNumber: 313,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 360,
-      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",
@@ -578,27 +580,26 @@ describe("DataclassDeclaration", () => {
               name: "__slots__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 379,
-              columnNumber: 15
+              lineNumber: 328,
+              columnNumber: 13
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 378,
-          columnNumber: 13
+          lineNumber: 327,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 377,
-      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",
@@ -611,24 +612,21 @@ describe("DataclassDeclaration", () => {
               type: "int"
             }, {
               fileName: import.meta.url,
-              lineNumber: 395,
-              columnNumber: 13
+              lineNumber: 341,
+              columnNumber: 11
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 394,
-          columnNumber: 11
+          lineNumber: 340,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 393,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 339,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -636,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: 424,
-          columnNumber: 11
+          lineNumber: 365,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 423,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 364,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -667,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",
@@ -690,8 +685,8 @@ describe("DataclassDeclaration", () => {
               omitNone: true
             }, {
               fileName: import.meta.url,
-              lineNumber: 450,
-              columnNumber: 15
+              lineNumber: 385,
+              columnNumber: 13
             }), _$createComponent(py.VariableDeclaration, {
               instanceVariable: true,
               get name() {
@@ -705,27 +700,26 @@ describe("DataclassDeclaration", () => {
               omitNone: true
             }, {
               fileName: import.meta.url,
-              lineNumber: 456,
-              columnNumber: 15
+              lineNumber: 391,
+              columnNumber: 13
             })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 449,
-          columnNumber: 13
+          lineNumber: 384,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 448,
-      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",
@@ -736,27 +730,26 @@ describe("DataclassDeclaration", () => {
               name: "__lt__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 476,
-              columnNumber: 15
+              lineNumber: 408,
+              columnNumber: 13
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 475,
-          columnNumber: 13
+          lineNumber: 407,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 474,
-      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",
@@ -765,69 +758,65 @@ describe("DataclassDeclaration", () => {
               name: "__lt__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 491,
-              columnNumber: 15
+              lineNumber: 420,
+              columnNumber: 13
             }), _$createComponent(py.DunderMethodDeclaration, {
               name: "__slots__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 492,
-              columnNumber: 15
+              lineNumber: 421,
+              columnNumber: 13
             }), _$createComponent(py.DunderMethodDeclaration, {
               name: "__hash__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 493,
-              columnNumber: 15
+              lineNumber: 422,
+              columnNumber: 13
             }), _$createComponent(py.DunderMethodDeclaration, {
               name: "__setattr__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 494,
-              columnNumber: 15
+              lineNumber: 423,
+              columnNumber: 13
             }), _$createComponent(py.DunderMethodDeclaration, {
               name: "__delattr__"
             }, {
               fileName: import.meta.url,
-              lineNumber: 495,
-              columnNumber: 15
+              lineNumber: 424,
+              columnNumber: 13
             })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 490,
-          columnNumber: 13
+          lineNumber: 419,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 489,
-      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: 508,
-          columnNumber: 11
+          lineNumber: 434,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 507,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 433,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -835,7 +824,6 @@ describe("DataclassDeclaration", () => {
         class User:
             pass
 
-
       `);
   });
   it("Counts KW_ONLY sentinels through wrappers (symbol-level)", () => {
@@ -853,12 +841,13 @@ describe("DataclassDeclaration", () => {
         omitNone: true
       }, {
         fileName: import.meta.url,
-        lineNumber: 530,
+        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",
@@ -876,49 +865,45 @@ describe("DataclassDeclaration", () => {
               omitNone: true
             }, {
               fileName: import.meta.url,
-              lineNumber: 543,
-              columnNumber: 15
+              lineNumber: 464,
+              columnNumber: 13
             }), _$createComponent(Wrapper, {}, {
               fileName: import.meta.url,
-              lineNumber: 549,
-              columnNumber: 15
+              lineNumber: 470,
+              columnNumber: 13
             })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 542,
-          columnNumber: 13
+          lineNumber: 463,
+          columnNumber: 11
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 541,
-      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: 562,
-          columnNumber: 11
+          lineNumber: 480,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 561,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 479,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -926,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: 585,
-          columnNumber: 11
+          lineNumber: 498,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 584,
-      columnNumber: 9
-    })], {
-      externals: [dataclassesModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 497,
+      columnNumber: 7
+    })).toRenderTo(`
         from dataclasses import dataclass
 
 
@@ -957,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: 610,
-              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: 616,
-              columnNumber: 13
-            })];
+              lineNumber: 518,
+              columnNumber: 11
+            });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 609,
-          columnNumber: 11
-        });
-      }
-    }, {
-      fileName: import.meta.url,
-      lineNumber: 608,
-      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: 630,
-                  columnNumber: 17
-                });
+                  lineNumber: 535,
+                  columnNumber: 13
+                }), _$createIntrinsic("hbr", {}, {
+                  fileName: import.meta.url,
+                  lineNumber: 545,
+                  columnNumber: 13
+                }), "return user"];
               }
             }, {
               fileName: import.meta.url,
-              lineNumber: 626,
-              columnNumber: 13
-            }), _$createIntrinsic("hbr", {}, {
-              fileName: import.meta.url,
-              lineNumber: 636,
-              columnNumber: 13
-            }), "return user"];
+              lineNumber: 534,
+              columnNumber: 11
+            });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 625,
-          columnNumber: 11
-        });
+          lineNumber: 533,
+          columnNumber: 9
+        })];
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 624,
-      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
 
-        `
+          `
     });
   });
 });