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

package/dist/test/functioncallexpressions.test.js.map

@@ -1 +1 @@
-{"version":3,"names":["refkey","d","describe","expect","it","py","toSourceText","result","_$createComponent","FunctionCallExpression","target","toRenderTo","args","methodRef","StatementList","children","FunctionDeclaration","name","Atom","jsValue","expected","returnType","parameters","type","VariableDeclaration","initializer","callStatementVar"],"sources":["../../test/functioncallexpressions.test.tsx"],"sourcesContent":[null],"mappings":";AAAA,SAASA,MAAM,QAAQ,gBAAgB;AACvC,SAASC,CAAC,QAAQ,wBAAwB;AAC1C,SAASC,QAAQ,EAAEC,MAAM,EAAEC,EAAE,QAAQ,QAAQ;AAC7C,OAAO,KAAKC,EAAE,MAAM,iBAAiB;AACrC,SAASC,YAAY;AAErBJ,QAAQ,CAAC,wBAAwB,EAAE,MAAM;EACvCE,EAAE,CAAC,SAAS,EAAE,MAAM;IAClB,MAAMG,MAAM,GAAGD,YAAY,CAAC,CAAAE,iBAAA,CAAEH,EAAE,CAACI,sBAAsB;MAACC,MAAM;IAAA,GAAU,CAAC;IACzEP,MAAM,CAACI,MAAM,CAAC,CAACI,UAAU,CAACV,CAAC;AAC/B;AACA,KAAK,CAAC;EACJ,CAAC,CAAC;EACFG,EAAE,CAAC,mBAAmB,EAAE,MAAM;IAC5B,MAAMG,MAAM,GAAGD,YAAY,CAAC,CAAAE,iBAAA,CACzBH,EAAE,CAACI,sBAAsB;MAACC,MAAM;MAAOE,IAAI,EAAE,CAAC,GAAG,EAAE,GAAG;IAAC,GACzD,CAAC;IACFT,MAAM,CAACI,MAAM,CAAC,CAACI,UAAU,CAACV,CAAC;AAC/B;AACA,KAAK,CAAC;EACJ,CAAC,CAAC;EAEFG,EAAE,CAAC,8BAA8B,EAAE,MAAM;IACvC;IACA,MAAMS,SAAS,GAAGb,MAAM,CAAC,CAAC;IAC1B,MAAMO,MAAM,GAAGD,YAAY,CAAC,CAAAE,iBAAA,CACzBH,EAAE,CAACS,aAAa;MAAA,IAAAC,SAAA;QAAA,QAAAP,iBAAA,CACdH,EAAE,CAACW,mBAAmB;UAACC,IAAI;UAAWjB,MAAM,EAAEa;QAAS,IAAAL,iBAAA,CACvDH,EAAE,CAACI,sBAAsB;UACxBC,MAAM,EAAEG,SAAS;UAAA,IACjBD,IAAIA,CAAA;YAAA,OAAE,CAAAJ,iBAAA,CACHH,EAAE,CAACa,IAAI;cAACC,OAAO,EAAE;YAAQ,IAAAX,iBAAA,CACzBH,EAAE,CAACa,IAAI;cAACC,OAAO,EAAE;YAAE,IAAAX,iBAAA,CACnBH,EAAE,CAACa,IAAI;cAACC,OAAO,EAAE;YAAI,GACvB;UAAA;QAAA;MAAA;IAAA,GAGN,CAAC;IACF,MAAMC,QAAQ,GAAGnB,CAAC;AACtB;AACA;AACA;AACA;AACA,KAAK;IACDE,MAAM,CAACI,MAAM,CAAC,CAACI,UAAU,CAACS,QAAQ,CAAC;EACrC,CAAC,CAAC;EAEFhB,EAAE,CAAC,6CAA6C,EAAE,MAAM;IACtD;IACA,MAAMS,SAAS,GAAGb,MAAM,CAAC,CAAC;IAC1B,MAAMO,MAAM,GAAGD,YAAY,CAAC,CAAAE,iBAAA,CACzBH,EAAE,CAACS,aAAa;MAAA,IAAAC,SAAA;QAAA,QAAAP,iBAAA,CACdH,EAAE,CAACW,mBAAmB;UACrBC,IAAI;UACJI,UAAU;UACVrB,MAAM,EAAEa,SAAS;UACjBS,UAAU,EAAE,CACV;YAAEL,IAAI,EAAE,MAAM;YAAEM,IAAI,EAAE;UAAM,CAAC,EAC7B;YAAEN,IAAI,EAAE,QAAQ;YAAEM,IAAI,EAAE;UAAM,CAAC,EAC/B;YAAEN,IAAI,EAAE,MAAM;YAAEM,IAAI,EAAE;UAAO,CAAC;QAC/B,IAAAf,iBAAA,CAEFH,EAAE,CAACmB,mBAAmB;UACrBP,IAAI;UACJM,IAAI;UAAA,IACJE,WAAWA,CAAA;YAAA,OAAAjB,iBAAA,CACRH,EAAE,CAACI,sBAAsB;cACxBC,MAAM,EAAEG,SAAS;cAAA,IACjBD,IAAIA,CAAA;gBAAA,OAAE,CAAAJ,iBAAA,CACHH,EAAE,CAACa,IAAI;kBAACC,OAAO,EAAE;gBAAQ,IAAAX,iBAAA,CACzBH,EAAE,CAACa,IAAI;kBAACC,OAAO,EAAE;gBAAE,IAAAX,iBAAA,CACnBH,EAAE,CAACa,IAAI;kBAACC,OAAO,EAAE;gBAAI,GACvB;cAAA;YAAA;UAAA;QAAA;MAAA;IAAA,GAKV,CAAC;IACF,MAAMC,QAAQ,GAAGnB,CAAC;AACtB;AACA;AACA;AACA;AACA,KAAK;IACDE,MAAM,CAACI,MAAM,CAAC,CAACI,UAAU,CAACS,QAAQ,CAAC;EACrC,CAAC,CAAC;EAEFhB,EAAE,CAAC,8DAA8D,EAAE,MAAM;IACvE,MAAMG,MAAM,GAAGD,YAAY,CAAC,CAAAE,iBAAA,CACzBH,EAAE,CAACS,aAAa;MAAA,IAAAC,SAAA;QAAA,OAAAP,iBAAA,CACdH,EAAE,CAACI,sBAAsB;UACxBC,MAAM,EAAE,gBAAgB;UAAA,IACxBE,IAAIA,CAAA;YAAA,OAAE,CAAAJ,iBAAA,CACHH,EAAE,CAACmB,mBAAmB;cACrBP,IAAI;cACJQ,WAAW,EAAE,QAAQ;cACrBC,gBAAgB;YAAA,IAAAlB,iBAAA,CAEjBH,EAAE,CAACmB,mBAAmB;cACrBP,IAAI;cACJQ,WAAW,EAAE,EAAE;cACfC,gBAAgB;YAAA,IAAAlB,iBAAA,CAEjBH,EAAE,CAACmB,mBAAmB;cACrBP,IAAI;cACJQ,WAAW,EAAE,IAAI;cACjBC,gBAAgB;YAAA,GAEnB;UAAA;QAAA;MAAA;IAAA,GAGN,CAAC;IACF,MAAMN,QAAQ,GAAGnB,CAAC;AACtB;AACA,KAAK;IACDE,MAAM,CAACI,MAAM,CAAC,CAACI,UAAU,CAACS,QAAQ,CAAC;EACrC,CAAC,CAAC;EAEFhB,EAAE,CAAC,+DAA+D,EAAE,MAAM;IACxE,MAAMG,MAAM,GAAGD,YAAY,CAAC,CAAAE,iBAAA,CACzBH,EAAE,CAACS,aAAa;MAAA,IAAAC,SAAA;QAAA,OAAAP,iBAAA,CACdH,EAAE,CAACI,sBAAsB;UACxBC,MAAM,EAAE,gBAAgB;UAAA,IACxBE,IAAIA,CAAA;YAAA,OAAE,CAAAJ,iBAAA,CACHH,EAAE,CAACa,IAAI;cAACC,OAAO,EAAE;YAAQ,IAAAX,iBAAA,CACzBH,EAAE,CAACmB,mBAAmB;cACrBP,IAAI;cACJQ,WAAW,EAAE,EAAE;cACfC,gBAAgB;YAAA,IAAAlB,iBAAA,CAEjBH,EAAE,CAACmB,mBAAmB;cACrBP,IAAI;cACJQ,WAAW,EAAE,IAAI;cACjBC,gBAAgB;YAAA,GAEnB;UAAA;QAAA;MAAA;IAAA,GAGN,CAAC;IACF,MAAMN,QAAQ,GAAGnB,CAAC;AACtB;AACA,KAAK;IACDE,MAAM,CAACI,MAAM,CAAC,CAACI,UAAU,CAACS,QAAQ,CAAC;EACrC,CAAC,CAAC;AACJ,CAAC,CAAC","ignoreList":[]}
+{"version":3,"names":["refkey","describe","expect","it","py","TestOutput","_$createComponent","children","FunctionCallExpression","target","toRenderTo","args","methodRef","StatementList","FunctionDeclaration","name","Atom","jsValue","returnType","parameters","type","VariableDeclaration","initializer","callStatementVar"],"sources":["../../test/functioncallexpressions.test.tsx"],"sourcesContent":[null],"mappings":";AAAA,SAASA,MAAM,QAAQ,gBAAgB;AACvC,SAASC,QAAQ,EAAEC,MAAM,EAAEC,EAAE,QAAQ,QAAQ;AAC7C,OAAO,KAAKC,EAAE,MAAM,iBAAiB;AACrC,SAASC,UAAU,QAAQ,YAAY;AAEvCJ,QAAQ,CAAC,wBAAwB,EAAE,MAAM;EACvCE,EAAE,CAAC,SAAS,EAAE,MAAM;IAClBD,MAAM,CAAAI,iBAAA,CACHD,UAAU;MAAA,IAAAE,SAAA;QAAA,OAAAD,iBAAA,CACRF,EAAE,CAACI,sBAAsB;UAACC,MAAM;QAAA;MAAA;IAAA,EAErC,CAAC,CAACC,UAAU,CACV;AACN;AACA,KACI,CAAC;EACH,CAAC,CAAC;EAEFP,EAAE,CAAC,mBAAmB,EAAE,MAAM;IAC5BD,MAAM,CAAAI,iBAAA,CACHD,UAAU;MAAA,IAAAE,SAAA;QAAA,OAAAD,iBAAA,CACRF,EAAE,CAACI,sBAAsB;UAACC,MAAM;UAAOE,IAAI,EAAE,CAAC,GAAG,EAAE,GAAG;QAAC;MAAA;IAAA,EAE5D,CAAC,CAACD,UAAU,CACV;AACN;AACA,KACI,CAAC;EACH,CAAC,CAAC;EAEFP,EAAE,CAAC,8BAA8B,EAAE,MAAM;IACvC,MAAMS,SAAS,GAAGZ,MAAM,CAAC,CAAC;IAC1BE,MAAM,CAAAI,iBAAA,CACHD,UAAU;MAAA,IAAAE,SAAA;QAAA,OAAAD,iBAAA,CACRF,EAAE,CAACS,aAAa;UAAA,IAAAN,SAAA;YAAA,QAAAD,iBAAA,CACdF,EAAE,CAACU,mBAAmB;cAACC,IAAI;cAAWf,MAAM,EAAEY;YAAS,IAAAN,iBAAA,CACvDF,EAAE,CAACI,sBAAsB;cACxBC,MAAM,EAAEG,SAAS;cAAA,IACjBD,IAAIA,CAAA;gBAAA,OAAE,CAAAL,iBAAA,CACHF,EAAE,CAACY,IAAI;kBAACC,OAAO,EAAE;gBAAQ,IAAAX,iBAAA,CACzBF,EAAE,CAACY,IAAI;kBAACC,OAAO,EAAE;gBAAE,IAAAX,iBAAA,CACnBF,EAAE,CAACY,IAAI;kBAACC,OAAO,EAAE;gBAAI,GACvB;cAAA;YAAA;UAAA;QAAA;MAAA;IAAA,EAIT,CAAC,CAACP,UAAU,CACV;AACN;AACA;AACA;AACA;AACA,KACI,CAAC;EACH,CAAC,CAAC;EAEFP,EAAE,CAAC,6CAA6C,EAAE,MAAM;IACtD,MAAMS,SAAS,GAAGZ,MAAM,CAAC,CAAC;IAC1BE,MAAM,CAAAI,iBAAA,CACHD,UAAU;MAAA,IAAAE,SAAA;QAAA,OAAAD,iBAAA,CACRF,EAAE,CAACS,aAAa;UAAA,IAAAN,SAAA;YAAA,QAAAD,iBAAA,CACdF,EAAE,CAACU,mBAAmB;cACrBC,IAAI;cACJG,UAAU;cACVlB,MAAM,EAAEY,SAAS;cACjBO,UAAU,EAAE,CACV;gBAAEJ,IAAI,EAAE,MAAM;gBAAEK,IAAI,EAAE;cAAM,CAAC,EAC7B;gBAAEL,IAAI,EAAE,QAAQ;gBAAEK,IAAI,EAAE;cAAM,CAAC,EAC/B;gBAAEL,IAAI,EAAE,MAAM;gBAAEK,IAAI,EAAE;cAAO,CAAC;YAC/B,IAAAd,iBAAA,CAEFF,EAAE,CAACiB,mBAAmB;cACrBN,IAAI;cACJK,IAAI;cAAA,IACJE,WAAWA,CAAA;gBAAA,OAAAhB,iBAAA,CACRF,EAAE,CAACI,sBAAsB;kBACxBC,MAAM,EAAEG,SAAS;kBAAA,IACjBD,IAAIA,CAAA;oBAAA,OAAE,CAAAL,iBAAA,CACHF,EAAE,CAACY,IAAI;sBAACC,OAAO,EAAE;oBAAQ,IAAAX,iBAAA,CACzBF,EAAE,CAACY,IAAI;sBAACC,OAAO,EAAE;oBAAE,IAAAX,iBAAA,CACnBF,EAAE,CAACY,IAAI;sBAACC,OAAO,EAAE;oBAAI,GACvB;kBAAA;gBAAA;cAAA;YAAA;UAAA;QAAA;MAAA;IAAA,EAMb,CAAC,CAACP,UAAU,CACV;AACN;AACA;AACA;AACA;AACA,KACI,CAAC;EACH,CAAC,CAAC;EAEFP,EAAE,CAAC,8DAA8D,EAAE,MAAM;IACvED,MAAM,CAAAI,iBAAA,CACHD,UAAU;MAAA,IAAAE,SAAA;QAAA,OAAAD,iBAAA,CACRF,EAAE,CAACS,aAAa;UAAA,IAAAN,SAAA;YAAA,OAAAD,iBAAA,CACdF,EAAE,CAACI,sBAAsB;cACxBC,MAAM,EAAE,gBAAgB;cAAA,IACxBE,IAAIA,CAAA;gBAAA,OAAE,CAAAL,iBAAA,CACHF,EAAE,CAACiB,mBAAmB;kBACrBN,IAAI;kBACJO,WAAW,EAAE,QAAQ;kBACrBC,gBAAgB;gBAAA,IAAAjB,iBAAA,CAEjBF,EAAE,CAACiB,mBAAmB;kBACrBN,IAAI;kBACJO,WAAW,EAAE,EAAE;kBACfC,gBAAgB;gBAAA,IAAAjB,iBAAA,CAEjBF,EAAE,CAACiB,mBAAmB;kBACrBN,IAAI;kBACJO,WAAW,EAAE,IAAI;kBACjBC,gBAAgB;gBAAA,GAEnB;cAAA;YAAA;UAAA;QAAA;MAAA;IAAA,EAIT,CAAC,CAACb,UAAU,CACV;AACN;AACA,KACI,CAAC;EACH,CAAC,CAAC;EAEFP,EAAE,CAAC,+DAA+D,EAAE,MAAM;IACxED,MAAM,CAAAI,iBAAA,CACHD,UAAU;MAAA,IAAAE,SAAA;QAAA,OAAAD,iBAAA,CACRF,EAAE,CAACS,aAAa;UAAA,IAAAN,SAAA;YAAA,OAAAD,iBAAA,CACdF,EAAE,CAACI,sBAAsB;cACxBC,MAAM,EAAE,gBAAgB;cAAA,IACxBE,IAAIA,CAAA;gBAAA,OAAE,CAAAL,iBAAA,CACHF,EAAE,CAACY,IAAI;kBAACC,OAAO,EAAE;gBAAQ,IAAAX,iBAAA,CACzBF,EAAE,CAACiB,mBAAmB;kBACrBN,IAAI;kBACJO,WAAW,EAAE,EAAE;kBACfC,gBAAgB;gBAAA,IAAAjB,iBAAA,CAEjBF,EAAE,CAACiB,mBAAmB;kBACrBN,IAAI;kBACJO,WAAW,EAAE,IAAI;kBACjBC,gBAAgB;gBAAA,GAEnB;cAAA;YAAA;UAAA;QAAA;MAAA;IAAA,EAIT,CAAC,CAACb,UAAU,CACV;AACN;AACA,KACI,CAAC;EACH,CAAC,CAAC;AACJ,CAAC,CAAC","ignoreList":[]}

