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

package/dist/dev/test/pydocs.test.js

@@ -1,38 +1,40 @@
-import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
+import { createComponent as _$createComponent, memo as _$memo } from "@alloy-js/core/jsx-runtime";
 import { Prose } from "@alloy-js/core";
-import { d } from "@alloy-js/core/testing";
 import { describe, expect, it } from "vitest";
 import { enumModule } 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("PyDoc", () => {
   it("formats properly", () => {
-    const res = toSourceText([_$createComponent(py.PyDoc, {
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return [_$createComponent(Prose, {
-          children: "This is an example of a long docstring that will be broken in lines. We will also render another paragraph after this one."
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 17,
-          columnNumber: 11
-        }), _$createComponent(Prose, {
-          children: "This is another paragraph, and there's a line break before it."
+        return _$createComponent(py.PyDoc, {
+          get children() {
+            return [_$createComponent(Prose, {
+              children: "This is an example of a long docstring that will be broken in lines. We will also render another paragraph after this one."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 12,
+              columnNumber: 11
+            }), _$createComponent(Prose, {
+              children: "This is another paragraph, and there's a line break before it."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 16,
+              columnNumber: 11
+            })];
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 21,
-          columnNumber: 11
-        })];
+          lineNumber: 11,
+          columnNumber: 9
+        });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 16,
-      columnNumber: 9
-    })], {
-      printOptions: {
-        printWidth: 40
-      }
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 10,
+      columnNumber: 7
+    })).toRenderTo(`
         """
         This is an example of a long docstring
         that will be broken in lines. We will
@@ -43,38 +45,43 @@ describe("PyDoc", () => {
         line break before it.
         """
 
-        
-      `);
+      `, {
+      tabWidth: 4,
+      printWidth: 40
+    });
   });
 });
 describe("PyDocExample", () => {
   it("creates docstring with a code sample", () => {
-    const res = toSourceText([_$createComponent(py.PyDoc, {
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return [_$createComponent(Prose, {
-          children: "This is an example of a docstring with a code sample."
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 52,
-          columnNumber: 11
-        }), _$createComponent(py.PyDocExample, {
-          children: "print(\"Hello world!\")"
+        return _$createComponent(py.PyDoc, {
+          get children() {
+            return [_$createComponent(Prose, {
+              children: "This is an example of a docstring with a code sample."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 44,
+              columnNumber: 11
+            }), _$createComponent(py.PyDocExample, {
+              children: "print(\"Hello world!\")"
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 45,
+              columnNumber: 11
+            })];
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 53,
-          columnNumber: 11
-        })];
+          lineNumber: 43,
+          columnNumber: 9
+        });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 51,
-      columnNumber: 9
-    })], {
-      printOptions: {
-        printWidth: 40
-      }
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 42,
+      columnNumber: 7
+    })).toRenderTo(`
       """
       This is an example of a docstring with a
       code sample.
@@ -82,42 +89,47 @@ describe("PyDocExample", () => {
       >> print("Hello world!")
       """
 
-
-      `);
+      `, {
+      tabWidth: 4,
+      printWidth: 40
+    });
   });
   it("creates docstring with more than one code sample", () => {
-    const res = toSourceText([_$createComponent(py.PyDoc, {
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return [_$createComponent(Prose, {
-          children: "This is an example of a docstring with a code sample."
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 77,
-          columnNumber: 11
-        }), _$createComponent(py.PyDocExample, {
-          children: "print(\"Hello world!\")"
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 78,
-          columnNumber: 11
-        }), _$createComponent(py.PyDocExample, {
-          children: "print(\"Hello world again!\")"
+        return _$createComponent(py.PyDoc, {
+          get children() {
+            return [_$createComponent(Prose, {
+              children: "This is an example of a docstring with a code sample."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 66,
+              columnNumber: 11
+            }), _$createComponent(py.PyDocExample, {
+              children: "print(\"Hello world!\")"
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 67,
+              columnNumber: 11
+            }), _$createComponent(py.PyDocExample, {
+              children: "print(\"Hello world again!\")"
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 68,
+              columnNumber: 11
+            })];
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 79,
-          columnNumber: 11
-        })];
+          lineNumber: 65,
+          columnNumber: 9
+        });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 76,
-      columnNumber: 9
-    })], {
-      printOptions: {
-        printWidth: 40
-      }
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 64,
+      columnNumber: 7
+    })).toRenderTo(`
       """
       This is an example of a docstring with a
       code sample.
@@ -127,36 +139,41 @@ describe("PyDocExample", () => {
       >> print("Hello world again!")
       """
 
-
-      `);
+      `, {
+      tabWidth: 4,
+      printWidth: 40
+    });
   });
   it("creates docstring with a multiline code sample", () => {
-    const res = toSourceText([_$createComponent(py.PyDoc, {
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return [_$createComponent(Prose, {
-          children: "This is an example of a docstring with a code sample."
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 105,
-          columnNumber: 11
-        }), _$createComponent(py.PyDocExample, {
-          children: `print("Hello world!")\nx = "Hello"\nprint(x)`
+        return _$createComponent(py.PyDoc, {
+          get children() {
+            return [_$createComponent(Prose, {
+              children: "This is an example of a docstring with a code sample."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 91,
+              columnNumber: 11
+            }), _$createComponent(py.PyDocExample, {
+              children: `print("Hello world!")\nx = "Hello"\nprint(x)`
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 92,
+              columnNumber: 11
+            })];
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 106,
-          columnNumber: 11
-        })];
+          lineNumber: 90,
+          columnNumber: 9
+        });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 104,
-      columnNumber: 9
-    })], {
-      printOptions: {
-        printWidth: 40
-      }
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 89,
+      columnNumber: 7
+    })).toRenderTo(`
       """
       This is an example of a docstring with a
       code sample.
@@ -166,102 +183,133 @@ describe("PyDocExample", () => {
       >> print(x)
       """
 
-
-      `);
+      `, {
+      tabWidth: 4,
+      printWidth: 40
+    });
   });
 });
 describe("SimpleCommentBlock", () => {
   it("renders simple comment block", () => {
-    const res = toSourceText([_$createComponent(py.SimpleCommentBlock, {
-      children: "This is a simple comment block that spans multiple lines and should be split automatically."
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.SimpleCommentBlock, {
+          children: "This is a simple comment block that spans multiple lines and should be split automatically."
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 118,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 134,
+      lineNumber: 117,
       columnNumber: 7
-    })]);
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
           # This is a simple comment block that spans multiple lines and should be split
           # automatically.
-          
           `);
   });
   it("renders comment block with line breaks", () => {
-    const res = toSourceText([_$createComponent(py.SimpleCommentBlock, {
-      children: "First line of comment.\\nSecond line of comment."
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.SimpleCommentBlock, {
+          children: "First line of comment.\\nSecond line of comment."
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 134,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 150,
+      lineNumber: 133,
       columnNumber: 7
-    })]);
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
           # First line of comment.
           # Second line of comment.
-          
           `);
   });
 });
 describe("SimpleInlineComment", () => {
   it("renders inline comment", () => {
-    const res = toSourceText([["x = 42", _$createComponent(py.SimpleInlineComment, {
-      children: "This is an inline comment"
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return [["x = 42", _$createComponent(py.SimpleInlineComment, {
+          children: "This is an inline comment"
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 154,
+          columnNumber: 13
+        })]];
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 169,
-      columnNumber: 9
-    })]]);
-    expect(res).toRenderTo(d`
-          x = 42  # This is an inline comment
-          `);
+      lineNumber: 150,
+      columnNumber: 7
+    })).toRenderTo(`x = 42  # This is an inline comment`);
   });
   it("renders inline comment with complex text", () => {
-    const res = toSourceText([["result = calculate()", _$createComponent(py.SimpleInlineComment, {
-      children: "TODO: Add error handling here"
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return [["result = calculate()", _$createComponent(py.SimpleInlineComment, {
+          children: "TODO: Add error handling here"
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 169,
+          columnNumber: 13
+        })]];
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 185,
-      columnNumber: 9
-    })]]);
-    expect(res).toRenderTo(d`
-          result = calculate()  # TODO: Add error handling here
-          `);
+      lineNumber: 165,
+      columnNumber: 7
+    })).toRenderTo(`result = calculate()  # TODO: Add error handling here`);
   });
 });
 describe("New Documentation Components", () => {
   it("ModuleDoc renders correctly", () => {
-    const res = toSourceText([_$createComponent(py.ModuleDoc, {
-      get description() {
-        return [_$createComponent(Prose, {
-          children: "This module demonstrates documentation as specified by the Google Python Style Guide."
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 203,
-          columnNumber: 11
-        })];
-      },
-      attributes: [{
-        name: "module_level_variable1",
-        type: "int",
-        children: "Module level variables may be documented."
-      }],
-      get examples() {
-        return [_$createComponent(py.PyDocExample, {
-          children: "print(\"mod\")"
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.ModuleDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "This module demonstrates documentation as specified by the Google Python Style Guide."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 185,
+              columnNumber: 13
+            })];
+          },
+          attributes: [{
+            name: "module_level_variable1",
+            type: "int",
+            children: "Module level variables may be documented."
+          }],
+          get examples() {
+            return [_$createComponent(py.PyDocExample, {
+              children: "print(\"mod\")"
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 197,
+              columnNumber: 22
+            })];
+          },
+          seeAlso: ["another_module.func", "RelatedClass"],
+          warning: "Internal API.",
+          deprecated: "Use new_module instead.",
+          todo: ["For module TODOs", "You have to also use sphinx.ext.todo extension"]
         }, {
           fileName: import.meta.url,
-          lineNumber: 215,
-          columnNumber: 20
-        })];
-      },
-      seeAlso: ["another_module.func", "RelatedClass"],
-      warning: "Internal API.",
-      deprecated: "Use new_module instead.",
-      todo: ["For module TODOs", "You have to also use sphinx.ext.todo extension"]
+          lineNumber: 183,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 201,
+      lineNumber: 182,
       columnNumber: 7
-    })]);
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         """
         This module demonstrates documentation as specified by the Google Python Style
         Guide.
@@ -287,40 +335,46 @@ describe("New Documentation Components", () => {
             * You have to also use sphinx.ext.todo extension
         """
 
