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

package/dist/test/pydocs.test.js

@@ -1,26 +1,24 @@
-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."
-        }), _$createComponent(Prose, {
-          children: "This is another paragraph, and there's a line break before it."
-        })];
-      }
-    })], {
-      printOptions: {
-        printWidth: 40
+        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."
+            }), _$createComponent(Prose, {
+              children: "This is another paragraph, and there's a line break before it."
+            })];
+          }
+        });
       }
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
         """
         This is an example of a long docstring
         that will be broken in lines. We will
@@ -31,26 +29,27 @@ 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."
-        }), _$createComponent(py.PyDocExample, {
-          children: "print(\"Hello world!\")"
-        })];
-      }
-    })], {
-      printOptions: {
-        printWidth: 40
+        return _$createComponent(py.PyDoc, {
+          get children() {
+            return [_$createComponent(Prose, {
+              children: "This is an example of a docstring with a code sample."
+            }), _$createComponent(py.PyDocExample, {
+              children: "print(\"Hello world!\")"
+            })];
+          }
+        });
       }
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
       """
       This is an example of a docstring with a
       code sample.
@@ -58,26 +57,27 @@ 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."
-        }), _$createComponent(py.PyDocExample, {
-          children: "print(\"Hello world!\")"
-        }), _$createComponent(py.PyDocExample, {
-          children: "print(\"Hello world again!\")"
-        })];
-      }
-    })], {
-      printOptions: {
-        printWidth: 40
+        return _$createComponent(py.PyDoc, {
+          get children() {
+            return [_$createComponent(Prose, {
+              children: "This is an example of a docstring with a code sample."
+            }), _$createComponent(py.PyDocExample, {
+              children: "print(\"Hello world!\")"
+            }), _$createComponent(py.PyDocExample, {
+              children: "print(\"Hello world again!\")"
+            })];
+          }
+        });
       }
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
       """
       This is an example of a docstring with a
       code sample.
@@ -87,24 +87,25 @@ 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."
-        }), _$createComponent(py.PyDocExample, {
-          children: `print("Hello world!")\nx = "Hello"\nprint(x)`
-        })];
-      }
-    })], {
-      printOptions: {
-        printWidth: 40
+        return _$createComponent(py.PyDoc, {
+          get children() {
+            return [_$createComponent(Prose, {
+              children: "This is an example of a docstring with a code sample."
+            }), _$createComponent(py.PyDocExample, {
+              children: `print("Hello world!")\nx = "Hello"\nprint(x)`
+            })];
+          }
+        });
       }
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
       """
       This is an example of a docstring with a
       code sample.
@@ -114,74 +115,85 @@ 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(res).toRenderTo(d`
+    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."
+        });
+      }
+    })).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(res).toRenderTo(d`
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.SimpleCommentBlock, {
+          children: "First line of comment.\\nSecond line of comment."
+        });
+      }
+    })).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(res).toRenderTo(d`
-          x = 42  # This is an inline comment
-          `);
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return [["x = 42", _$createComponent(py.SimpleInlineComment, {
+          children: "This is an inline comment"
+        })]];
+      }
+    })).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(res).toRenderTo(d`
-          result = calculate()  # TODO: Add error handling here
-          `);
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return [["result = calculate()", _$createComponent(py.SimpleInlineComment, {
+          children: "TODO: Add error handling here"
+        })]];
+      }
+    })).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."
-        })];
-      },
-      attributes: [{
-        name: "module_level_variable1",
-        type: "int",
-        children: "Module level variables may be documented."
-      }],
-      get examples() {
-        return [_$createComponent(py.PyDocExample, {
-          children: "print(\"mod\")"
-        })];
-      },
-      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"]
-    })]);
-    expect(res).toRenderTo(d`
+    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."
+            })];
+          },
+          attributes: [{
+            name: "module_level_variable1",
+            type: "int",
+            children: "Module level variables may be documented."
+          }],
+          get examples() {
+            return [_$createComponent(py.PyDocExample, {
+              children: "print(\"mod\")"
+            })];
+          },
+          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"]
+        });
+      }
+    })).toRenderTo(`
         """
         This module demonstrates documentation as specified by the Google Python Style
         Guide.
@@ -207,28 +219,30 @@ 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."
-        })];
-      },
-      returns: "str: The readonly property value.",
-      get examples() {
-        return [_$createComponent(py.PyDocExample, {
-          children: "print(obj.name)"
-        })];
-      },
-      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."
-    })]);
-    expect(res).toRenderTo(d`
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.PropertyDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "Properties should be documented in their getter method."
+            })];
+          },
+          returns: "str: The readonly property value.",
+          get examples() {
+            return [_$createComponent(py.PyDocExample, {
+              children: "print(obj.name)"
+            })];
+          },
+          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."
+        });
+      }
+    })).toRenderTo(`
         """
         Properties should be documented in their getter method.
 
@@ -251,33 +265,35 @@ 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."
-        })];
-      },
-      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))"
-        })];
-      },
-      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."
-    })]);
-    expect(res).toRenderTo(d`
+    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."
+            })];
+          },
+          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))"
+            })];
+          },
+          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."
+        });
+      }
+    })).toRenderTo(`
         """
         Generators have a Yields section instead of a Returns section.
 
@@ -303,40 +319,42 @@ 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."
-        })];
-      },
-      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."
-    })]);
-    expect(res).toRenderTo(d`
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.ExceptionDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "Exceptions are documented in the same way as classes."
+            })];
+          },
+          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."
+        });
+      }
+    })).toRenderTo(`
         """
         Exceptions are documented in the same way as classes.
 
@@ -360,27 +378,29 @@ 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."
-        })];
-      },
-      parameters: [{
-        name: "param1",
-        doc: "The first parameter."
-      }, {
-        name: "param2",
-        doc: "The second parameter."
-      }],
-      returns: "True if successful, False otherwise.",
-      overrides: "Base.method"
-    })]);
-    expect(res).toRenderTo(d`
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.MethodDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "Class methods are similar to regular functions."
+            })];
+          },
+          parameters: [{
+            name: "param1",
+            doc: "The first parameter."
+          }, {
+            name: "param2",
+            doc: "The second parameter."
+          }],
+          returns: "True if successful, False otherwise.",
+          overrides: "Base.method"
+        });
+      }
+    })).toRenderTo(`
         """
         Class methods are similar to regular functions.
 
@@ -396,24 +416,26 @@ 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."
-        })];
-      },
-      parameters: [{
-        name: "param1",
-        doc: "The first parameter."
-      }],
-      returns: "True if successful, False otherwise.",
-      note: "This method has special behavior when called multiple times."
-    })]);
-    expect(res).toRenderTo(d`
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.MethodDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "Class methods are similar to regular functions."
+            })];
+          },
+          parameters: [{
+            name: "param1",
+            doc: "The first parameter."
+          }],
+          returns: "True if successful, False otherwise.",
+          note: "This method has special behavior when called multiple times."
+        });
+      }
+    })).toRenderTo(`
         """
         Class methods are similar to regular functions.
 
@@ -427,35 +449,39 @@ 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."
+            })];
+          }
+        });
       }