package/dist/test/functiondeclaration.test.js

@@ -1,183 +1,208 @@
 import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
-import { code, namekey, refkey } from "@alloy-js/core";
-import { d } from "@alloy-js/core/testing";
+import { code, namekey, refkey, render } from "@alloy-js/core";
 import { describe, expect, it } from "vitest";
 import * as py from "../src/index.js";
 import { abcModule } from "../src/index.js";
-import { assertFileContents, toSourceText, toSourceTextMultiple } from "./utils.js";
+import { TestOutput, TestOutputDirectory } from "./utils.js";
 describe("Function Declaration", () => {
   it("renders multiple decorators above def without blank lines", () => {
-    const result = toSourceText([_$createComponent(py.FunctionDeclaration, {
-      name: "f",
-      decorators: ["@a", "@b", "@c"],
-      children: "pass"
-    })]);
-    expect(result).toRenderTo(d`
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.FunctionDeclaration, {
+          name: "f",
+          decorators: ["@a", "@b", "@c"],
+          children: "pass"
+        });
+      }
+    })).toRenderTo(`
       @a
       @b
       @c
       def f():
           pass
 
-        
     `);
   });
   it("renders a function with no body as 'pass'", () => {
-    const result = toSourceText([_$createComponent(py.FunctionDeclaration, {
-      name: "foo"
-    })]);
-    expect(result).toRenderTo(d`
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.FunctionDeclaration, {
+          name: "foo"
+        });
+      }
+    })).toRenderTo(`
       def foo():
           pass
 
-        
     `);
   });
   it("takes a namekey", () => {
-    const result = toSourceText([_$createComponent(py.FunctionDeclaration, {
-      get name() {
-        return namekey("foo-bar");
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.FunctionDeclaration, {
+          get name() {
+            return namekey("foo-bar");
+          }
+        });
       }
-    })]);
-    expect(result).toRenderTo(d`
+    })).toRenderTo(`
       def foo_bar():
           pass
 
-
     `);
   });
   it("renders a function with no body as 'pass' with return type", () => {
-    const result = toSourceText([_$createComponent(py.FunctionDeclaration, {
-      name: "foo",
-      returnType: "int"
-    })]);
-    expect(result).toRenderTo(d`
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.FunctionDeclaration, {
+          name: "foo",
+          returnType: "int"
+        });
+      }
+    })).toRenderTo(`
       def foo() -> int:
           pass
 
-
     `);
   });
   it("renders a function that calls another function", () => {
     const refkeyFoo = refkey();
-    const result = toSourceText([_$createComponent(py.StatementList, {
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return [_$createComponent(py.FunctionDeclaration, {
-          name: "foo",
-          returnType: "int",
-          refkey: refkeyFoo
-        }), _$createComponent(py.FunctionDeclaration, {
-          name: "bar",
-          returnType: "int",
+        return _$createComponent(py.StatementList, {
           get children() {
-            return _$createComponent(py.VariableDeclaration, {
-              name: "result",
-              type: "int",
-              get initializer() {
-                return _$createComponent(py.FunctionCallExpression, {
-                  target: refkeyFoo,
-                  args: []
+            return [_$createComponent(py.FunctionDeclaration, {
+              name: "foo",
+              returnType: "int",
+              refkey: refkeyFoo
+            }), _$createComponent(py.FunctionDeclaration, {
+              name: "bar",
+              returnType: "int",
+              get children() {
+                return _$createComponent(py.VariableDeclaration, {
+                  name: "result",
+                  type: "int",
+                  get initializer() {
+                    return _$createComponent(py.FunctionCallExpression, {
+                      target: refkeyFoo,
+                      args: []
+                    });
+                  }
                 });
               }
-            });
+            })];
           }
-        })];
+        });
       }
-    })]);
-    expect(result).toRenderTo(d`
+    })).toRenderTo(`
       def foo() -> int:
           pass
 
       def bar() -> int:
           result: int = foo()
 
-
     `);
   });
   it("renders a function with parameters", () => {
-    const result = toSourceText([_$createComponent(py.FunctionDeclaration, {
-      name: "baz",
-      parameters: [{
-        name: "x",
-        type: "int"
-      }, {
-        name: "y",
-        default: 0
-      }, {
-        name: "z",
-        type: "int",
-        default: 42
-      }],
-      args: true,
-      kwargs: true,
-      children: "print(x, y)"
-    })]);
-    expect(result).toRenderTo(d`
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.FunctionDeclaration, {
+          name: "baz",
+          parameters: [{
+            name: "x",
+            type: "int"
+          }, {
+            name: "y",
+            default: 0
+          }, {
+            name: "z",
+            type: "int",
+            default: 42
+          }],
+          args: true,
+          kwargs: true,
+          children: "print(x, y)"
+        });
+      }
+    })).toRenderTo(`
         def baz(x: int, y=0, z: int = 42, *args, **kwargs):
             print(x, y)
 
-
       `);
   });
   it("renders an __init__ function with no body as 'pass'", () => {
-    const result = toSourceText([_$createComponent(py.ClassDeclaration, {
-      name: "MyClass",
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return _$createComponent(py.DunderMethodDeclaration, {
-          name: "__init__",
-          parameters: [{
-            name: "x"
-          }]
+        return _$createComponent(py.ClassDeclaration, {
+          name: "MyClass",
+          get children() {
+            return _$createComponent(py.DunderMethodDeclaration, {
+              name: "__init__",
+              parameters: [{
+                name: "x"
+              }]
+            });
+          }
         });
       }
-    })]);
-    expect(result).toRenderTo(d`
+    })).toRenderTo(`
       class MyClass:
           def __init__(self, x):
               pass
 
-      
-      
+
     `);
   });
   it("can be an async function", () => {
-    expect(toSourceText([_$createComponent(py.FunctionDeclaration, {
-      async: true,
-      name: "foo"
-    })])).toBe(d`
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.FunctionDeclaration, {
+          async: true,
+          name: "foo"
+        });
+      }
+    })).toRenderTo(`
       async def foo():
           pass
 
     `);
   });
   it("can be an async function with returnType", () => {
-    expect(toSourceText([_$createComponent(py.FunctionDeclaration, {
-      async: true,
-      name: "foo",
-      returnType: "Foo"
-    })])).toBe(d`
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.FunctionDeclaration, {
+          async: true,
+          name: "foo",
+          returnType: "Foo"
+        });
+      }
+    })).toRenderTo(`
       async def foo() -> Foo:
           pass
 
     `);
   });
   it("can be an async function with returnType element with Reference", () => {
-    expect(toSourceText([_$createComponent(py.StatementList, {
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return [_$createComponent(py.ClassDeclaration, {
-          name: "Foo",
-          get refkey() {
-            return refkey("Foo");
-          }
-        }), _$createComponent(py.FunctionDeclaration, {
-          async: true,
-          name: "foo",
-          get returnType() {
-            return refkey("Foo");
+        return _$createComponent(py.StatementList, {
+          get children() {
+            return [_$createComponent(py.ClassDeclaration, {
+              name: "Foo",
+              get refkey() {
+                return refkey("Foo");
+              }
+            }), _$createComponent(py.FunctionDeclaration, {
+              async: true,
+              name: "foo",
+              get returnType() {
+                return refkey("Foo");
+              }
+            })];
           }
-        })];
+        });
       }
-    })])).toBe(d`
+    })).toRenderTo(`
       class Foo:
           pass
 
@@ -187,22 +212,26 @@ describe("Function Declaration", () => {
     `);
   });
   it("can be an async function with returnType element with list of References", () => {
-    expect(toSourceText([_$createComponent(py.StatementList, {
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return [_$createComponent(py.ClassDeclaration, {
-          name: "Foo",
-          get refkey() {
-            return refkey("Foo");
-          }
-        }), _$createComponent(py.FunctionDeclaration, {
-          async: true,
-          name: "foo",
-          get returnType() {
-            return code`list[${refkey("Foo")}]`;
+        return _$createComponent(py.StatementList, {
+          get children() {
+            return [_$createComponent(py.ClassDeclaration, {
+              name: "Foo",
+              get refkey() {
+                return refkey("Foo");
+              }
+            }), _$createComponent(py.FunctionDeclaration, {
+              async: true,
+              name: "foo",
+              get returnType() {
+                return code`list[${refkey("Foo")}]`;
+              }
+            })];
           }
-        })];
+        });
       }
-    })])).toBe(d`
+    })).toRenderTo(`
       class Foo:
           pass
 
@@ -231,7 +260,9 @@ describe("Function Declaration", () => {
         });
       }
     });