-
         `);
   });
   it("PropertyDoc renders correctly", () => {
-    const res = toSourceText([_$createComponent(py.PropertyDoc, {
-      get description() {
-        return [_$createComponent(Prose, {
-          children: "Properties should be documented in their getter method."
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 262,
-          columnNumber: 11
-        })];
-      },
-      returns: "str: The readonly property value.",
-      get examples() {
-        return [_$createComponent(py.PyDocExample, {
-          children: "print(obj.name)"
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.PropertyDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "Properties should be documented in their getter method."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 243,
+              columnNumber: 13
+            })];
+          },
+          returns: "str: The readonly property value.",
+          get examples() {
+            return [_$createComponent(py.PyDocExample, {
+              children: "print(obj.name)"
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 248,
+              columnNumber: 22
+            })];
+          },
+          seeAlso: ["other_property"],
+          warning: "Access may be slow.",
+          deprecated: "Use full_name instead.",
+          note: "If the setter method contains notable behavior, it should be mentioned here."
         }, {
           fileName: import.meta.url,
-          lineNumber: 267,
-          columnNumber: 20
-        })];
-      },
-      seeAlso: ["other_property"],
-      warning: "Access may be slow.",
-      deprecated: "Use full_name instead.",
-      note: "If the setter method contains notable behavior, it should be mentioned here."
+          lineNumber: 241,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 260,
+      lineNumber: 240,
       columnNumber: 7
-    })]);
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         """
         Properties should be documented in their getter method.
 
@@ -343,45 +397,51 @@ describe("New Documentation Components", () => {
             If the setter method contains notable behavior, it should be mentioned here.
         """
 
-
         `);
   });
   it("GeneratorDoc renders correctly", () => {
-    const res = toSourceText([_$createComponent(py.GeneratorDoc, {
-      get description() {
-        return [_$createComponent(Prose, {
-          children: "Generators have a Yields section instead of a Returns section."
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 308,
-          columnNumber: 11
-        })];
-      },
-      parameters: [{
-        name: "n",
-        type: "int",
-        doc: "The upper limit of the range to generate, from 0 to n - 1."
-      }],
-      yields: "int: The next number in the range of 0 to n - 1.",
-      get examples() {
-        return [_$createComponent(py.PyDocExample, {
-          children: "print(next(gen))"
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.GeneratorDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "Generators have a Yields section instead of a Returns section."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 288,
+              columnNumber: 13
+            })];
+          },
+          parameters: [{
+            name: "n",
+            type: "int",
+            doc: "The upper limit of the range to generate, from 0 to n - 1."
+          }],
+          yields: "int: The next number in the range of 0 to n - 1.",
+          get examples() {
+            return [_$createComponent(py.PyDocExample, {
+              children: "print(next(gen))"
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 300,
+              columnNumber: 22
+            })];
+          },
+          seeAlso: ["make_generator"],
+          warning: "Do not consume in tight loops without sleep.",
+          deprecated: "Use new_generator instead.",
+          note: "Examples should be written in doctest format."
         }, {
           fileName: import.meta.url,
-          lineNumber: 320,
-          columnNumber: 20
-        })];
-      },
-      seeAlso: ["make_generator"],
-      warning: "Do not consume in tight loops without sleep.",
-      deprecated: "Use new_generator instead.",
-      note: "Examples should be written in doctest format."
+          lineNumber: 286,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 306,
+      lineNumber: 285,
       columnNumber: 7
-    })]);
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         """
         Generators have a Yields section instead of a Returns section.
 
@@ -407,48 +467,54 @@ describe("New Documentation Components", () => {
             Examples should be written in doctest format.
         """
 