-    })]);
-    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."
-        })];
-      },
-      todo: ["Implement feature X", "Add more tests", "Update documentation"]
-    })]);
-    expect(res).toRenderTo(d`
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.ModuleDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "Module with pending tasks."
+            })];
+          },
+          todo: ["Implement feature X", "Add more tests", "Update documentation"]
+        });
+      }
+    })).toRenderTo(`
         """
         Module with pending tasks.
 
@@ -465,36 +491,40 @@ 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."
+            })];
+          }
+        });
       }
-    })]);
-    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."
-        })];
-      },
-      returns: ":obj:`list` of :obj:`str`: The property value.",
-      note: "If the setter method contains notable behavior, it should be mentioned here."
-    })]);
-    expect(res).toRenderTo(d`
+    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."
+            })];
+          },
+          returns: ":obj:`list` of :obj:`str`: The property value.",
+          note: "If the setter method contains notable behavior, it should be mentioned here."
+        });
+      }
+    })).toRenderTo(`
         """
         Properties with both a getter and setter should only be documented in their
         getter method.
@@ -506,35 +536,37 @@ 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."
-        })];
-      },
-      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."]
-    })]);
-    expect(res).toRenderTo(d`
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.GeneratorDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "A more complex generator example with multiple parameters."
+            })];
+          },
+          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."]
+        });
+      }
+    })).toRenderTo(`
         """
         A more complex generator example with multiple parameters.
 
@@ -555,49 +587,51 @@ 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."
-        }), _$createComponent(Prose, {
-          children: "This exception is raised when authentication credentials are invalid or when authentication tokens have expired."
-        })];
-      },
-      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."
-    })]);
-    expect(res).toRenderTo(d`
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.ExceptionDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "A custom exception for authentication failures."
+            }), _$createComponent(Prose, {
+              children: "This exception is raised when authentication credentials are invalid or when authentication tokens have expired."
+            })];
+          },
+          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."
+        });
+      }
+    })).toRenderTo(`
         """
         A custom exception for authentication failures.
 
@@ -624,24 +658,26 @@ 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."
-        })];
-      },
-      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."]
-    })]);
-    expect(res).toRenderTo(d`
+    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."
+            })];
+          },
+          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."]
+        });
+      }
+    })).toRenderTo(`
         """
         A method that performs an action but doesn't return a value.
 
@@ -655,20 +691,22 @@ 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)."
-        })];
-      },
-      returns: "bool: True if the operation was successful.",
-      note: "This is a parameterless method that only operates on instance state."
-    })]);
-    expect(res).toRenderTo(d`
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.MethodDoc, {
+          get description() {
+            return [_$createComponent(Prose, {
+              children: "A simple method with no parameters (except self)."
+            })];
+          },
+          returns: "bool: True if the operation was successful.",
+          note: "This is a parameterless method that only operates on instance state."
+        });
+      }
+    })).toRenderTo(`
         """
         A simple method with no parameters (except self).
 
@@ -679,46 +717,50 @@ 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."
+            });
+          }
         });
       }
