@alloy-js/python 0.3.0-dev.4 → 0.3.0-dev.6

package/dist/test/sourcefiles.test.js

@@ -3,7 +3,8 @@ import { Prose } from "@alloy-js/core";
 import { d } from "@alloy-js/core/testing";
 import { expect, it } from "vitest";
 import * as py from "../src/index.js";
-import { assertFileContents, toSourceText, toSourceTextMultiple } from "./utils.js";
+import { dataclassesModule } from "../src/index.js";
+import { toSourceText } from "./utils.js";
 
 /**
  * toSourceText wraps the children in a SourceFile component
@@ -11,11 +12,10 @@ import { assertFileContents, toSourceText, toSourceTextMultiple } from "./utils.
  */
 it("renders an empty source file", () => {
   const result = toSourceText([]);
-  const expected = d`
+  expect(result).toRenderTo(d`
 
 
-  `;
-  expect(result).toRenderTo(expected);
+  `);
 });
 it("correct formatting of source file", () => {
   const result = toSourceText([_$createComponent(py.ClassDeclaration, {
@@ -162,7 +162,7 @@ it("correct formatting of source file", () => {
       })];
     }
   })]);
-  const expected = d`
+  expect(result).toRenderTo(d`
     class SomeClass:
         def some_method() -> str:
             x: int = 42