-
         `);
   });
   it("ExceptionDoc renders correctly", () => {
-    const res = toSourceText([_$createComponent(py.ExceptionDoc, {
-      get description() {
-        return [_$createComponent(Prose, {
-          children: "Exceptions are documented in the same way as classes."
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.ExceptionDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "Exceptions are documented in the same way as classes."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 343,
+              columnNumber: 13
+            })];
+          },
+          parameters: [{
+            name: "msg",
+            type: "str",
+            doc: "Human readable string describing the exception."
+          }, {
+            name: "code",
+            type: "int",
+            default: undefined,
+            doc: "Error code."
+          }],
+          attributes: [{
+            name: "msg",
+            type: "str",
+            children: "Human readable string describing the exception."
+          }, {
+            name: "code",
+            type: "int",
+            children: "Exception error code."
+          }],
+          seeAlso: ["BaseException"],
+          deprecated: "Use NewException instead.",
+          note: "Do not include the 'self' parameter in the Args section."
         }, {
           fileName: import.meta.url,
-          lineNumber: 364,
-          columnNumber: 11
-        })];
-      },
-      parameters: [{
-        name: "msg",
-        type: "str",
-        doc: "Human readable string describing the exception."
-      }, {
-        name: "code",
-        type: "int",
-        default: undefined,
-        doc: "Error code."
-      }],
-      attributes: [{
-        name: "msg",
-        type: "str",
-        children: "Human readable string describing the exception."
-      }, {
-        name: "code",
-        type: "int",
-        children: "Exception error code."
-      }],
-      seeAlso: ["BaseException"],
-      deprecated: "Use NewException instead.",
-      note: "Do not include the 'self' parameter in the Args section."
+          lineNumber: 341,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 362,
+      lineNumber: 340,
       columnNumber: 7
-    })]);
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         """
         Exceptions are documented in the same way as classes.
 
@@ -472,35 +538,41 @@ describe("New Documentation Components", () => {
             Do not include the 'self' parameter in the Args section.
         """
 
-
         `);
   });
   it("MethodDoc renders correctly without default note", () => {
-    const res = toSourceText([_$createComponent(py.MethodDoc, {
-      get description() {
-        return [_$createComponent(Prose, {
-          children: "Class methods are similar to regular functions."
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.MethodDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "Class methods are similar to regular functions."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 411,
+              columnNumber: 13
+            })];
+          },
+          parameters: [{
+            name: "param1",
+            doc: "The first parameter."
+          }, {
+            name: "param2",
+            doc: "The second parameter."
+          }],
+          returns: "True if successful, False otherwise.",
+          overrides: "Base.method"
         }, {
           fileName: import.meta.url,
-          lineNumber: 431,
-          columnNumber: 11
-        })];
-      },
-      parameters: [{
-        name: "param1",
-        doc: "The first parameter."
-      }, {
-        name: "param2",
-        doc: "The second parameter."
-      }],
-      returns: "True if successful, False otherwise.",
-      overrides: "Base.method"
+          lineNumber: 409,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 429,
+      lineNumber: 408,
       columnNumber: 7
-    })]);
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         """
         Class methods are similar to regular functions.
 
@@ -516,32 +588,38 @@ describe("New Documentation Components", () => {
             Base.method
         """
 
-
         `);
   });
   it("MethodDoc renders correctly with custom note", () => {
-    const res = toSourceText([_$createComponent(py.MethodDoc, {
-      get description() {
-        return [_$createComponent(Prose, {
-          children: "Class methods are similar to regular functions."
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.MethodDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "Class methods are similar to regular functions."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 453,
+              columnNumber: 13
+            })];
+          },
+          parameters: [{
+            name: "param1",
+            doc: "The first parameter."
+          }],
+          returns: "True if successful, False otherwise.",
+          note: "This method has special behavior when called multiple times."
         }, {
           fileName: import.meta.url,
-          lineNumber: 474,
-          columnNumber: 11
-        })];
-      },
-      parameters: [{
-        name: "param1",
-        doc: "The first parameter."
-      }],
-      returns: "True if successful, False otherwise.",
-      note: "This method has special behavior when called multiple times."
+          lineNumber: 451,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 472,
+      lineNumber: 450,
       columnNumber: 7
-    })]);
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         """
         Class methods are similar to regular functions.
 
@@ -555,51 +633,63 @@ describe("New Documentation Components", () => {
             This method has special behavior when called multiple times.
         """
 
-
         `);
   });
   it("ModuleDoc with minimal content", () => {
-    const res = toSourceText([_$createComponent(py.ModuleDoc, {
-      get description() {
-        return [_$createComponent(Prose, {
-          children: "Simple module description."
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.ModuleDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "Simple module description."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 488,
+              columnNumber: 25
+            })];
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 510,
-          columnNumber: 23
-        })];
+          lineNumber: 487,
+          columnNumber: 9
+        });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 509,
+      lineNumber: 486,
       columnNumber: 7
-    })]);
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         """
         Simple module description.
         """
 
