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

package/dist/dev/test/type-checking-imports.test.js

@@ -1,103 +1,118 @@
 import { createComponent as _$createComponent, memo as _$memo } from "@alloy-js/core/jsx-runtime";
 import { refkey } from "@alloy-js/core";
-import { describe, it } from "vitest";
+import { describe, expect, it } from "vitest";
 import { createModule } from "../src/create-module.js";
 import * as py from "../src/index.js";
-import { assertFileContents, toSourceTextMultiple } from "./utils.js";
+import { TestOutputDirectory } from "./utils.js";
 describe("TYPE_CHECKING imports", () => {
   it("imports type-only references inside TYPE_CHECKING block", () => {
     const userClassRef = refkey();
-    const result = toSourceTextMultiple([_$createComponent(py.SourceFile, {
-      path: "models.py",
+    expect(_$createComponent(TestOutputDirectory, {
       get children() {
-        return _$createComponent(py.ClassDeclaration, {
-          name: "User",
-          refkey: userClassRef
+        return [_$createComponent(py.SourceFile, {
+          path: "models.py",
+          get children() {
+            return _$createComponent(py.ClassDeclaration, {
+              name: "User",
+              refkey: userClassRef
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 14,
+              columnNumber: 11
+            });
+          }
         }, {
           fileName: import.meta.url,
           lineNumber: 13,
           columnNumber: 9
-        });
-      }
-    }, {
-      fileName: import.meta.url,
-      lineNumber: 12,
-      columnNumber: 7
-    }), _$createComponent(py.SourceFile, {
-      path: "service.py",
-      get children() {
-        return _$createComponent(py.FunctionDeclaration, {
-          name: "process_user",
-          parameters: [{
-            name: "user",
-            type: userClassRef
-          }],
-          returnType: "None",
-          children: "pass"
+        }), _$createComponent(py.SourceFile, {
+          path: "service.py",
+          get children() {
+            return _$createComponent(py.FunctionDeclaration, {
+              name: "process_user",
+              parameters: [{
+                name: "user",
+                type: userClassRef
+              }],
+              returnType: "None",
+              children: "pass"
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 17,
+              columnNumber: 11
+            });
+          }
         }, {
           fileName: import.meta.url,
           lineNumber: 16,
           columnNumber: 9
-        });
+        })];
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 15,
+      lineNumber: 12,
       columnNumber: 7
-    })]);
-    assertFileContents(result, {
+    })).toRenderTo({
       "models.py": `
-        class User:
-            pass
+          class User:
+              pass
 
-        `,
+          `,
       "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 process_user(user: User) -> None:
-            pass
+          def process_user(user: User) -> None:
+              pass
 
-        `
+          `
     });
   });
   it("imports value references outside TYPE_CHECKING block", () => {
     const userClassRef = refkey();
-    const result = toSourceTextMultiple([_$createComponent(py.SourceFile, {
-      path: "models.py",
+    expect(_$createComponent(TestOutputDirectory, {
       get children() {
-        return _$createComponent(py.ClassDeclaration, {
-          name: "User",
-          refkey: userClassRef
+        return [_$createComponent(py.SourceFile, {
+          path: "models.py",
+          get children() {
+            return _$createComponent(py.ClassDeclaration, {
+              name: "User",
+              refkey: userClassRef
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 52,
+              columnNumber: 11
+            });
+          }
         }, {
           fileName: import.meta.url,
           lineNumber: 51,
           columnNumber: 9
-        });
-      }
-    }, {
-      fileName: import.meta.url,
-      lineNumber: 50,
-      columnNumber: 7
-    }), _$createComponent(py.SourceFile, {
-      path: "service.py",
-      get children() {
-        return _$createComponent(py.FunctionDeclaration, {
-          name: "create_user",
-          returnType: "None",
+        }), _$createComponent(py.SourceFile, {
+          path: "service.py",
           get children() {
-            return _$createComponent(py.VariableDeclaration, {
-              name: "user",
-              get initializer() {
-                return _$createComponent(py.ClassInstantiation, {
-                  target: userClassRef
+            return _$createComponent(py.FunctionDeclaration, {
+              name: "create_user",
+              returnType: "None",
+              get children() {
+                return _$createComponent(py.VariableDeclaration, {
+                  name: "user",
+                  get initializer() {
+                    return _$createComponent(py.ClassInstantiation, {
+                      target: userClassRef
+                    }, {
+                      fileName: import.meta.url,
+                      lineNumber: 58,
+                      columnNumber: 28
+                    });
+                  }
                 }, {
                   fileName: import.meta.url,
-                  lineNumber: 57,
-                  columnNumber: 26
+                  lineNumber: 56,
+                  columnNumber: 13
                 });
               }
             }, {
@@ -110,80 +125,87 @@ describe("TYPE_CHECKING imports", () => {
           fileName: import.meta.url,
           lineNumber: 54,
           columnNumber: 9
-        });
+        })];
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 53,
+      lineNumber: 50,
       columnNumber: 7
-    })]);
-    assertFileContents(result, {
+    })).toRenderTo({
       "models.py": `
-        class User:
-            pass
+          class User:
+              pass
 
-        `,
+          `,
       "service.py": `
-        from models import User
+          from models import User
 
 
-        def create_user() -> None:
-            user = User()
+          def create_user() -> None:
+              user = User()
 
-        `
+          `
     });
   });
   it("upgrades type-only import to regular import when also used as value", () => {
     const userClassRef = refkey();
-    const result = toSourceTextMultiple([_$createComponent(py.SourceFile, {
-      path: "models.py",
+    expect(_$createComponent(TestOutputDirectory, {
       get children() {
-        return _$createComponent(py.ClassDeclaration, {
-          name: "User",
-          refkey: userClassRef
+        return [_$createComponent(py.SourceFile, {
+          path: "models.py",
+          get children() {
+            return _$createComponent(py.ClassDeclaration, {
+              name: "User",
+              refkey: userClassRef
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 86,
+              columnNumber: 11
+            });
+          }
         }, {
           fileName: import.meta.url,
           lineNumber: 85,
           columnNumber: 9
-        });
-      }
-    }, {
-      fileName: import.meta.url,
-      lineNumber: 84,
-      columnNumber: 7
-    }), _$createComponent(py.SourceFile, {
-      path: "service.py",
-      get children() {
-        return _$createComponent(py.FunctionDeclaration, {
-          name: "create_user",
-          parameters: [{
-            name: "existing",
-            type: userClassRef
-          }],
-          returnType: userClassRef,
+        }), _$createComponent(py.SourceFile, {
+          path: "service.py",
           get children() {
-            return _$createComponent(py.StatementList, {
+            return _$createComponent(py.FunctionDeclaration, {
+              name: "create_user",
+              parameters: [{
+                name: "existing",
+                type: userClassRef
+              }],
+              returnType: userClassRef,
               get children() {
-                return [_$createComponent(py.VariableDeclaration, {
-                  name: "user",
-                  get initializer() {
-                    return _$createComponent(py.ClassInstantiation, {
-                      target: userClassRef
+                return _$createComponent(py.StatementList, {
+                  get children() {
+                    return [_$createComponent(py.VariableDeclaration, {
+                      name: "user",
+                      get initializer() {
+                        return _$createComponent(py.ClassInstantiation, {
+                          target: userClassRef
+                        }, {
+                          fileName: import.meta.url,
+                          lineNumber: 97,
+                          columnNumber: 30
+                        });
+                      }
                     }, {
                       fileName: import.meta.url,
-                      lineNumber: 96,
-                      columnNumber: 28
-                    });
+                      lineNumber: 95,
+                      columnNumber: 15
+                    }), ["return user"]];
                   }
                 }, {
                   fileName: import.meta.url,
                   lineNumber: 94,
                   columnNumber: 13
-                }), ["return user"]];
+                });
               }
             }, {
               fileName: import.meta.url,
-              lineNumber: 93,
+              lineNumber: 89,
               columnNumber: 11
             });
           }
@@ -191,347 +213,370 @@ describe("TYPE_CHECKING imports", () => {
           fileName: import.meta.url,
           lineNumber: 88,
           columnNumber: 9
-        });
+        })];
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 87,
+      lineNumber: 84,
       columnNumber: 7
-    })]);
-
-    // Since User is used both as a type (parameter type, return type) and
-    // as a value (ClassInstantiation), it should be a regular import
-    assertFileContents(result, {
+    })).toRenderTo({
       "models.py": `
-        class User:
-            pass
+          class User:
+              pass
 
-        `,
+          `,
       "service.py": `
-        from models import User
+          from models import User
 
 
-        def create_user(existing: User) -> User:
-            user = User()
-            return user
+          def create_user(existing: User) -> User:
+              user = User()
+              return user
 
-        `
+          `
     });
   });
   it("handles mixed type-only and regular imports from same module", () => {
     const userClassRef = refkey();
     const helperFuncRef = refkey();
-    const result = toSourceTextMultiple([_$createComponent(py.SourceFile, {
-      path: "models.py",
+    expect(_$createComponent(TestOutputDirectory, {
       get children() {
-        return [_$createComponent(py.ClassDeclaration, {
-          name: "User",
-          refkey: userClassRef
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 130,
-          columnNumber: 9
-        }), _$createComponent(py.FunctionDeclaration, {
-          name: "helper",
-          refkey: helperFuncRef,
-          children: "pass"
+        return [_$createComponent(py.SourceFile, {
+          path: "models.py",
+          get children() {
+            return [_$createComponent(py.ClassDeclaration, {
+              name: "User",
+              refkey: userClassRef
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 129,
+              columnNumber: 11
+            }), _$createComponent(py.FunctionDeclaration, {
+              name: "helper",
+              refkey: helperFuncRef,
+              children: "pass"
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 130,
+              columnNumber: 11
+            })];
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 131,
+          lineNumber: 128,
           columnNumber: 9
-        })];
-      }
-    }, {
-      fileName: import.meta.url,
-      lineNumber: 129,
-      columnNumber: 7
-    }), _$createComponent(py.SourceFile, {
-      path: "service.py",
-      get children() {
-        return _$createComponent(py.FunctionDeclaration, {
-          name: "process",
-          parameters: [{
-            name: "user",
-            type: userClassRef
-          }],
-          returnType: "None",
+        }), _$createComponent(py.SourceFile, {
+          path: "service.py",
           get children() {
-            return _$createComponent(py.FunctionCallExpression, {
-              target: helperFuncRef
+            return _$createComponent(py.FunctionDeclaration, {
+              name: "process",
+              parameters: [{
+                name: "user",
+                type: userClassRef
+              }],
+              returnType: "None",
+              get children() {
+                return _$createComponent(py.FunctionCallExpression, {
+                  target: helperFuncRef
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 140,
+                  columnNumber: 13
+                });
+              }
             }, {
               fileName: import.meta.url,
-              lineNumber: 141,
+              lineNumber: 135,
               columnNumber: 11
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 136,
+          lineNumber: 134,
           columnNumber: 9
-        });
+        })];
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 135,
+      lineNumber: 127,
       columnNumber: 7
-    })]);
-
-    // helper is used as a value (function call), User is only used as type
-    assertFileContents(result, {
+    })).toRenderTo({
       "models.py": `
-        class User:
-            pass
+          class User:
+              pass
 
 
-        def helper():
-            pass
+          def helper():
+              pass
 
-        `,
+          `,
       "service.py": `
-        from models import helper
-        from typing import TYPE_CHECKING
+          from models import helper
+          from typing import TYPE_CHECKING
 
-        if TYPE_CHECKING:
-            from models import User
+          if TYPE_CHECKING:
+              from models import User
 
 
-        def process(user: User) -> None:
-            helper()
+          def process(user: User) -> None:
+              helper()
 
-        `
+          `
     });
   });
   it("handles return type as type-only import", () => {
     const resultTypeRef = refkey();
-    const result = toSourceTextMultiple([_$createComponent(py.SourceFile, {
-      path: "types.py",
+    expect(_$createComponent(TestOutputDirectory, {
       get children() {
-        return _$createComponent(py.ClassDeclaration, {
-          name: "Result",
-          refkey: resultTypeRef
+        return [_$createComponent(py.SourceFile, {
+          path: "types.py",
+          get children() {
+            return _$createComponent(py.ClassDeclaration, {
+              name: "Result",
+              refkey: resultTypeRef
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 175,
+              columnNumber: 11
+            });
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 177,
+          lineNumber: 174,
           columnNumber: 9
-        });
-      }
-    }, {
-      fileName: import.meta.url,
-      lineNumber: 176,
-      columnNumber: 7
-    }), _$createComponent(py.SourceFile, {
-      path: "main.py",
-      get children() {
-        return _$createComponent(py.FunctionDeclaration, {
-          name: "get_result",
-          returnType: resultTypeRef,
-          children: "pass"
+        }), _$createComponent(py.SourceFile, {
+          path: "main.py",
+          get children() {
+            return _$createComponent(py.FunctionDeclaration, {
+              name: "get_result",
+              returnType: resultTypeRef,
+              children: "pass"
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 178,
+              columnNumber: 11
+            });
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 180,
+          lineNumber: 177,
           columnNumber: 9
-        });
+        })];
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 179,
+      lineNumber: 173,
       columnNumber: 7
-    })]);
-    assertFileContents(result, {
+    })).toRenderTo({
       "types.py": `
-        class Result:
-            pass
+          class Result:
+              pass
 
-        `,
+          `,
       "main.py": `
-        from typing import TYPE_CHECKING
+          from typing import TYPE_CHECKING
 
-        if TYPE_CHECKING:
-            from types import Result
+          if TYPE_CHECKING:
+              from types import Result
 
 
-        def get_result() -> Result:
-            pass
+          def get_result() -> Result:
+              pass
 
-        `
+          `
     });
   });
   it("handles variable type annotation as type-only import", () => {
     const configTypeRef = refkey();
-    const result = toSourceTextMultiple([_$createComponent(py.SourceFile, {
-      path: "types.py",
+    expect(_$createComponent(TestOutputDirectory, {
       get children() {
-        return _$createComponent(py.ClassDeclaration, {
-          name: "Config",
-          refkey: configTypeRef
+        return [_$createComponent(py.SourceFile, {
+          path: "types.py",
+          get children() {
+            return _$createComponent(py.ClassDeclaration, {
+              name: "Config",
+              refkey: configTypeRef
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 209,
+              columnNumber: 11
+            });
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 211,
+          lineNumber: 208,
           columnNumber: 9
-        });
-      }
-    }, {
-      fileName: import.meta.url,
-      lineNumber: 210,
-      columnNumber: 7
-    }), _$createComponent(py.SourceFile, {
-      path: "main.py",
-      get children() {
-        return _$createComponent(py.VariableDeclaration, {
-          name: "config",
-          type: configTypeRef,
-          omitNone: true
+        }), _$createComponent(py.SourceFile, {
+          path: "main.py",
+          get children() {
+            return _$createComponent(py.VariableDeclaration, {
+              name: "config",
+              type: configTypeRef,
+              omitNone: true
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 212,
+              columnNumber: 11
+            });
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 214,
+          lineNumber: 211,
           columnNumber: 9
-        });
+        })];
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 213,
+      lineNumber: 207,
       columnNumber: 7
-    })]);
-    assertFileContents(result, {
+    })).toRenderTo({
       "types.py": `
-        class Config:
-            pass
+          class Config:
+              pass
 
-        `,
+          `,
       "main.py": `
-        from typing import TYPE_CHECKING
+          from typing import TYPE_CHECKING
 
-        if TYPE_CHECKING:
-            from types import Config
+          if TYPE_CHECKING:
+              from types import Config
 
-        config: Config
-        `
+          config: Config
+          `
     });
   });
   it("handles TypeReference component as type-only import", () => {
     const myTypeRef = refkey();
-    const result = toSourceTextMultiple([_$createComponent(py.SourceFile, {
-      path: "types.py",
+    expect(_$createComponent(TestOutputDirectory, {
       get children() {
-        return _$createComponent(py.ClassDeclaration, {
-          name: "MyType",
-          refkey: myTypeRef
+        return [_$createComponent(py.SourceFile, {
+          path: "types.py",
+          get children() {
+            return _$createComponent(py.ClassDeclaration, {
+              name: "MyType",
+              refkey: myTypeRef
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 238,
+              columnNumber: 11
+            });
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 240,
+          lineNumber: 237,
           columnNumber: 9
-        });
-      }
-    }, {
-      fileName: import.meta.url,
-      lineNumber: 239,
-      columnNumber: 7
-    }), _$createComponent(py.SourceFile, {
-      path: "main.py",
-      get children() {
-        return _$createComponent(py.VariableDeclaration, {
-          name: "value",
-          get type() {
-            return _$createComponent(py.TypeReference, {
-              refkey: myTypeRef
+        }), _$createComponent(py.SourceFile, {
+          path: "main.py",
+          get children() {
+            return _$createComponent(py.VariableDeclaration, {
+              name: "value",
+              get type() {
+                return _$createComponent(py.TypeReference, {
+                  refkey: myTypeRef
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 243,
+                  columnNumber: 19
+                });
+              },
+              omitNone: true
             }, {
               fileName: import.meta.url,
-              lineNumber: 245,
-              columnNumber: 17
+              lineNumber: 241,
+              columnNumber: 11
             });
-          },
-          omitNone: true
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 243,
+          lineNumber: 240,
           columnNumber: 9
-        });
+        })];
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 242,
+      lineNumber: 236,
       columnNumber: 7
-    })]);
-    assertFileContents(result, {
+    })).toRenderTo({
       "types.py": `
-        class MyType:
-            pass
+          class MyType:
+              pass
 
-        `,
+          `,
       "main.py": `
-        from typing import TYPE_CHECKING
+          from typing import TYPE_CHECKING
 
-        if TYPE_CHECKING:
-            from types import MyType
+          if TYPE_CHECKING:
+              from types import MyType
 
-        value: MyType
-        `
+          value: MyType
+          `
     });
   });
   it("handles class bases as regular import (runtime requirement)", () => {
     const baseClassRef = refkey();
-    const result = toSourceTextMultiple([_$createComponent(py.SourceFile, {
-      path: "base.py",
+    expect(_$createComponent(TestOutputDirectory, {
       get children() {
-        return _$createComponent(py.ClassDeclaration, {
-          name: "BaseClass",
-          refkey: baseClassRef
+        return [_$createComponent(py.SourceFile, {
+          path: "base.py",
+          get children() {
+            return _$createComponent(py.ClassDeclaration, {
+              name: "BaseClass",
+              refkey: baseClassRef
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 271,
+              columnNumber: 11
+            });
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 273,
+          lineNumber: 270,
           columnNumber: 9
-        });
-      }
-    }, {
-      fileName: import.meta.url,
-      lineNumber: 272,
-      columnNumber: 7
-    }), _$createComponent(py.SourceFile, {
-      path: "derived.py",
-      get children() {
-        return _$createComponent(py.ClassDeclaration, {
-          name: "DerivedClass",
-          bases: [baseClassRef]
+        }), _$createComponent(py.SourceFile, {
+          path: "derived.py",
+          get children() {
+            return _$createComponent(py.ClassDeclaration, {
+              name: "DerivedClass",
+              bases: [baseClassRef]
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 274,
+              columnNumber: 11
+            });
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 276,
+          lineNumber: 273,
           columnNumber: 9
-        });
+        })];
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 275,
+      lineNumber: 269,
       columnNumber: 7
-    })]);
-
-    // Class bases require runtime access, so they should NOT be
-    // inside a TYPE_CHECKING block
-    assertFileContents(result, {
+    })).toRenderTo({
       "base.py": `
-        class BaseClass:
-            pass
+          class BaseClass:
+              pass
 
-        `,
+          `,
       "derived.py": `
-        from base import BaseClass
+          from base import BaseClass
 
 
-        class DerivedClass(BaseClass):
-            pass
+          class DerivedClass(BaseClass):
+              pass
 
-        `
+          `
     });
   });
   it("renders regular imports before TYPE_CHECKING block", () => {
-    // Create a typing module with multiple exports
     const typingModule = createModule({
       name: "typing",
       descriptor: {
         ".": ["TYPE_CHECKING", "cast"]
       }
     });
-
-    // Create a third-party module
     const requestsModule = createModule({
       name: "requests",
       descriptor: {
@@ -539,78 +584,82 @@ describe("TYPE_CHECKING imports", () => {
       }
     });
     const userClassRef = refkey();
-    const result = toSourceTextMultiple([_$createComponent(py.SourceFile, {
-      path: "models.py",
+    expect(_$createComponent(TestOutputDirectory, {
+      externals: [typingModule, requestsModule],
       get children() {
-        return _$createComponent(py.ClassDeclaration, {
-          name: "User",
-          refkey: userClassRef
+        return [_$createComponent(py.SourceFile, {
+          path: "models.py",
+          get children() {
+            return _$createComponent(py.ClassDeclaration, {
+              name: "User",
+              refkey: userClassRef
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 314,
+              columnNumber: 11
+            });
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 321,
-          columnNumber: 11
-        });
-      }
-    }, {
-      fileName: import.meta.url,
-      lineNumber: 320,
-      columnNumber: 9
-    }), _$createComponent(py.SourceFile, {
-      path: "service.py",
-      get children() {
-        return _$createComponent(py.FunctionDeclaration, {
-          name: "get_users",
-          parameters: [{
-            name: "user",
-            type: userClassRef
-          }],
-          returnType: "str",
+          lineNumber: 313,
+          columnNumber: 9
+        }), _$createComponent(py.SourceFile, {
+          path: "service.py",
           get children() {
-            return _$createComponent(py.StatementList, {
+            return _$createComponent(py.FunctionDeclaration, {
+              name: "get_users",
+              parameters: [{
+                name: "user",
+                type: userClassRef
+              }],
+              returnType: "str",
               get children() {
-                return [["response = ", _$memo(() => requestsModule["."].get, false, "requestsModule[.].get"), "(\"https://example.com\")"], ["return ", _$memo(() => typingModule["."].cast, false, "typingModule[.].cast"), "(str, user)"]];
+                return _$createComponent(py.StatementList, {
+                  get children() {
+                    return [["response = ", _$memo(() => requestsModule["."].get, false, "requestsModule[.].get"), "(\"https://example.com\")"], ["return ", _$memo(() => typingModule["."].cast, false, "typingModule[.].cast"), "(str, user)"]];
+                  }
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 322,
+                  columnNumber: 13
+                });
               }
             }, {
               fileName: import.meta.url,
-              lineNumber: 329,
-              columnNumber: 13
+              lineNumber: 317,
+              columnNumber: 11
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 324,
-          columnNumber: 11
-        });
+          lineNumber: 316,
+          columnNumber: 9
+        })];
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 323,
-      columnNumber: 9
-    })], {
-      externals: [typingModule, requestsModule]
-    });
-
-    // Regular imports first (sorted alphabetically), then TYPE_CHECKING block
-    assertFileContents(result, {
+      lineNumber: 312,
+      columnNumber: 7
+    })).toRenderTo({
       "models.py": `
-        class User:
-            pass
+          class User:
+              pass
 
-        `,
+          `,
       "service.py": `
-        from requests import get
-        from typing import cast
-        from typing import TYPE_CHECKING
+          from requests import get
+          from typing import cast
+          from typing import TYPE_CHECKING
 
-        if TYPE_CHECKING:
-            from models import User
+          if TYPE_CHECKING:
+              from models import User
 
 
-        def get_users(user: User) -> str:
-            response = get("https://example.com")
-            return cast(str, user)
+          def get_users(user: User) -> str:
+              response = get("https://example.com")
+              return cast(str, user)
 
-        `
+          `
     });
   });
 });