-    expect(toSourceText([decl])).toBe(d`
+    expect(_$createComponent(TestOutput, {
+      children: decl
+    })).toRenderTo(`
       class MyClass:
           async def __aenter__(self) -> MyClass:
               return self
@@ -258,7 +289,9 @@ describe("Function Declaration", () => {
         });
       }
     });
-    expect(toSourceText([decl])).toBe(d`
+    expect(_$createComponent(TestOutput, {
+      children: decl
+    })).toRenderTo(`
       class MyClass:
           async def __new__(cls) -> MyClass:
               return super().__new__(cls)
@@ -276,7 +309,9 @@ describe("Function Declaration", () => {
       }],
       children: "return a + b"
     });
-    expect(toSourceText([decl])).toBe(d`
+    expect(_$createComponent(TestOutput, {
+      children: decl
+    })).toRenderTo(`
       def foo(a, b):
           return a + b
 
@@ -293,7 +328,9 @@ describe("Function Declaration", () => {
       typeParameters: ["T", "U"],
       children: "return a + b"
     });
-    expect(toSourceText([decl])).toBe(d`
+    expect(_$createComponent(TestOutput, {
+      children: decl
+    })).toRenderTo(`
       def foo[T, U](a, b):
           return a + b
 
@@ -330,9 +367,10 @@ describe("Function Declaration", () => {
         });
       }
     });