-
         `);
   });
   it("ModuleDoc with only todo items", () => {
-    const res = toSourceText([_$createComponent(py.ModuleDoc, {
-      get description() {
-        return [_$createComponent(Prose, {
-          children: "Module with pending tasks."
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.ModuleDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "Module with pending tasks."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 505,
+              columnNumber: 25
+            })];
+          },
+          todo: ["Implement feature X", "Add more tests", "Update documentation"]
         }, {
           fileName: import.meta.url,
-          lineNumber: 528,
-          columnNumber: 23
-        })];
-      },
-      todo: ["Implement feature X", "Add more tests", "Update documentation"]
+          lineNumber: 504,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 527,
+      lineNumber: 503,
       columnNumber: 7
-    })]);
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         """
         Module with pending tasks.
 
@@ -609,52 +699,64 @@ describe("New Documentation Components", () => {
             * Update documentation
         """
 
-
         `);
   });
   it("PropertyDoc minimal (description only)", () => {
-    const res = toSourceText([_$createComponent(py.PropertyDoc, {
-      get description() {
-        return [_$createComponent(Prose, {
-          children: "A simple readonly property."
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.PropertyDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "A simple readonly property."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 532,
+              columnNumber: 25
+            })];
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 552,
-          columnNumber: 23
-        })];
+          lineNumber: 531,
+          columnNumber: 9
+        });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 551,
+      lineNumber: 530,
       columnNumber: 7
-    })]);
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         """
         A simple readonly property.
         """
 
-
         `);
   });
   it("PropertyDoc with getter and setter info", () => {
-    const res = toSourceText([_$createComponent(py.PropertyDoc, {
-      get description() {
-        return [_$createComponent(Prose, {
-          children: "Properties with both a getter and setter should only be documented in their getter method."
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.PropertyDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "Properties with both a getter and setter should only be documented in their getter method."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 550,
+              columnNumber: 13
+            })];
+          },
+          returns: ":obj:`list` of :obj:`str`: The property value.",
+          note: "If the setter method contains notable behavior, it should be mentioned here."
         }, {
           fileName: import.meta.url,
-          lineNumber: 571,
-          columnNumber: 11
-        })];
-      },
-      returns: ":obj:`list` of :obj:`str`: The property value.",
-      note: "If the setter method contains notable behavior, it should be mentioned here."
+          lineNumber: 548,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 569,
+      lineNumber: 547,
       columnNumber: 7
-    })]);
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         """
         Properties with both a getter and setter should only be documented in their
         getter method.
@@ -666,43 +768,49 @@ describe("New Documentation Components", () => {
             If the setter method contains notable behavior, it should be mentioned here.
         """
 
-
         `);
   });
   it("GeneratorDoc with complex parameters", () => {
-    const res = toSourceText([_$createComponent(py.GeneratorDoc, {
-      get description() {
-        return [_$createComponent(Prose, {
-          children: "A more complex generator example with multiple parameters."
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.GeneratorDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "A more complex generator example with multiple parameters."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 581,
+              columnNumber: 13
+            })];
+          },
+          parameters: [{
+            name: "start",
+            type: "int",
+            default: "0",
+            doc: "Starting value for the sequence."
+          }, {
+            name: "stop",
+            type: "int",
+            doc: "Ending value for the sequence (exclusive)."
+          }, {
+            name: "step",
+            type: "int",
+            default: "1",
+            doc: "Step size between values."
+          }],
+          yields: "int: The next number in the sequence.",
+          raises: ["ValueError: If step is zero.", "TypeError: If parameters are not integers."]
         }, {
           fileName: import.meta.url,
-          lineNumber: 603,
-          columnNumber: 11
-        })];
-      },
-      parameters: [{
-        name: "start",
-        type: "int",
-        default: "0",
-        doc: "Starting value for the sequence."
-      }, {
-        name: "stop",
-        type: "int",
-        doc: "Ending value for the sequence (exclusive)."
-      }, {
-        name: "step",
-        type: "int",
-        default: "1",
-        doc: "Step size between values."
-      }],
-      yields: "int: The next number in the sequence.",
-      raises: ["ValueError: If step is zero.", "TypeError: If parameters are not integers."]
+          lineNumber: 579,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 601,
+      lineNumber: 578,
       columnNumber: 7
-    })]);
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         """
         A more complex generator example with multiple parameters.
 
@@ -723,61 +831,67 @@ describe("New Documentation Components", () => {
             TypeError: If parameters are not integers.
         """
 
-
         `);
   });
   it("ExceptionDoc with comprehensive documentation", () => {
-    const res = toSourceText([_$createComponent(py.ExceptionDoc, {
-      get description() {
-        return [_$createComponent(Prose, {
-          children: "A custom exception for authentication failures."
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 665,
-          columnNumber: 11
-        }), _$createComponent(Prose, {
-          children: "This exception is raised when authentication credentials are invalid or when authentication tokens have expired."
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.ExceptionDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "A custom exception for authentication failures."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 642,
+              columnNumber: 13
+            }), _$createComponent(Prose, {
+              children: "This exception is raised when authentication credentials are invalid or when authentication tokens have expired."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 643,
+              columnNumber: 13
+            })];
+          },
+          parameters: [{
+            name: "message",
+            type: "str",
+            doc: "Human readable error message describing the authentication failure."
+          }, {
+            name: "error_code",
+            type: "int",
+            default: "401",
+            doc: "HTTP error code associated with the authentication failure."
+          }, {
+            name: "retry_after",
+            type: "int",
+            default: undefined,
+            doc: "Number of seconds to wait before retrying authentication."
+          }],
+          attributes: [{
+            name: "message",
+            type: "str",
+            children: "The error message."
+          }, {
+            name: "error_code",
+            type: "int",
+            children: "HTTP status code."
+          }, {
+            name: "retry_after",
+            type: "int",
+            children: "Retry delay in seconds, if applicable."
+          }],
+          note: "This exception should be caught and handled gracefully in production code."
         }, {
           fileName: import.meta.url,
-          lineNumber: 666,
-          columnNumber: 11
-        })];
-      },
-      parameters: [{
-        name: "message",
-        type: "str",
-        doc: "Human readable error message describing the authentication failure."
-      }, {
-        name: "error_code",
-        type: "int",
-        default: "401",
-        doc: "HTTP error code associated with the authentication failure."
-      }, {
-        name: "retry_after",
-        type: "int",
-        default: undefined,
-        doc: "Number of seconds to wait before retrying authentication."
-      }],
-      attributes: [{
-        name: "message",
-        type: "str",
-        children: "The error message."
-      }, {
-        name: "error_code",
-        type: "int",
-        children: "HTTP status code."
-      }, {
-        name: "retry_after",
-        type: "int",
-        children: "Retry delay in seconds, if applicable."
-      }],
-      note: "This exception should be caught and handled gracefully in production code."
+          lineNumber: 640,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 663,
+      lineNumber: 639,
       columnNumber: 7
-    })]);
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         """
         A custom exception for authentication failures.
 
@@ -804,32 +918,38 @@ describe("New Documentation Components", () => {
             This exception should be caught and handled gracefully in production code.
         """
 
-
         `);
   });
   it("MethodDoc with raises but no returns", () => {
-    const res = toSourceText([_$createComponent(py.MethodDoc, {
-      get description() {
-        return [_$createComponent(Prose, {
-          children: "A method that performs an action but doesn't return a value."
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.MethodDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "A method that performs an action but doesn't return a value."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 724,
+              columnNumber: 13
+            })];
+          },
+          parameters: [{
+            name: "data",
+            type: "bytes",
+            doc: "Raw data to process."
+          }],
+          raises: ["ValueError: If data is empty or invalid.", "IOError: If processing fails due to I/O issues."]
         }, {
           fileName: import.meta.url,
-          lineNumber: 748,
-          columnNumber: 11
-        })];
-      },
-      parameters: [{
-        name: "data",
-        type: "bytes",
-        doc: "Raw data to process."
-      }],
-      raises: ["ValueError: If data is empty or invalid.", "IOError: If processing fails due to I/O issues."]
+          lineNumber: 722,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 746,
+      lineNumber: 721,
       columnNumber: 7
-    })]);
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         """
         A method that performs an action but doesn't return a value.
 
@@ -843,28 +963,34 @@ describe("New Documentation Components", () => {
             IOError: If processing fails due to I/O issues.
         """
 
