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

package/dist/dev/test/pydanticclassdeclarations.test.js

@@ -1,14 +1,14 @@
 import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
 import { Prose, code, refkey } from "@alloy-js/core";
-import { d } from "@alloy-js/core/testing";
 import { describe, expect, it } from "vitest";
 import { pydanticModule, pydanticSettingsModule, typingModule } 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("PydanticClassDeclaration", () => {
   it("forwards class-level decorators above `class`", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule, typingModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "User",
@@ -23,24 +23,21 @@ describe("PydanticClassDeclaration", () => {
               type: "int"
             }, {
               fileName: import.meta.url,
-              lineNumber: 25,
-              columnNumber: 13
+              lineNumber: 19,
+              columnNumber: 11
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 21,
-          columnNumber: 11
+          lineNumber: 15,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 20,
-      columnNumber: 9
-    })], {
-      externals: [pydanticModule, typingModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 14,
+      columnNumber: 7
+    })).toRenderTo(`
         from pydantic import BaseModel
         from typing import final
 
@@ -49,12 +46,12 @@ describe("PydanticClassDeclaration", () => {
         class User(BaseModel):
             id: int
 
-
       `);
   });
   it("emits a pydantic model with BaseModel, fields, and Field import", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "User",
@@ -66,8 +63,8 @@ describe("PydanticClassDeclaration", () => {
               type: "int"
             }, {
               fileName: import.meta.url,
-              lineNumber: 57,
-              columnNumber: 13
+              lineNumber: 45,
+              columnNumber: 11
             }), _$createComponent(py.VariableDeclaration, {
               instanceVariable: true,
               name: "name",
@@ -77,24 +74,21 @@ describe("PydanticClassDeclaration", () => {
               }
             }, {
               fileName: import.meta.url,
-              lineNumber: 63,
-              columnNumber: 13
+              lineNumber: 51,
+              columnNumber: 11
             })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 56,
-          columnNumber: 11
+          lineNumber: 44,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 55,
-      columnNumber: 9
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 43,
+      columnNumber: 7
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import Field
 
@@ -103,13 +97,13 @@ describe("PydanticClassDeclaration", () => {
             id: int
             name: str = Field(default="anon")
 
-
       `);
   });
   it("inherits from another pydantic model via bases", () => {
     const baseRef = refkey();
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.StatementList, {
           get children() {
@@ -124,14 +118,14 @@ describe("PydanticClassDeclaration", () => {
                   type: "int"
                 }, {
                   fileName: import.meta.url,
-                  lineNumber: 97,
-                  columnNumber: 15
+                  lineNumber: 79,
+                  columnNumber: 13
                 });
               }
             }, {
               fileName: import.meta.url,
-              lineNumber: 96,
-              columnNumber: 13
+              lineNumber: 78,
+              columnNumber: 11
             }), _$createComponent(py.PydanticClassDeclaration, {
               name: "Admin",
               bases: [baseRef],
@@ -143,30 +137,27 @@ describe("PydanticClassDeclaration", () => {
                   type: "str"
                 }, {
                   fileName: import.meta.url,
-                  lineNumber: 105,
-                  columnNumber: 15
+                  lineNumber: 87,
+                  columnNumber: 13
                 });
               }
             }, {
               fileName: import.meta.url,
-              lineNumber: 104,
-              columnNumber: 13
+              lineNumber: 86,
+              columnNumber: 11
             })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 95,
-          columnNumber: 11
+          lineNumber: 77,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 94,
-      columnNumber: 9
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 76,
+      columnNumber: 7
+    })).toRenderTo(`
         from pydantic import BaseModel
 
 
@@ -176,12 +167,12 @@ describe("PydanticClassDeclaration", () => {
         class Admin(User):
             role: str
 
-
       `);
   });
   it("supports required, optional, Field default, and plain default", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "Item",
@@ -193,8 +184,8 @@ describe("PydanticClassDeclaration", () => {
               type: "str"
             }, {
               fileName: import.meta.url,
-              lineNumber: 139,
-              columnNumber: 13
+              lineNumber: 115,
+              columnNumber: 11
             }), _$createComponent(py.VariableDeclaration, {
               instanceVariable: true,
               name: "notes",
@@ -203,14 +194,14 @@ describe("PydanticClassDeclaration", () => {
                   children: ["str", "None"]
                 }, {
                   fileName: import.meta.url,
-                  lineNumber: 148,
-                  columnNumber: 21
+                  lineNumber: 124,
+                  columnNumber: 19
                 });
               }
             }, {
               fileName: import.meta.url,
-              lineNumber: 145,
-              columnNumber: 13
+              lineNumber: 121,
+              columnNumber: 11
             }), _$createComponent(py.VariableDeclaration, {
               instanceVariable: true,
               name: "label",
@@ -220,8 +211,8 @@ describe("PydanticClassDeclaration", () => {
               }
             }, {
               fileName: import.meta.url,
-              lineNumber: 150,
-              columnNumber: 13
+              lineNumber: 126,
+              columnNumber: 11
             }), _$createComponent(py.VariableDeclaration, {
               instanceVariable: true,
               name: "qty",
@@ -229,24 +220,21 @@ describe("PydanticClassDeclaration", () => {
               initializer: 1
             }, {
               fileName: import.meta.url,
-              lineNumber: 156,
-              columnNumber: 13
+              lineNumber: 132,
+              columnNumber: 11
             })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 138,
-          columnNumber: 11
+          lineNumber: 114,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 137,
-      columnNumber: 9
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 113,
+      columnNumber: 7
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import Field
 
@@ -257,7 +245,6 @@ describe("PydanticClassDeclaration", () => {
             label: str = Field(default="untitled")
             qty: int = 1
 
-
       `);
   });
   it("emits class docstring", () => {
@@ -267,17 +254,18 @@ describe("PydanticClassDeclaration", () => {
           children: "Payload for an API request."
         }, {
           fileName: import.meta.url,
-          lineNumber: 187,
+          lineNumber: 158,
           columnNumber: 34
         })];
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 187,
+      lineNumber: 158,
       columnNumber: 7
     });
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "RequestBody",
@@ -290,24 +278,21 @@ describe("PydanticClassDeclaration", () => {
               type: "str"
             }, {
               fileName: import.meta.url,
-              lineNumber: 193,
-              columnNumber: 13
+              lineNumber: 163,
+              columnNumber: 11
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 192,
-          columnNumber: 11
+          lineNumber: 162,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 191,
-      columnNumber: 9
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 161,
+      columnNumber: 7
+    })).toRenderTo(`
         from pydantic import BaseModel
 
 
@@ -318,88 +303,94 @@ describe("PydanticClassDeclaration", () => {
 
             value: str
 
-
       `);
   });
   it("resolves refkey across files with pydantic imports", () => {
     const modelRef = refkey();
-    const res = toSourceTextMultiple([_$createComponent(py.SourceFile, {
-      path: "models.py",
+    expect(_$createComponent(TestOutputDirectory, {
+      externals: [pydanticModule],
       get children() {
-        return _$createComponent(py.PydanticClassDeclaration, {
-          name: "User",
-          refkey: modelRef,
+        return [_$createComponent(py.SourceFile, {
+          path: "models.py",
           get children() {
-            return _$createComponent(py.VariableDeclaration, {
-              instanceVariable: true,
-              omitNone: true,
-              name: "id",
-              type: "int"
+            return _$createComponent(py.PydanticClassDeclaration, {
+              name: "User",
+              refkey: modelRef,
+              get children() {
+                return _$createComponent(py.VariableDeclaration, {
+                  instanceVariable: true,
+                  omitNone: true,
+                  name: "id",
+                  type: "int"
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 193,
+                  columnNumber: 13
+                });
+              }
             }, {
               fileName: import.meta.url,
-              lineNumber: 228,
-              columnNumber: 13
+              lineNumber: 192,
+              columnNumber: 11
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 227,
-          columnNumber: 11
-        });
-      }
-    }, {
-      fileName: import.meta.url,
-      lineNumber: 226,
-      columnNumber: 9
-    }), _$createComponent(py.SourceFile, {
-      path: "service.py",
-      get children() {
-        return _$createComponent(py.FunctionDeclaration, {
-          name: "load_user",
-          returnType: modelRef,
-          parameters: [{
-            name: "user_id",
-            type: "int"
-          }],
-          children: "return User(id=user_id)"
+          lineNumber: 191,
+          columnNumber: 9
+        }), _$createComponent(py.SourceFile, {
+          path: "service.py",
+          get children() {
+            return _$createComponent(py.FunctionDeclaration, {
+              name: "load_user",
+              returnType: modelRef,
+              parameters: [{
+                name: "user_id",
+                type: "int"
+              }],
+              children: "return User(id=user_id)"
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 202,
+              columnNumber: 11
+            });
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 237,
-          columnNumber: 11
-        });
+          lineNumber: 201,
+          columnNumber: 9
+        })];
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 236,
-      columnNumber: 9
-    })], {
-      externals: [pydanticModule]
-    });
-    assertFileContents(res, {
+      lineNumber: 190,
+      columnNumber: 7
+    })).toRenderTo({
       "models.py": `
-        from pydantic import BaseModel
+          from pydantic import BaseModel
 
 
-        class User(BaseModel):
-            id: int
+          class User(BaseModel):
+              id: int
 
-        `,
+          `,
       "service.py": `
-        from typing import TYPE_CHECKING
+          from typing import TYPE_CHECKING
 
-        if TYPE_CHECKING:
-            from models import User
+          if TYPE_CHECKING:
+              from models import User
 
 
-        def load_user(user_id: int) -> User:
-            return User(id=user_id)
+          def load_user(user_id: int) -> User:
+              return User(id=user_id)
 
-        `
+          `
     });
   });
   it("emits model_config = ConfigDict(...) from top-level config props", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "User",
@@ -414,24 +405,21 @@ describe("PydanticClassDeclaration", () => {
               type: "int"
             }, {
               fileName: import.meta.url,
-              lineNumber: 282,
-              columnNumber: 13
+              lineNumber: 243,
+              columnNumber: 11
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 276,
-          columnNumber: 11
+          lineNumber: 237,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 275,
-      columnNumber: 9
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 236,
+      columnNumber: 7
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import ConfigDict
 
@@ -440,12 +428,12 @@ describe("PydanticClassDeclaration", () => {
             model_config = ConfigDict(frozen=True, extra="forbid", validate_assignment=True)
             id: int
 
-
       `);
   });
   it("emits model_config = ConfigDict(...) from modelConfig", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "User",
@@ -462,24 +450,21 @@ describe("PydanticClassDeclaration", () => {
               type: "int"
             }, {
               fileName: import.meta.url,
-              lineNumber: 321,
-              columnNumber: 13
+              lineNumber: 276,
+              columnNumber: 11
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 313,
-          columnNumber: 11
+          lineNumber: 268,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 312,
-      columnNumber: 9
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 267,
+      columnNumber: 7
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import ConfigDict
 
@@ -488,12 +473,12 @@ describe("PydanticClassDeclaration", () => {
             model_config = ConfigDict(frozen=True, extra="forbid", validate_assignment=True)
             id: int
 
-
       `);
   });
   it("gives precedence to top-level config props over modelConfig", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "User",
@@ -505,18 +490,15 @@ describe("PydanticClassDeclaration", () => {
           extra: "forbid"
         }, {
           fileName: import.meta.url,
-          lineNumber: 352,
-          columnNumber: 11
+          lineNumber: 301,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 351,
-      columnNumber: 9
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 300,
+      columnNumber: 7
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import ConfigDict
 
@@ -524,12 +506,12 @@ describe("PydanticClassDeclaration", () => {
         class User(BaseModel):
             model_config = ConfigDict(frozen=True, extra="forbid")
 
-
       `);
   });
   it("supports additional typed ConfigDict props", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "User",
@@ -542,18 +524,15 @@ describe("PydanticClassDeclaration", () => {
           valJsonBytes: "hex"
         }, {
           fileName: import.meta.url,
-          lineNumber: 381,
-          columnNumber: 11
+          lineNumber: 324,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 380,
-      columnNumber: 9
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 323,
+      columnNumber: 7
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import ConfigDict
 
@@ -561,12 +540,12 @@ describe("PydanticClassDeclaration", () => {
         class User(BaseModel):
             model_config = ConfigDict(coerce_numbers_to_str=True, ser_json_bytes="base64", str_min_length=1, str_max_length=128, use_enum_values=True, val_json_bytes="hex", validate_return=True)
 
-
       `);
   });
   it("imports SecretStr when used as a field type", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "Credentials",
@@ -580,24 +559,21 @@ describe("PydanticClassDeclaration", () => {
               }
             }, {
               fileName: import.meta.url,
-              lineNumber: 415,
-              columnNumber: 13
+              lineNumber: 352,
+              columnNumber: 11
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 414,
-          columnNumber: 11
+          lineNumber: 351,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 413,
-      columnNumber: 9
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 350,
+      columnNumber: 7
+    })).toRenderTo(`
         from pydantic import BaseModel
         from typing import TYPE_CHECKING
 
@@ -608,12 +584,12 @@ describe("PydanticClassDeclaration", () => {
         class Credentials(BaseModel):
             token: SecretStr
 
-
       `);
   });
   it("emits arbitrary model_config via modelConfigExpression", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "M",
@@ -622,18 +598,15 @@ describe("PydanticClassDeclaration", () => {
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 448,
-          columnNumber: 11
+          lineNumber: 379,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 447,
-      columnNumber: 9
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 378,
+      columnNumber: 7
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import ConfigDict
 
@@ -641,12 +614,12 @@ describe("PydanticClassDeclaration", () => {
         class M(BaseModel):
             model_config = ConfigDict(frozen=True, extra="allow")
 
-
       `);
   });
   it("supports RootModel as explicit bases entry", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "Tags",
@@ -655,30 +628,27 @@ describe("PydanticClassDeclaration", () => {
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 475,
-          columnNumber: 11
+          lineNumber: 400,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 474,
-      columnNumber: 9
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 399,
+      columnNumber: 7
+    })).toRenderTo(`
         from pydantic import RootModel
 
 
         class Tags(RootModel[list[str]]):
             pass
 
-
       `);
   });
   it("places Pydantic validators above classmethod", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "User",
@@ -690,8 +660,8 @@ describe("PydanticClassDeclaration", () => {
               type: "str"
             }, {
               fileName: import.meta.url,
-              lineNumber: 502,
-              columnNumber: 13
+              lineNumber: 421,
+              columnNumber: 11
             }), _$createComponent(py.ClassMethodDeclaration, {
               name: "strip_name",
               get decorators() {
@@ -705,24 +675,21 @@ describe("PydanticClassDeclaration", () => {
               children: "return value.strip()"
             }, {
               fileName: import.meta.url,
-              lineNumber: 508,
-              columnNumber: 13
+              lineNumber: 427,
+              columnNumber: 11
             })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 501,
-          columnNumber: 11
+          lineNumber: 420,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 500,
-      columnNumber: 9
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 419,
+      columnNumber: 7
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import field_validator
 
@@ -735,14 +702,14 @@ describe("PydanticClassDeclaration", () => {
                 return value.strip()
 
 
-      
       `);
   });
 });
 describe("Pydantic ecosystem emitters", () => {
   it("typing module resolves Any and similar annotations", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [typingModule],
       get children() {
         return _$createComponent(py.FunctionDeclaration, {
           name: "identity",
@@ -758,18 +725,15 @@ describe("Pydantic ecosystem emitters", () => {
           children: "return x"
         }, {
           fileName: import.meta.url,
-          lineNumber: 549,
-          columnNumber: 11
+          lineNumber: 462,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 548,
-      columnNumber: 9
-    })], {
-      externals: [typingModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 461,
+      columnNumber: 7
+    })).toRenderTo(`
         from typing import TYPE_CHECKING
 
         if TYPE_CHECKING:
@@ -779,12 +743,12 @@ describe("Pydantic ecosystem emitters", () => {
         def identity(x: Any) -> Any:
             return x
 
-
       `);
   });
   it("pydantic.types constrains field annotations", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "Score",
@@ -798,24 +762,21 @@ describe("Pydantic ecosystem emitters", () => {
               }
             }, {
               fileName: import.meta.url,
-              lineNumber: 582,
-              columnNumber: 13
+              lineNumber: 489,
+              columnNumber: 11
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 581,
-          columnNumber: 11
+          lineNumber: 488,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 580,
-      columnNumber: 9
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 487,
+      columnNumber: 7
+    })).toRenderTo(`
         from pydantic import BaseModel
         from typing import TYPE_CHECKING
 
@@ -826,73 +787,80 @@ describe("Pydantic ecosystem emitters", () => {
         class Score(BaseModel):
             points: PositiveInt
 
-
       `);
   });
   it("postponed annotations support forward references in fields", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
-      path: "models.py",
-      get futureImports() {
-        return [_$createComponent(py.FutureStatement, {
-          feature: "annotations"
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 616,
-          columnNumber: 27
-        })];
-      },
+    expect(_$createComponent(TestOutputDirectory, {
+      externals: [pydanticModule],
       get children() {
-        return _$createComponent(py.PydanticClassDeclaration, {
-          name: "Node",
-          get children() {
-            return [_$createComponent(py.VariableDeclaration, {
-              instanceVariable: true,
-              omitNone: true,
-              name: "label",
-              type: "str"
+        return _$createComponent(py.SourceFile, {
+          path: "models.py",
+          get futureImports() {
+            return [_$createComponent(py.FutureStatement, {
+              feature: "annotations"
             }, {
               fileName: import.meta.url,
-              lineNumber: 619,
-              columnNumber: 13
-            }), _$createComponent(py.VariableDeclaration, {
-              instanceVariable: true,
-              name: "child",
-              type: code`"Node" | None`
+              lineNumber: 518,
+              columnNumber: 27
+            })];
+          },
+          get children() {
+            return _$createComponent(py.PydanticClassDeclaration, {
+              name: "Node",
+              get children() {
+                return [_$createComponent(py.VariableDeclaration, {
+                  instanceVariable: true,
+                  omitNone: true,
+                  name: "label",
+                  type: "str"
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 521,
+                  columnNumber: 13
+                }), _$createComponent(py.VariableDeclaration, {
+                  instanceVariable: true,
+                  name: "child",
+                  type: code`"Node" | None`
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 527,
+                  columnNumber: 13
+                })];
+              }
             }, {
               fileName: import.meta.url,
-              lineNumber: 625,
-              columnNumber: 13
-            })];
+              lineNumber: 520,
+              columnNumber: 11
+            });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 618,
-          columnNumber: 11
+          lineNumber: 516,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 614,
-      columnNumber: 9
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
-        from __future__ import annotations
-
-        from pydantic import BaseModel
+      lineNumber: 515,
+      columnNumber: 7
+    })).toRenderTo({
+      "models.py": `
+          from __future__ import annotations
 
+          from pydantic import BaseModel
 
-        class Node(BaseModel):
-            label: str
-            child: "Node" | None = None
 
+          class Node(BaseModel):
+              label: str
+              child: "Node" | None = None
 
-      `);
+        `
+    });
   });
   it("model_config can use pydantic.alias_generators", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "M",
@@ -901,18 +869,15 @@ describe("Pydantic ecosystem emitters", () => {
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 656,
-          columnNumber: 11
+          lineNumber: 553,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 655,
-      columnNumber: 9
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 552,
+      columnNumber: 7
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import ConfigDict
         from pydantic.alias_generators import to_camel
@@ -921,12 +886,12 @@ describe("Pydantic ecosystem emitters", () => {
         class M(BaseModel):
             model_config = ConfigDict(alias_generator=to_camel)
 
-
       `);
   });
   it("pydantic_settings exposes BaseSettings", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "config.py",
+      externals: [pydanticSettingsModule],
       get children() {
         return _$createComponent(py.ClassDeclaration, {
           name: "AppSettings",
@@ -935,30 +900,27 @@ describe("Pydantic ecosystem emitters", () => {
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 684,
-          columnNumber: 11
+          lineNumber: 575,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 683,
-      columnNumber: 9
-    })], {
-      externals: [pydanticSettingsModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 574,
+      columnNumber: 7
+    })).toRenderTo(`
         from pydantic_settings import BaseSettings
 
 
         class AppSettings(BaseSettings):
             pass
 
-
       `);
   });
   it("emits @computed_field above @property via PropertyDeclaration", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "Square",
@@ -970,8 +932,8 @@ describe("Pydantic ecosystem emitters", () => {
               type: "float"
             }, {
               fileName: import.meta.url,
-              lineNumber: 711,
-              columnNumber: 13
+              lineNumber: 596,
+              columnNumber: 11
             }), _$createComponent(py.PropertyDeclaration, {
               name: "area",
               type: "float",
@@ -981,24 +943,21 @@ describe("Pydantic ecosystem emitters", () => {
               children: "return self.width ** 2"
             }, {
               fileName: import.meta.url,
-              lineNumber: 717,
-              columnNumber: 13
+              lineNumber: 602,
+              columnNumber: 11
             })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 710,
-          columnNumber: 11
+          lineNumber: 595,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 709,
-      columnNumber: 9
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 594,
+      columnNumber: 7
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import computed_field
 
@@ -1011,12 +970,12 @@ describe("Pydantic ecosystem emitters", () => {
                 return self.width ** 2
 
 
-      
       `);
   });
   it("emits computed_field on an instance method", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "Square",
@@ -1028,8 +987,8 @@ describe("Pydantic ecosystem emitters", () => {
               type: "float"
             }, {
               fileName: import.meta.url,
-              lineNumber: 754,
-              columnNumber: 13
+              lineNumber: 633,
+              columnNumber: 11
             }), _$createComponent(py.MethodDeclaration, {
               name: "area",
               get decorators() {
@@ -1039,24 +998,21 @@ describe("Pydantic ecosystem emitters", () => {
               children: "return self.width ** 2"
             }, {
               fileName: import.meta.url,
-              lineNumber: 760,
-              columnNumber: 13
+              lineNumber: 639,
+              columnNumber: 11
             })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 753,
-          columnNumber: 11
+          lineNumber: 632,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 752,
-      columnNumber: 9
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 631,
+      columnNumber: 7
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import computed_field
 
@@ -1068,12 +1024,12 @@ describe("Pydantic ecosystem emitters", () => {
                 return self.width ** 2
 
 
-      
       `);
   });
   it("emits model_validator above classmethod", () => {
-    const res = toSourceText([_$createComponent(py.SourceFile, {
+    expect(_$createComponent(TestOutput, {
       path: "models.py",
+      externals: [pydanticModule],
       get children() {
         return _$createComponent(py.PydanticClassDeclaration, {
           name: "Bag",
@@ -1085,8 +1041,8 @@ describe("Pydantic ecosystem emitters", () => {
               type: "list"
             }, {
               fileName: import.meta.url,
-              lineNumber: 796,
-              columnNumber: 13
+              lineNumber: 669,
+              columnNumber: 11
             }), _$createComponent(py.ClassMethodDeclaration, {
               name: "ensure_items",
               get decorators() {
@@ -1100,24 +1056,21 @@ describe("Pydantic ecosystem emitters", () => {
               children: "return data"
             }, {
               fileName: import.meta.url,
-              lineNumber: 802,
-              columnNumber: 13
+              lineNumber: 675,
+              columnNumber: 11
             })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 795,
-          columnNumber: 11
+          lineNumber: 668,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 794,
-      columnNumber: 9
-    })], {
-      externals: [pydanticModule]
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 667,
+      columnNumber: 7
+    })).toRenderTo(`
         from pydantic import BaseModel
         from pydantic import model_validator
 
@@ -1130,7 +1083,6 @@ describe("Pydantic ecosystem emitters", () => {
                 return data
 
 
-      
       `);
   });
 });