-    })]);
-    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."
-        }), _$createComponent(py.PyDocExample, {
-          children: `print([i for i in example_generator(4)])\n[0, 1, 2, 3]`
-        })];
-      },
-      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."
-    })]);
-    expect(res).toRenderTo(d`
+    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."
+            }), _$createComponent(py.PyDocExample, {
+              children: `print([i for i in example_generator(4)])\n[0, 1, 2, 3]`
+            })];
+          },
+          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."
+        });
+      }
+    })).toRenderTo(`
         """
         Generators have a Yields section instead of a Returns section.
 
@@ -735,7 +777,6 @@ describe("New Documentation Components", () => {
             Examples should be written in doctest format, and should illustrate how to use the function.
         """
 
-
         `);
   });
 });
@@ -779,32 +820,30 @@ describe("Full example", () => {
       note: "Do not include the 'self' parameter in the Args section.",
       style: "google"
     });
-    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"
-            }), _$createComponent(py.VariableDeclaration, {
-              name: "name_and_type",
-              type: "int"
-            }), _$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"
+                }), _$createComponent(py.VariableDeclaration, {
+                  name: "name_and_type",
+                  type: "int"
+                }), _$createComponent(py.VariableDeclaration, {
+                  name: "name_type_and_value",
+                  type: "int",
+                  initializer: 12
+                })];
+              }
+            });
           }
         });
       }
-    })], {
-      printOptions: {
-        printWidth: 80,
-        tabWidth: 4
-      }
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
           class A:
               """
               This is an example of a long docstring that will be broken in lines. We will
@@ -825,7 +864,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
@@ -854,7 +893,6 @@ describe("Full example", () => {
               name_and_type: int = None
               name_type_and_value: int = 12
 
-
           `);
   });
   it("renders correctly in a Function", () => {
@@ -882,32 +920,30 @@ describe("Full example", () => {
       note: "This function can be used as both a regular function and a generator.",
       style: "google"
     });
-    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"
-            }), _$createComponent(py.VariableDeclaration, {
-              name: "name_and_type",
-              type: "number"
-            }), _$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"
+                }), _$createComponent(py.VariableDeclaration, {
+                  name: "name_and_type",
+                  type: "number"
+                }), _$createComponent(py.VariableDeclaration, {
+                  name: "name_type_and_value",
+                  type: "number",
+                  initializer: 12
+                })];
+              }
+            });
           }
         });
       }