@@ -194,8 +194,7 @@ it("correct formatting of source file", () => {
 
     a.b["special-prop"]
 
-  `;
-  expect(result).toRenderTo(expected);
+  `);
 });
 it("renders module documentation correctly", () => {
   const moduleDoc = _$createComponent(py.ModuleDoc, {
@@ -217,7 +216,7 @@ it("renders module documentation correctly", () => {
     style: "google"
   });
   const content = _$createComponent(py.SourceFile, {
-    path: "utils.py",
+    path: "test.py",
     doc: moduleDoc,
     get children() {
       return [_$createComponent(py.VariableDeclaration, {
@@ -232,40 +231,34 @@ it("renders module documentation correctly", () => {
       })];
     }
   });
-  const res = toSourceTextMultiple([content]);
-  const file = res.contents.find(f => f.kind === "file" && f.path === "utils.py");
-  expect(file).toBeDefined();
-  assertFileContents(res, {
-    "utils.py": d`
-        """
-        This module provides utility functions for data processing. It includes
-        functions for validation, transformation, and analysis.
+  expect(toSourceText(content)).toRenderTo(d`
+    """
+    This module provides utility functions for data processing. It includes
+    functions for validation, transformation, and analysis.
 
-        Attributes:
-            DEFAULT_TIMEOUT (int): Default timeout value in seconds.
+    Attributes:
+        DEFAULT_TIMEOUT (int): Default timeout value in seconds.
 
-            MAX_RETRIES (int): Maximum number of retry attempts.
+        MAX_RETRIES (int): Maximum number of retry attempts.
 
-        Todo:
-            * Add caching functionality
-            * Improve error messages
-        """
+    Todo:
+        * Add caching functionality
+        * Improve error messages
+    """
 
+    default_timeout = 30
 
-        default_timeout = 30
+    max_retries = 3
 
-        max_retries = 3
-
-        def process_data():
-            pass
+    def process_data():
+        pass
 
 
-        `
-  });
+  `);
 });
 it("renders source file without documentation correctly", () => {
   const content = _$createComponent(py.SourceFile, {
-    path: "simple.py",
+    path: "test.py",
     get children() {
       return _$createComponent(py.FunctionDeclaration, {
         name: "hello_world",
@@ -273,16 +266,905 @@ it("renders source file without documentation correctly", () => {
       });
     }
   });
-  const res = toSourceTextMultiple([content]);
-  const file = res.contents.find(f => f.kind === "file" && f.path === "simple.py");
-  expect(file).toBeDefined();
-  assertFileContents(res, {
-    "simple.py": d`
-        def hello_world():
-            print("Hello, World!")
+  expect(toSourceText(content)).toRenderTo(d`
+    def hello_world():
+        print("Hello, World!")
+
+
+  `);
+});
+it("nothing before top-level definition", () => {
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    get children() {
+      return _$createComponent(py.FunctionDeclaration, {
+        name: "hello",
+        children: "pass"
+      });
+    }
+  });
+  expect(toSourceText(content)).toRenderTo(d`
+    def hello():
+        pass
+
+
+  `);
+});
+it("nothing before non-definition", () => {
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    get children() {
+      return _$createComponent(py.VariableDeclaration, {
+        name: "x",
+        initializer: 42
+      });
+    }
+  });
+  expect(toSourceText(content)).toRenderTo("x = 42");
+});
+it("only doc before definition", () => {
+  const moduleDoc = _$createComponent(py.ModuleDoc, {
+    get description() {
+      return [_$createComponent(Prose, {
+        children: "Module description."
+      })];
+    }
+  });
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    doc: moduleDoc,
+    get children() {
+      return _$createComponent(py.FunctionDeclaration, {
+        name: "hello",
+        children: "pass"
+      });
+    }
+  });
+  expect(toSourceText(content)).toRenderTo(d`
+    """
+    Module description.
+    """
+
+
+    def hello():
+        pass
+
+
+  `);
+});
+it("only doc before non-definition", () => {
+  const moduleDoc = _$createComponent(py.ModuleDoc, {
+    get description() {
+      return [_$createComponent(Prose, {
+        children: "Module description."
+      })];
+    }
+  });
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    doc: moduleDoc,
+    get children() {
+      return _$createComponent(py.VariableDeclaration, {
+        name: "x",
+        initializer: 42
+      });
+    }
+  });
+  expect(toSourceText(content)).toRenderTo(d`
+    """
+    Module description.
+    """
+
+    x = 42`);
+});
+it("only header before definition", () => {
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    header: "#!/usr/bin/env python3",
+    get children() {
+      return _$createComponent(py.FunctionDeclaration, {
+        name: "hello",
+        children: "pass"
+      });
+    }
+  });
+
+  // 2 blank lines before definition (PEP 8)
+  expect(toSourceText(content)).toRenderTo(d`
+    #!/usr/bin/env python3
+
+
+    def hello():
+        pass
+
+
+  `);
+});
+it("only header before non-definition", () => {
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    header: "#!/usr/bin/env python3",
+    get children() {
+      return _$createComponent(py.VariableDeclaration, {
+        name: "x",
+        initializer: 42
+      });
+    }
+  });
+
+  // 1 blank line for non-definition
+  expect(toSourceText(content)).toRenderTo(d`
+    #!/usr/bin/env python3
+
+    x = 42`);
+});
+it("only futureImports before definition", () => {
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    get futureImports() {
+      return _$createComponent(py.FutureStatement, {
+        feature: "annotations"
+      });
+    },
+    get children() {
+      return _$createComponent(py.FunctionDeclaration, {
+        name: "hello",
+        children: "pass"
+      });
+    }
+  });
+  expect(toSourceText(content)).toRenderTo(d`
+    from __future__ import annotations
+
+
+    def hello():
+        pass
+
+
+  `);
+});
+it("only futureImports before non-definition", () => {
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    get futureImports() {
+      return _$createComponent(py.FutureStatement, {
+        feature: "annotations"
+      });
+    },
+    get children() {
+      return _$createComponent(py.VariableDeclaration, {
+        name: "x",
+        initializer: 42
+      });
+    }
+  });
+  expect(toSourceText(content)).toRenderTo(d`
+    from __future__ import annotations
+
+    x = 42`);
+});
+it("only imports before definition", () => {
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    get children() {
+      return _$createComponent(py.DataclassDeclaration, {
+        name: "User",
+        get children() {
+          return _$createComponent(py.VariableDeclaration, {
+            name: "name",
+            type: "str"
+          });
+        }
+      });
+    }
+  });
+  expect(toSourceText(content, {
+    externals: [dataclassesModule]
+  })).toRenderTo(d`
+    from dataclasses import dataclass
+
+
+    @dataclass
+    class User:
+        name: str = None
+
+
+  `);
+});
+it("only imports before non-definition", () => {
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    get children() {
+      return _$createComponent(py.VariableDeclaration, {
+        name: "x",
+        get initializer() {
+          return _$createComponent(py.Reference, {
+            get refkey() {
+              return dataclassesModule["."].dataclass;
+            }
+          });
+        }
+      });
+    }
+  });
+  expect(toSourceText(content, {
+    externals: [dataclassesModule]
+  })).toRenderTo(d`
+    from dataclasses import dataclass
+
+    x = dataclass`);
+});
+it("doc + futureImports before definition", () => {
+  const moduleDoc = _$createComponent(py.ModuleDoc, {
+    get description() {
+      return [_$createComponent(Prose, {
+        children: "Module description."
+      })];
+    }
+  });
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    doc: moduleDoc,
+    get futureImports() {
+      return _$createComponent(py.FutureStatement, {
+        feature: "annotations"
+      });
+    },
+    get children() {
+      return _$createComponent(py.FunctionDeclaration, {
+        name: "hello",
+        children: "pass"
+      });
+    }
+  });
+  expect(toSourceText(content)).toRenderTo(d`
+    """
+    Module description.
+    """
+
+    from __future__ import annotations
+
+
+    def hello():
+        pass
+
+
+  `);
+});
+it("doc + futureImports before non-definition", () => {
+  const moduleDoc = _$createComponent(py.ModuleDoc, {
+    get description() {
+      return [_$createComponent(Prose, {
+        children: "Module description."
+      })];
+    }
+  });
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    doc: moduleDoc,
+    get futureImports() {
+      return _$createComponent(py.FutureStatement, {
+        feature: "annotations"
+      });
+    },
+    get children() {
+      return _$createComponent(py.VariableDeclaration, {
+        name: "x",
+        initializer: 42
+      });
+    }
+  });
+  expect(toSourceText(content)).toRenderTo(d`
+    """
+    Module description.
+    """
+
+    from __future__ import annotations
+
+    x = 42`);
+});
+it("doc + imports before definition", () => {
+  const moduleDoc = _$createComponent(py.ModuleDoc, {
+    get description() {
+      return [_$createComponent(Prose, {
+        children: "Module description."
+      })];
+    }
+  });
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    doc: moduleDoc,
+    get children() {
+      return _$createComponent(py.DataclassDeclaration, {
+        name: "User",
+        get children() {
+          return _$createComponent(py.VariableDeclaration, {
+            name: "name",
+            type: "str"
+          });
+        }
+      });
+    }
+  });
+  expect(toSourceText(content, {
+    externals: [dataclassesModule]
+  })).toRenderTo(d`
+    """
+    Module description.
+    """
+
+    from dataclasses import dataclass
+
+
+    @dataclass
+    class User:
+        name: str = None
+
+
+  `);
+});
+it("doc + imports before non-definition", () => {
+  const moduleDoc = _$createComponent(py.ModuleDoc, {
+    get description() {
+      return [_$createComponent(Prose, {
+        children: "Module description."
+      })];
+    }
+  });
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    doc: moduleDoc,
+    get children() {
+      return _$createComponent(py.VariableDeclaration, {
+        name: "x",
+        get initializer() {
+          return _$createComponent(py.Reference, {
+            get refkey() {
+              return dataclassesModule["."].dataclass;
+            }
+          });
+        }
+      });
+    }
+  });
+  expect(toSourceText(content, {
+    externals: [dataclassesModule]
+  })).toRenderTo(d`
+    """
+    Module description.
+    """
+
+    from dataclasses import dataclass
+
+    x = dataclass`);
+});
+it("futureImports + imports before definition", () => {
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    get futureImports() {
+      return _$createComponent(py.FutureStatement, {
+        feature: "annotations"
+      });
+    },
+    get children() {
+      return _$createComponent(py.DataclassDeclaration, {
+        name: "User",
+        get children() {
+          return _$createComponent(py.VariableDeclaration, {
+            name: "name",
+            type: "str"
+          });
+        }
+      });
+    }
+  });
+  expect(toSourceText(content, {
+    externals: [dataclassesModule]
+  })).toRenderTo(d`
+    from __future__ import annotations
+
+    from dataclasses import dataclass
+
+
+    @dataclass
+    class User:
+        name: str = None
+
+
+  `);
+});
+it("futureImports + imports before non-definition", () => {
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    get futureImports() {
+      return _$createComponent(py.FutureStatement, {
+        feature: "annotations"
+      });
+    },
+    get children() {
+      return _$createComponent(py.VariableDeclaration, {
+        name: "x",
+        get initializer() {
+          return _$createComponent(py.Reference, {
+            get refkey() {
+              return dataclassesModule["."].dataclass;
+            }
+          });
+        }
+      });
+    }
+  });
+  expect(toSourceText(content, {
+    externals: [dataclassesModule]
+  })).toRenderTo(d`
+    from __future__ import annotations
+
+    from dataclasses import dataclass
+
+    x = dataclass`);
+});
+it("doc + futureImports + imports before definition", () => {
+  const moduleDoc = _$createComponent(py.ModuleDoc, {
+    get description() {
+      return [_$createComponent(Prose, {
+        children: "Module description."
+      })];
+    }
+  });
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    doc: moduleDoc,
+    get futureImports() {
+      return _$createComponent(py.FutureStatement, {
+        feature: "annotations"
+      });
+    },
+    get children() {
+      return _$createComponent(py.DataclassDeclaration, {
+        name: "User",
+        get children() {
+          return _$createComponent(py.VariableDeclaration, {
+            name: "name",
+            type: "str"
+          });
+        }
+      });
+    }
+  });
+  expect(toSourceText(content, {
+    externals: [dataclassesModule]
+  })).toRenderTo(d`
+    """
+    Module description.
+    """
+
+    from __future__ import annotations
+
+    from dataclasses import dataclass
+
+
+    @dataclass
+    class User:
+        name: str = None
+
+
+  `);
+});
+it("doc + futureImports + imports before non-definition", () => {
+  const moduleDoc = _$createComponent(py.ModuleDoc, {
+    get description() {
+      return [_$createComponent(Prose, {
+        children: "Module description."
+      })];
+    }
+  });
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    doc: moduleDoc,
+    get futureImports() {
+      return _$createComponent(py.FutureStatement, {
+        feature: "annotations"
+      });
+    },
+    get children() {
+      return _$createComponent(py.VariableDeclaration, {
+        name: "x",
+        get initializer() {
+          return _$createComponent(py.Reference, {
+            get refkey() {
+              return dataclassesModule["."].dataclass;
+            }
+          });
+        }
+      });
+    }
+  });
+  expect(toSourceText(content, {
+    externals: [dataclassesModule]
+  })).toRenderTo(d`
+    """
+    Module description.
+    """
+
+    from __future__ import annotations
+
+    from dataclasses import dataclass
+
+    x = dataclass`);
+});
+it("only doc in file (no children)", () => {
+  const moduleDoc = _$createComponent(py.ModuleDoc, {
+    get description() {
+      return [_$createComponent(Prose, {
+        children: "Module description."
+      })];
+    }
+  });
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    doc: moduleDoc
+  });
+  expect(toSourceText(content)).toRenderTo(d`
+    """
+    Module description.
+    """
+
+
+  `);
+});
+it("only header in file (no children)", () => {
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    header: "#!/usr/bin/env python3"
+  });
+  expect(toSourceText(content)).toRenderTo(d`
+    #!/usr/bin/env python3
+
+
+  `);
+});
+it("only futureImports in file (no children)", () => {
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    get futureImports() {
+      return _$createComponent(py.FutureStatement, {
+        feature: "annotations"
+      });
+    }
+  });
+  expect(toSourceText(content)).toRenderTo(d`
+    from __future__ import annotations
+
+  `);
+});
+it("doc + futureImports in file (no children)", () => {
+  const moduleDoc = _$createComponent(py.ModuleDoc, {
+    get description() {
+      return [_$createComponent(Prose, {
+        children: "Module description."
+      })];
+    }
+  });
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    doc: moduleDoc,
+    get futureImports() {
+      return _$createComponent(py.FutureStatement, {
+        feature: "annotations"
+      });
+    }
+  });
+  expect(toSourceText(content)).toRenderTo(d`
+    """
+    Module description.
+    """
+
+    from __future__ import annotations
+
+  `);
+});
+
+// headerComment tests
+it("only headerComment before definition", () => {
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    headerComment: "Copyright 2024 My Company",
+    get children() {
+      return _$createComponent(py.FunctionDeclaration, {
+        name: "hello",
+        children: "pass"
+      });
+    }
+  });
+
+  // 2 blank lines before definition (PEP 8)
+  expect(toSourceText(content)).toRenderTo(d`
+    # Copyright 2024 My Company
+
+
+    def hello():
+        pass
+
+
+  `);
+});
+it("only headerComment before non-definition", () => {
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    headerComment: "Copyright 2024 My Company",
+    get children() {
+      return _$createComponent(py.VariableDeclaration, {
+        name: "x",
+        initializer: 42
+      });
+    }
+  });
+  expect(toSourceText(content)).toRenderTo(d`
+    # Copyright 2024 My Company
+
+    x = 42`);
+});
+it("headerComment + doc before definition", () => {
+  const moduleDoc = _$createComponent(py.ModuleDoc, {
+    get description() {
+      return [_$createComponent(Prose, {
+        children: "Module description."
+      })];
+    }
+  });
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    headerComment: "Copyright 2024 My Company",
+    doc: moduleDoc,
+    get children() {
+      return _$createComponent(py.FunctionDeclaration, {
+        name: "hello",
+        children: "pass"
+      });
+    }
+  });
 
+  // headerComment and doc adjacent, then PEP 8 spacing before definition
+  expect(toSourceText(content)).toRenderTo(d`
+    # Copyright 2024 My Company
+    """
+    Module description.
+    """
 
-        `
+
+    def hello():
+        pass
+
+
+  `);
+});
+it("headerComment + futureImports before definition", () => {
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    headerComment: "Copyright 2024 My Company",
+    get futureImports() {
+      return _$createComponent(py.FutureStatement, {
+        feature: "annotations"
+      });
+    },
+    get children() {
+      return _$createComponent(py.FunctionDeclaration, {
+        name: "hello",
+        children: "pass"
+      });
+    }
+  });
+
+  // headerComment and futureImports adjacent, then PEP 8 spacing
+  expect(toSourceText(content)).toRenderTo(d`
+    # Copyright 2024 My Company
+
+    from __future__ import annotations
+
+
+    def hello():
+        pass
+
+
+  `);
+});
+it("headerComment + imports before definition", () => {
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    headerComment: "Copyright 2024 My Company",
+    get children() {
+      return _$createComponent(py.DataclassDeclaration, {
+        name: "User",
+        get children() {
+          return _$createComponent(py.VariableDeclaration, {
+            name: "name",
+            type: "str"
+          });
+        }
+      });
+    }
+  });
+  expect(toSourceText(content, {
+    externals: [dataclassesModule]
+  })).toRenderTo(d`
+    # Copyright 2024 My Company
+
+    from dataclasses import dataclass
+
+
+    @dataclass
+    class User:
+        name: str = None
+
+
+  `);
+});
+it("headerComment + doc + futureImports before definition", () => {
+  const moduleDoc = _$createComponent(py.ModuleDoc, {
+    get description() {
+      return [_$createComponent(Prose, {
+        children: "Module description."
+      })];
+    }
   });
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    headerComment: "Copyright 2024 My Company",
+    doc: moduleDoc,
+    get futureImports() {
+      return _$createComponent(py.FutureStatement, {
+        feature: "annotations"
+      });
+    },
+    get children() {
+      return _$createComponent(py.FunctionDeclaration, {
+        name: "hello",
+        children: "pass"
+      });
+    }
+  });
+  expect(toSourceText(content)).toRenderTo(d`
+    # Copyright 2024 My Company
+    """
+    Module description.
+    """
+
+    from __future__ import annotations
+
+
+    def hello():
+        pass
+
+
+  `);
+});
+it("headerComment + doc + futureImports + imports before definition", () => {
+  const moduleDoc = _$createComponent(py.ModuleDoc, {
+    get description() {
+      return [_$createComponent(Prose, {
+        children: "Module description."
+      })];
+    }
+  });
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    headerComment: "Copyright 2024 My Company",
+    doc: moduleDoc,
+    get futureImports() {
+      return _$createComponent(py.FutureStatement, {
+        feature: "annotations"
+      });
+    },
+    get children() {
+      return _$createComponent(py.DataclassDeclaration, {
+        name: "User",
+        get children() {
+          return _$createComponent(py.VariableDeclaration, {
+            name: "name",
+            type: "str"
+          });
+        }
+      });
+    }
+  });
+  expect(toSourceText(content, {
+    externals: [dataclassesModule]
+  })).toRenderTo(d`
+    # Copyright 2024 My Company
+    """
+    Module description.
+    """
+
+    from __future__ import annotations
+
+    from dataclasses import dataclass
+
+
+    @dataclass
+    class User:
+        name: str = None
+
+
+  `);
+});
+it("only headerComment in file (no children)", () => {
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    headerComment: "Copyright 2024 My Company"
+  });
+  expect(toSourceText(content)).toRenderTo(d`
+    # Copyright 2024 My Company
+
+  `);
+});
+it("header + headerComment before definition", () => {
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    header: "#!/usr/bin/env python3",
+    headerComment: "Copyright 2024 My Company",
+    get children() {
+      return _$createComponent(py.FunctionDeclaration, {
+        name: "hello",
+        children: "pass"
+      });
+    }
+  });
+
+  // 2 blank lines before definition (PEP 8)
+  expect(toSourceText(content)).toRenderTo(d`
+    #!/usr/bin/env python3
+    # Copyright 2024 My Company
+
+
+    def hello():
+        pass
+
+
+  `);
+});
+it("header + headerComment before class definition", () => {
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    header: "#!/usr/bin/env python3",
+    headerComment: "Copyright 2024 My Company",
+    get children() {
+      return _$createComponent(py.ClassDeclaration, {
+        name: "MyClass",
+        children: "pass"
+      });
+    }
+  });
+
+  // 2 blank lines before definition (PEP 8)
+  expect(toSourceText(content)).toRenderTo(d`
+    #!/usr/bin/env python3
+    # Copyright 2024 My Company
+
+
+    class MyClass:
+        pass
+
+
+  `);
+});
+it("header + headerComment before non-definition", () => {
+  const content = _$createComponent(py.SourceFile, {
+    path: "test.py",
+    header: "#!/usr/bin/env python3",
+    headerComment: "Copyright 2024 My Company",
+    get children() {
+      return _$createComponent(py.VariableDeclaration, {
+        name: "x",
+        initializer: 42
+      });
+    }
+  });
+
+  // 1 blank line for non-definition
+  expect(toSourceText(content)).toRenderTo(d`
+    #!/usr/bin/env python3
+    # Copyright 2024 My Company
+
+    x = 42`);
 });
 //# sourceMappingURL=sourcefiles.test.js.map