-
         `);
   });
   it("MethodDoc with no parameters", () => {
-    const res = toSourceText([_$createComponent(py.MethodDoc, {
-      get description() {
-        return [_$createComponent(Prose, {
-          children: "A simple method with no parameters (except self)."
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.MethodDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "A simple method with no parameters (except self)."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 765,
+              columnNumber: 13
+            })];
+          },
+          returns: "bool: True if the operation was successful.",
+          note: "This is a parameterless method that only operates on instance state."
         }, {
           fileName: import.meta.url,
-          lineNumber: 790,
-          columnNumber: 11
-        })];
-      },
-      returns: "bool: True if the operation was successful.",
-      note: "This is a parameterless method that only operates on instance state."
+          lineNumber: 763,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 788,
+      lineNumber: 762,
       columnNumber: 7
-    })]);
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         """
         A simple method with no parameters (except self).
 
@@ -875,66 +1001,78 @@ describe("New Documentation Components", () => {
             This is a parameterless method that only operates on instance state.
         """
 
-
         `);
   });
   it("AttributeDoc standalone usage", () => {
-    const res = toSourceText([_$createComponent(py.PyDoc, {
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return _$createComponent(py.AttributeDoc, {
-          name: "connection_timeout",
-          type: "float",
-          children: "Maximum time in seconds to wait for a connection to be established."
+        return _$createComponent(py.PyDoc, {
+          get children() {
+            return _$createComponent(py.AttributeDoc, {
+              name: "connection_timeout",
+              type: "float",
+              children: "Maximum time in seconds to wait for a connection to be established."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 791,
+              columnNumber: 11
+            });
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 817,
+          lineNumber: 790,
           columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 816,
+      lineNumber: 789,
       columnNumber: 7
-    })]);
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         """
         connection_timeout (float): Maximum time in seconds to wait for a connection to
             be established.
         """
 
-
         `);
   });
   it("GeneratorDoc with examples in description", () => {
-    const res = toSourceText([_$createComponent(py.GeneratorDoc, {
-      get description() {
-        return [_$createComponent(Prose, {
-          children: "Generators have a Yields section instead of a Returns section."
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 839,
-          columnNumber: 11
-        }), _$createComponent(py.PyDocExample, {
-          children: `print([i for i in example_generator(4)])\n[0, 1, 2, 3]`
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.GeneratorDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "Generators have a Yields section instead of a Returns section."
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 812,
+              columnNumber: 13
+            }), _$createComponent(py.PyDocExample, {
+              children: `print([i for i in example_generator(4)])\n[0, 1, 2, 3]`
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 815,
+              columnNumber: 13
+            })];
+          },
+          parameters: [{
+            name: "n",
+            type: "int",
+            doc: "The upper limit of the range to generate, from 0 to n - 1."
+          }],
+          yields: "int: The next number in the range of 0 to n - 1.",
+          note: "Examples should be written in doctest format, and should illustrate how to use the function."
         }, {
           fileName: import.meta.url,
-          lineNumber: 842,
-          columnNumber: 11
-        })];
-      },
-      parameters: [{
-        name: "n",
-        type: "int",
-        doc: "The upper limit of the range to generate, from 0 to n - 1."
-      }],
-      yields: "int: The next number in the range of 0 to n - 1.",
-      note: "Examples should be written in doctest format, and should illustrate how to use the function."
+          lineNumber: 810,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 837,
+      lineNumber: 809,
       columnNumber: 7
-    })]);
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         """
         Generators have a Yields section instead of a Returns section.
 
@@ -951,7 +1089,6 @@ describe("New Documentation Components", () => {
             Examples should be written in doctest format, and should illustrate how to use the function.
         """
 
-
         `);
   });
 });
@@ -963,13 +1100,13 @@ describe("Full example", () => {
           children: "This is an example of a long docstring that will be broken in lines. We will also render another paragraph after this one."
         }, {
           fileName: import.meta.url,
-          lineNumber: 887,
+          lineNumber: 858,
           columnNumber: 11
         }), _$createComponent(py.PyDocExample, {
           children: `print("Hello world!")\nx = "Hello"\nprint(x)`
         }, {
           fileName: import.meta.url,
-          lineNumber: 891,
+          lineNumber: 862,
           columnNumber: 11
         })];
       },
@@ -987,7 +1124,7 @@ describe("Full example", () => {
           children: "print(\"class-doc\")"
         }, {
           fileName: import.meta.url,
-          lineNumber: 907,
+          lineNumber: 878,
           columnNumber: 20
         })];
       },