-    })], {
-      printOptions: {
-        printWidth: 80,
-        tabWidth: 4
-      }
-    });
-    expect(res).toRenderTo(d`
+    })).toRenderTo(`
           def some_function():
               """
               This is an example of a long docstring that will be broken in lines. We will
@@ -923,7 +959,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
@@ -947,21 +983,18 @@ 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."
-    })], {
-      printOptions: {
-        printWidth: 80,
-        tabWidth: 4
+    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."
+        });
       }
-    });
-    expect(res).toRenderTo(d`
+    })).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
@@ -975,27 +1008,29 @@ describe("Full example", () => {
         })];
       }
     });
-    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
-    })], {
-      externals: [enumModule]
-    });
-    const expected = d`
+    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
+        });
+      }
+    })).toRenderTo(`
       from enum import IntEnum
 
 
@@ -1017,9 +1052,7 @@ describe("Full example", () => {
           The color blue.
           """
 
-
-    `;
-    expect(result).toRenderTo(expected);
+    `);
   });
   it("ModuleDoc with SourceFile integration", () => {
     const moduleDoc = _$createComponent(py.ModuleDoc, {
@@ -1040,50 +1073,49 @@ describe("Full example", () => {
       todo: ["Add caching functionality", "Improve error messages"],
       style: "google"
     });
-    const content = _$createComponent(py.SourceFile, {
-      path: "utils.py",
-      doc: moduleDoc,
+    expect(_$createComponent(TestOutputDirectory, {
       get children() {
-        return [_$createComponent(py.VariableDeclaration, {
-          name: "DEFAULT_TIMEOUT",
-          initializer: 30
-        }), _$createComponent(py.VariableDeclaration, {
-          name: "MAX_RETRIES",
-          initializer: 3
-        }), _$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
+            }), _$createComponent(py.VariableDeclaration, {
+              name: "MAX_RETRIES",
+              initializer: 3
+            }), _$createComponent(py.FunctionDeclaration, {
+              name: "process_data",
+              children: "pass"
+            })];
+          }
+        });
       }
-    });
-    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", () => {
@@ -1101,12 +1133,15 @@ describe("Full example", () => {
       yields: "int: The next fibonacci number in the sequence.",
       style: "google"
     });
-    const result = toSourceText([_$createComponent(py.FunctionDeclaration, {
-      name: "fibonacci_generator",
-      doc: generatorDoc,
-      children: "yield 0"
-    })]);
-    expect(result).toRenderTo(d`
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.FunctionDeclaration, {
+          name: "fibonacci_generator",
+          doc: generatorDoc,
+          children: "yield 0"
+        });
+      }
+    })).toRenderTo(`
         def fibonacci_generator():
             """
             A generator function that yields fibonacci numbers. This is an efficient way
@@ -1120,7 +1155,6 @@ describe("Full example", () => {
             """
             yield 0
 
-
         `);
   });
   it("ExceptionDoc with ClassDeclaration integration", () => {
@@ -1141,25 +1175,28 @@ describe("Full example", () => {
       }],
       style: "google"
     });
-    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"
-            }), _$createComponent(py.VariableDeclaration, {
-              name: "error_code",
-              type: "int"
-            })];
+            return _$createComponent(py.StatementList, {
+              get children() {
+                return [_$createComponent(py.VariableDeclaration, {
+                  name: "field_name",
+                  type: "str"
+                }), _$createComponent(py.VariableDeclaration, {
+                  name: "error_code",
+                  type: "int"
+                })];
+              }
+            });
           }
         });
       }
-    })]);
-    expect(result).toRenderTo(d`
+    })).toRenderTo(`
         class ValidationError(Exception):
             """
             Custom exception raised when data validation fails. This exception includes
@@ -1174,7 +1211,6 @@ describe("Full example", () => {
             field_name: str = None
             error_code: int = None
 
-
         `);
   });
   it("PropertyDoc with FunctionDeclaration (as property method) integration", () => {
@@ -1186,18 +1222,21 @@ describe("Full example", () => {
       },
       style: "google"
     });
-    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\""
+            });
+          }
         });
       }
-    })]);
-    expect(result).toRenderTo(`
+    })).toRenderTo(`
       class Person:
           @property
           def full_name(self) -> str:
@@ -1232,26 +1271,29 @@ describe("Full example", () => {
       note: "This method modifies the internal validation state.",
       style: "google"
     });
-    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)"
+            });
+          }
         });
       }
-    })]);
-    expect(result).toRenderTo(d`
+    })).toRenderTo(`
         class DataValidator:
             def validate(self, data: dict, strict: bool = True) -> bool:
                 """
@@ -1276,7 +1318,6 @@ describe("Full example", () => {
                 return self.validate(data, strict)
 
 
-
         `);
   });
 });