-    expect(toSourceText([decl], {
-      externals: [abcModule]
-    })).toBe(d`
+    expect(_$createComponent(TestOutput, {
+      externals: [abcModule],
+      children: decl
+    })).toRenderTo(`
       from abc import abstractmethod
 
 
@@ -377,7 +415,9 @@ describe("Function Declaration", () => {
         });
       }
     });
-    expect(toSourceText([decl])).toBe(d`
+    expect(_$createComponent(TestOutput, {
+      children: decl
+    })).toRenderTo(`
       class MyClass:
           def __init__(self, x: int):
               self.attribute = "value"
@@ -403,7 +443,9 @@ describe("Function Declaration", () => {
         });
       }
     });
-    expect(toSourceText([decl])).toBe(d`
+    expect(_$createComponent(TestOutput, {
+      children: decl
+    })).toRenderTo(`
       class MyClass:
           def __new__(cls, *args, **kwargs):
               pass
@@ -461,7 +503,9 @@ describe("Function Declaration", () => {
         })];
       }
     });
-    expect(toSourceText([decl])).toBe(d`
+    expect(_$createComponent(TestOutput, {
+      children: decl
+    })).toRenderTo(`
       def foo(x: int):
           def bar(y: int):
               def foobar(z: int):
@@ -472,55 +516,58 @@ describe("Function Declaration", () => {
     `);
   });
   it("renders complex typing structure", () => {
-    const res = toSourceTextMultiple([_$createComponent(py.SourceFile, {
-      path: "mod1.py",
+    expect(_$createComponent(TestOutputDirectory, {
       get children() {
-        return _$createComponent(py.ClassDeclaration, {
-          name: "Foo",
-          get refkey() {
-            return refkey("Foo");
+        return [_$createComponent(py.SourceFile, {
+          path: "mod1.py",
+          get children() {
+            return _$createComponent(py.ClassDeclaration, {
+              name: "Foo",
+              get refkey() {
+                return refkey("Foo");
+              }
+            });
           }
-        });
-      }
-    }), _$createComponent(py.SourceFile, {
-      path: "mod2.py",
-      get children() {
-        return [_$createComponent(py.ClassDeclaration, {
-          name: "A",
-          get refkey() {
-            return refkey("A");
+        }), _$createComponent(py.SourceFile, {
+          path: "mod2.py",
+          get children() {
+            return [_$createComponent(py.ClassDeclaration, {
+              name: "A",
+              get refkey() {
+                return refkey("A");
+              }
+            }), _$createComponent(py.ClassDeclaration, {
+              name: "B",
+              get refkey() {
+                return refkey("B");
+              }
+            })];
           }
-        }), _$createComponent(py.ClassDeclaration, {
-          name: "B",
-          get refkey() {
-            return refkey("B");
+        }), _$createComponent(py.SourceFile, {
+          path: "usage.py",
+          get children() {
+            return _$createComponent(py.FunctionDeclaration, {
+              async: true,
+              name: "foo",
+              get parameters() {
+                return [{
+                  name: "x",
+                  type: refkey("A")
+                }, {
+                  name: "y",
+                  type: refkey("B")
+                }];
+              },
+              args: true,
+              kwargs: true,
+              get returnType() {
+                return refkey("Foo");
+              }
+            });
           }
         })];
       }
-    }), _$createComponent(py.SourceFile, {
-      path: "usage.py",
-      get children() {
-        return _$createComponent(py.FunctionDeclaration, {
-          async: true,
-          name: "foo",
-          get parameters() {
-            return [{
-              name: "x",
-              type: refkey("A")
-            }, {
-              name: "y",
-              type: refkey("B")
-            }];
-          },
-          args: true,
-          kwargs: true,
-          get returnType() {
-            return refkey("Foo");
-          }
-        });
-      }
-    })]);
-    assertFileContents(res, {
+    })).toRenderTo({
       "mod1.py": `
             class Foo:
                 pass
@@ -552,37 +599,57 @@ describe("Function Declaration", () => {
   });
   it("throws error when PropertyDeclaration is used outside of a class", () => {
     expect(() => {
-      toSourceText([_$createComponent(py.PropertyDeclaration, {
-        name: "x"
-      })]);
+      render(_$createComponent(TestOutput, {
+        get children() {
+          return _$createComponent(py.PropertyDeclaration, {
+            name: "x"
+          });
+        }
+      }));
     }).toThrow('Method "x" must be declared inside a class (member scope)');
   });
   it("throws error when MethodDeclaration is used outside of a class", () => {
     expect(() => {
-      toSourceText([_$createComponent(py.MethodDeclaration, {
-        name: "my_method"
-      })]);
+      render(_$createComponent(TestOutput, {
+        get children() {
+          return _$createComponent(py.MethodDeclaration, {
+            name: "my_method"
+          });
+        }
+      }));
     }).toThrow('Method "my_method" must be declared inside a class (member scope)');
   });
   it("throws error when ClassMethodDeclaration is used outside of a class", () => {
     expect(() => {
-      toSourceText([_$createComponent(py.ClassMethodDeclaration, {
-        name: "my_class_method"
-      })]);
+      render(_$createComponent(TestOutput, {
+        get children() {
+          return _$createComponent(py.ClassMethodDeclaration, {
+            name: "my_class_method"
+          });
+        }
+      }));
     }).toThrow('Method "my_class_method" must be declared inside a class (member scope)');
   });
   it("throws error when StaticMethodDeclaration is used outside of a class", () => {
     expect(() => {
-      toSourceText([_$createComponent(py.StaticMethodDeclaration, {
-        name: "my_static_method"
-      })]);
+      render(_$createComponent(TestOutput, {
+        get children() {
+          return _$createComponent(py.StaticMethodDeclaration, {
+            name: "my_static_method"
+          });
+        }
+      }));
     }).toThrow('Method "my_static_method" must be declared inside a class (member scope)');
   });
   it("throws error when DunderMethodDeclaration is used outside of a class", () => {
     expect(() => {
-      toSourceText([_$createComponent(py.DunderMethodDeclaration, {
-        name: "__init__"
-      })]);
+      render(_$createComponent(TestOutput, {
+        get children() {
+          return _$createComponent(py.DunderMethodDeclaration, {
+            name: "__init__"
+          });
+        }
+      }));
     }).toThrow('Method "__init__" must be declared inside a class (member scope)');
   });
 });