@@ -1008,55 +1145,57 @@ describe("Full example", () => {
       style: "google"
     }, {
       fileName: import.meta.url,
-      lineNumber: 885,
+      lineNumber: 856,
       columnNumber: 7
     });
-    const res = toSourceText([_$createComponent(py.ClassDeclaration, {
-      name: "A",
-      doc: doc,
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return _$createComponent(py.StatementList, {
+        return _$createComponent(py.ClassDeclaration, {
+          name: "A",
+          doc: doc,
           get children() {
-            return [_$createComponent(py.VariableDeclaration, {
-              name: "just_name"
-            }, {
-              fileName: import.meta.url,
-              lineNumber: 932,
-              columnNumber: 13
-            }), _$createComponent(py.VariableDeclaration, {
-              name: "name_and_type",
-              type: "int"
-            }, {
-              fileName: import.meta.url,
-              lineNumber: 933,
-              columnNumber: 13
-            }), _$createComponent(py.VariableDeclaration, {
-              name: "name_type_and_value",
-              type: "int",
-              initializer: 12
+            return _$createComponent(py.StatementList, {
+              get children() {
+                return [_$createComponent(py.VariableDeclaration, {
+                  name: "just_name"
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 904,
+                  columnNumber: 13
+                }), _$createComponent(py.VariableDeclaration, {
+                  name: "name_and_type",
+                  type: "int"
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 905,
+                  columnNumber: 13
+                }), _$createComponent(py.VariableDeclaration, {
+                  name: "name_type_and_value",
+                  type: "int",
+                  initializer: 12
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 906,
+                  columnNumber: 13
+                })];
+              }
             }, {
               fileName: import.meta.url,
-              lineNumber: 934,
-              columnNumber: 13
-            })];
+              lineNumber: 903,
+              columnNumber: 11
+            });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 931,
-          columnNumber: 11
+          lineNumber: 902,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 930,
-      columnNumber: 9
-    })], {
-      printOptions: {
-        printWidth: 80,
-        tabWidth: 4
-      }
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 901,
+      columnNumber: 7
+    })).toRenderTo(`
           class A:
               """
               This is an example of a long docstring that will be broken in lines. We will
@@ -1077,7 +1216,7 @@ describe("Full example", () => {
                       nickname, or even a codename (e.g., 'Agent X'). It's used primarily
                       for display purposes, logging, or greeting messages and is not
                       required to be unique or validated unless specified by the caller.
-                      Defaults to \"John Doe\".
+                      Defaults to "John Doe".
 
                   somebody2 (str): Somebody's name. This can be any string representing a
                       person, whether it's a first name, full name, nickname, or even a
@@ -1106,7 +1245,6 @@ describe("Full example", () => {
               name_and_type: int = None
               name_type_and_value: int = 12
 
-
           `);
   });
   it("renders correctly in a Function", () => {
@@ -1116,13 +1254,13 @@ describe("Full example", () => {
           children: "This is an example of a long docstring that will be broken in lines. We will also render another paragraph after this one."
         }, {
           fileName: import.meta.url,
-          lineNumber: 1005,
+          lineNumber: 973,
           columnNumber: 11
         }), _$createComponent(py.PyDocExample, {
           children: `print("Hello world!")\nx = "Hello"\nprint(x)`
         }, {
           fileName: import.meta.url,
-          lineNumber: 1009,
+          lineNumber: 977,
           columnNumber: 11
         })];
       },
@@ -1143,55 +1281,57 @@ describe("Full example", () => {
       style: "google"
     }, {
       fileName: import.meta.url,
-      lineNumber: 1003,
+      lineNumber: 971,
       columnNumber: 7
     });
-    const res = toSourceText([_$createComponent(py.FunctionDeclaration, {
-      name: "some_function",
-      doc: doc,
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return _$createComponent(py.StatementList, {
+        return _$createComponent(py.FunctionDeclaration, {
+          name: "some_function",
+          doc: doc,
           get children() {
-            return [_$createComponent(py.VariableDeclaration, {
-              name: "just_name"
-            }, {
-              fileName: import.meta.url,
-              lineNumber: 1037,
-              columnNumber: 13
-            }), _$createComponent(py.VariableDeclaration, {
-              name: "name_and_type",
-              type: "number"
-            }, {
-              fileName: import.meta.url,
-              lineNumber: 1038,
-              columnNumber: 13
-            }), _$createComponent(py.VariableDeclaration, {
-              name: "name_type_and_value",
-              type: "number",
-              initializer: 12
+            return _$createComponent(py.StatementList, {
+              get children() {
+                return [_$createComponent(py.VariableDeclaration, {
+                  name: "just_name"
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 1006,
+                  columnNumber: 13
+                }), _$createComponent(py.VariableDeclaration, {
+                  name: "name_and_type",
+                  type: "number"
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 1007,
+                  columnNumber: 13
+                }), _$createComponent(py.VariableDeclaration, {
+                  name: "name_type_and_value",
+                  type: "number",
+                  initializer: 12
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 1008,
+                  columnNumber: 13
+                })];
+              }
             }, {
               fileName: import.meta.url,
-              lineNumber: 1039,
-              columnNumber: 13
-            })];
+              lineNumber: 1005,
+              columnNumber: 11
+            });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 1036,
-          columnNumber: 11
+          lineNumber: 1004,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 1035,
-      columnNumber: 9
-    })], {
-      printOptions: {
-        printWidth: 80,
-        tabWidth: 4
-      }
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 1003,
+      columnNumber: 7
+    })).toRenderTo(`
           def some_function():
               """
               This is an example of a long docstring that will be broken in lines. We will
@@ -1207,7 +1347,7 @@ describe("Full example", () => {
                       nickname, or even a codename (e.g., 'Agent X'). It's used primarily
                       for display purposes, logging, or greeting messages and is not
                       required to be unique or validated unless specified by the caller.
-                      Defaults to \"John Doe\".
+                      Defaults to "John Doe".
 
                   somebody2 (str): Somebody's name. This can be any string representing a
                       person, whether it's a first name, full name, nickname, or even a
@@ -1231,25 +1371,26 @@ describe("Full example", () => {
               name_and_type: number = None
               name_type_and_value: number = 12
 
-
           `);
   });
   it("renders correctly in a Variable", () => {
-    const res = toSourceText([_$createComponent(py.VariableDeclaration, {
-      name: "myVar",
-      initializer: 42,
-      doc: "This is a very long docstring that will be broken in two lines when rendered. This part of the docstring will be in the second line."
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.VariableDeclaration, {
+          name: "myVar",
+          initializer: 42,
+          doc: "This is a very long docstring that will be broken in two lines when rendered. This part of the docstring will be in the second line."
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 1064,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 1099,
-      columnNumber: 9
-    })], {
-      printOptions: {
-        printWidth: 80,
-        tabWidth: 4
-      }
-    });
-    expect(res).toRenderTo(d`
+      lineNumber: 1063,
+      columnNumber: 7
+    })).toRenderTo(`
             # This is a very long docstring that will be broken in two lines when rendered.
             # This part of the docstring will be in the second line.
             my_var = 42
@@ -1262,40 +1403,46 @@ describe("Full example", () => {
           children: "An enum representing colors."
         }, {
           fileName: import.meta.url,
-          lineNumber: 1120,
+          lineNumber: 1082,
           columnNumber: 23
         })];
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 1119,
+      lineNumber: 1081,
       columnNumber: 7
     });
-    const result = toSourceText([_$createComponent(py.ClassEnumDeclaration, {
-      name: "Color",
-      baseType: "IntEnum",
-      members: [{
-        name: "RED",
-        value: "1",
-        doc: "The color red."
-      }, {
-        name: "GREEN",
-        value: "2",
-        doc: "The color green."
-      }, {
-        name: "BLUE",
-        value: "3",
-        doc: "The color blue."
-      }],
-      doc: doc
+    expect(_$createComponent(TestOutput, {
+      externals: [enumModule],
+      get children() {
+        return _$createComponent(py.ClassEnumDeclaration, {
+          name: "Color",
+          baseType: "IntEnum",
+          members: [{
+            name: "RED",
+            value: "1",
+            doc: "The color red."
+          }, {
+            name: "GREEN",
+            value: "2",
+            doc: "The color green."
+          }, {
+            name: "BLUE",
+            value: "3",
+            doc: "The color blue."
+          }],
+          doc: doc
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 1087,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 1125,
-      columnNumber: 9
-    })], {
-      externals: [enumModule]
-    });
-    const expected = d`
+      lineNumber: 1086,
+      columnNumber: 7
+    })).toRenderTo(`
       from enum import IntEnum
 
 
@@ -1317,9 +1464,7 @@ describe("Full example", () => {
           The color blue.
           """
 
-
-    `;
-    expect(result).toRenderTo(expected);
+    `);
   });
   it("ModuleDoc with SourceFile integration", () => {
     const moduleDoc = _$createComponent(py.ModuleDoc, {
@@ -1328,7 +1473,7 @@ describe("Full example", () => {
           children: "This module provides utility functions for data processing. It includes functions for validation, transformation, and analysis."
         }, {
           fileName: import.meta.url,
-          lineNumber: 1169,
+          lineNumber: 1129,
           columnNumber: 11
         })];
       },
@@ -1345,69 +1490,72 @@ describe("Full example", () => {
       style: "google"
     }, {
       fileName: import.meta.url,
-      lineNumber: 1167,
+      lineNumber: 1127,
       columnNumber: 7
     });
-    const content = _$createComponent(py.SourceFile, {
-      path: "utils.py",
-      doc: moduleDoc,
+    expect(_$createComponent(TestOutputDirectory, {
       get children() {
-        return [_$createComponent(py.VariableDeclaration, {
-          name: "DEFAULT_TIMEOUT",
-          initializer: 30
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 1193,
-          columnNumber: 9
-        }), _$createComponent(py.VariableDeclaration, {
-          name: "MAX_RETRIES",
-          initializer: 3
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 1194,
-          columnNumber: 9
-        }), _$createComponent(py.FunctionDeclaration, {
-          name: "process_data",
-          children: "pass"
+        return _$createComponent(py.SourceFile, {
+          path: "utils.py",
+          doc: moduleDoc,
+          get children() {
+            return [_$createComponent(py.VariableDeclaration, {
+              name: "DEFAULT_TIMEOUT",
+              initializer: 30
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 1154,
+              columnNumber: 11
+            }), _$createComponent(py.VariableDeclaration, {
+              name: "MAX_RETRIES",
+              initializer: 3
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 1155,
+              columnNumber: 11
+            }), _$createComponent(py.FunctionDeclaration, {
+              name: "process_data",
+              children: "pass"
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 1156,
+              columnNumber: 11
+            })];
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 1195,
+          lineNumber: 1153,
           columnNumber: 9
-        })];
+        });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 1192,
+      lineNumber: 1152,
       columnNumber: 7
-    });
-    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.
-
-        Attributes:
-            DEFAULT_TIMEOUT (int): Default timeout value in seconds.
+    })).toRenderTo({
+      "utils.py": `
+          """
+          This module provides utility functions for data processing. It includes
+          functions for validation, transformation, and analysis.
 
-            MAX_RETRIES (int): Maximum number of retry attempts.
+          Attributes:
+              DEFAULT_TIMEOUT (int): Default timeout value in seconds.
 
-        Todo:
-            * Add caching functionality
-            * Improve error messages
-        """
+              MAX_RETRIES (int): Maximum number of retry attempts.
 
-        default_timeout = 30
+          Todo:
+              * Add caching functionality
+              * Improve error messages
+          """
 
-        max_retries = 3
+          default_timeout = 30
 
-        def process_data():
-            pass
+          max_retries = 3
 
+          def process_data():
+              pass
 
-        `
+          `
     });
   });
   it("GeneratorDoc with FunctionDeclaration integration", () => {
@@ -1417,7 +1565,7 @@ describe("Full example", () => {
           children: "A generator function that yields fibonacci numbers. This is an efficient way to generate the sequence on demand."
         }, {
           fileName: import.meta.url,
-          lineNumber: 1239,
+          lineNumber: 1192,
           columnNumber: 11
         })];
       },
@@ -1430,19 +1578,26 @@ describe("Full example", () => {
       style: "google"
     }, {
       fileName: import.meta.url,
-      lineNumber: 1237,
+      lineNumber: 1190,
       columnNumber: 7
     });
-    const result = toSourceText([_$createComponent(py.FunctionDeclaration, {
-      name: "fibonacci_generator",
-      doc: generatorDoc,
-      children: "yield 0"
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.FunctionDeclaration, {
+          name: "fibonacci_generator",
+          doc: generatorDoc,
+          children: "yield 0"
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 1211,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 1257,
+      lineNumber: 1210,
       columnNumber: 7
-    })]);
-    expect(result).toRenderTo(d`
+    })).toRenderTo(`
         def fibonacci_generator():
             """
             A generator function that yields fibonacci numbers. This is an efficient way
@@ -1456,7 +1611,6 @@ describe("Full example", () => {
             """
             yield 0
 
-
         `);
   });
   it("ExceptionDoc with ClassDeclaration integration", () => {
@@ -1466,7 +1620,7 @@ describe("Full example", () => {
           children: "Custom exception raised when data validation fails. This exception includes details about the validation error."
         }, {
           fileName: import.meta.url,
-          lineNumber: 1286,
+          lineNumber: 1238,
           columnNumber: 11
         })];
       },
@@ -1482,44 +1636,51 @@ describe("Full example", () => {
       style: "google"
     }, {
       fileName: import.meta.url,
-      lineNumber: 1284,
+      lineNumber: 1236,
       columnNumber: 7
     });
-    const result = toSourceText([_$createComponent(py.ClassDeclaration, {
-      name: "ValidationError",
-      bases: ["Exception"],
-      doc: exceptionDoc,
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return _$createComponent(py.StatementList, {
+        return _$createComponent(py.ClassDeclaration, {
+          name: "ValidationError",
+          bases: ["Exception"],
+          doc: exceptionDoc,
           get children() {
-            return [_$createComponent(py.VariableDeclaration, {
-              name: "field_name",
-              type: "str"
-            }, {
-              fileName: import.meta.url,
-              lineNumber: 1314,
-              columnNumber: 11
-            }), _$createComponent(py.VariableDeclaration, {
-              name: "error_code",
-              type: "int"
+            return _$createComponent(py.StatementList, {
+              get children() {
+                return [_$createComponent(py.VariableDeclaration, {
+                  name: "field_name",
+                  type: "str"
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 1267,
+                  columnNumber: 13
+                }), _$createComponent(py.VariableDeclaration, {
+                  name: "error_code",
+                  type: "int"
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 1268,
+                  columnNumber: 13
+                })];
+              }
             }, {
               fileName: import.meta.url,
-              lineNumber: 1315,
+              lineNumber: 1266,
               columnNumber: 11
-            })];
+            });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 1313,
+          lineNumber: 1261,
           columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 1308,
+      lineNumber: 1260,
       columnNumber: 7
-    })]);
-    expect(result).toRenderTo(d`
+    })).toRenderTo(`
         class ValidationError(Exception):
             """
             Custom exception raised when data validation fails. This exception includes
@@ -1534,7 +1695,6 @@ describe("Full example", () => {
             field_name: str = None
             error_code: int = None
 
-
         `);
   });
   it("PropertyDoc with FunctionDeclaration (as property method) integration", () => {
@@ -1544,36 +1704,43 @@ describe("Full example", () => {
           children: "The full name of the person, combining first and last name. This property automatically formats the name with proper capitalization."
         }, {
           fileName: import.meta.url,
-          lineNumber: 1345,
+          lineNumber: 1296,
           columnNumber: 11
         })];
       },
       style: "google"
     }, {
       fileName: import.meta.url,
-      lineNumber: 1343,
+      lineNumber: 1294,
       columnNumber: 7
     });
-    const result = toSourceText([_$createComponent(py.ClassDeclaration, {
-      name: "Person",
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return _$createComponent(py.PropertyDeclaration, {
-          name: "full_name",
-          doc: propertyDoc,
-          type: "str",
-          children: "return \"John Doe\""
+        return _$createComponent(py.ClassDeclaration, {
+          name: "Person",
+          get children() {
+            return _$createComponent(py.PropertyDeclaration, {
+              name: "full_name",
+              doc: propertyDoc,
+              type: "str",
+              children: "return \"John Doe\""
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 1308,
+              columnNumber: 11
+            });
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 1356,
+          lineNumber: 1307,
           columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 1355,
+      lineNumber: 1306,
       columnNumber: 7
-    })]);
-    expect(result).toRenderTo(`
+    })).toRenderTo(`
       class Person:
           @property
           def full_name(self) -> str:
@@ -1593,7 +1760,7 @@ describe("Full example", () => {
           children: "Validates the input data according to the defined schema. This method performs comprehensive validation including type checking."
         }, {
           fileName: import.meta.url,
-          lineNumber: 1380,
+          lineNumber: 1333,
           columnNumber: 11
         })];
       },
@@ -1613,37 +1780,44 @@ describe("Full example", () => {
       style: "google"
     }, {
       fileName: import.meta.url,
-      lineNumber: 1378,
+      lineNumber: 1331,
       columnNumber: 7
     });
-    const result = toSourceText([_$createComponent(py.ClassDeclaration, {
-      name: "DataValidator",
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return _$createComponent(py.MethodDeclaration, {
-          name: "validate",
-          doc: methodDoc,
-          parameters: [{
-            name: "data",
-            type: "dict"
-          }, {
-            name: "strict",
-            type: "bool",
-            default: true
-          }],
-          returnType: "bool",
-          children: "return self.validate(data, strict)"
+        return _$createComponent(py.ClassDeclaration, {
+          name: "DataValidator",
+          get children() {
+            return _$createComponent(py.MethodDeclaration, {
+              name: "validate",
+              doc: methodDoc,
+              parameters: [{
+                name: "data",
+                type: "dict"
+              }, {
+                name: "strict",
+                type: "bool",
+                default: true
+              }],
+              returnType: "bool",
+              children: "return self.validate(data, strict)"
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 1361,
+              columnNumber: 11
+            });
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 1407,
+          lineNumber: 1360,
           columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 1406,
+      lineNumber: 1359,
       columnNumber: 7
-    })]);
-    expect(result).toRenderTo(d`
+    })).toRenderTo(`
         class DataValidator:
             def validate(self, data: dict, strict: bool = True) -> bool:
                 """
@@ -1668,7 +1842,6 @@ describe("Full example", () => {
                 return self.validate(data, strict)
 
 
-
         `);
   });
 });