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

package/dist/test/memberexpressions.test.js

@@ -1,125 +1,148 @@
 import { createComponent as _$createComponent, memo as _$memo, createIntrinsic as _$createIntrinsic } from "@alloy-js/core/jsx-runtime";
-import { code, memberRefkey, refkey } from "@alloy-js/core";
-import { d } from "@alloy-js/core/testing";
+import { code, memberRefkey, refkey, render } from "@alloy-js/core";
 import { describe, expect, it } from "vitest";
 import * as py from "../src/index.js";
 import { ClassDeclaration, FunctionDeclaration, SourceFile, VariableDeclaration } from "../src/index.js";
-import { assertFileContents, toSourceText, toSourceTextMultiple } from "./utils.js";
+import { TestOutput, TestOutputDirectory } from "./utils.js";
 it("renders basic member expression with dot notation", () => {
-  expect(toSourceText([_$createComponent(py.MemberExpression, {
+  expect(_$createComponent(TestOutput, {
     get children() {
-      return [_$createComponent(py.MemberExpression.Part, {
-        id: "obj"
-      }), _$createComponent(py.MemberExpression.Part, {
-        id: "property"
-      })];
+      return _$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "obj"
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "property"
+          })];
+        }
+      });
     }
-  })])).toBe(d`
+  })).toRenderTo(`
     obj.property
   `);
 });
 it("renders basic member expression with key", () => {
-  expect(toSourceText([_$createComponent(py.MemberExpression, {
+  expect(_$createComponent(TestOutput, {
     get children() {
-      return [_$createComponent(py.MemberExpression.Part, {
-        id: "arr"
-      }), _$createComponent(py.MemberExpression.Part, {
-        key: 12345
-      }), _$createComponent(py.MemberExpression.Part, {
-        id: "foo-bar"
-      })];
+      return _$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "arr"
+          }), _$createComponent(py.MemberExpression.Part, {
+            key: 12345
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "foo-bar"
+          })];
+        }
+      });
     }
-  })])).toBe(d`
+  })).toRenderTo(`
     arr[12345].foo-bar
   `);
 });
 it("renders basic member expression with key with reference", () => {
   const rk1 = refkey();
-  expect(toSourceText([_$createComponent(py.StatementList, {
+  expect(_$createComponent(TestOutput, {
     get children() {
-      return [_$createComponent(py.VariableDeclaration, {
-        name: "test1",
-        refkey: rk1,
-        initializer: 1
-      }), _$createComponent(py.MemberExpression, {
+      return _$createComponent(py.StatementList, {
         get children() {
-          return [_$createComponent(py.MemberExpression.Part, {
-            id: "arr"
-          }), _$createComponent(py.MemberExpression.Part, {
-            key: rk1
-          }), _$createComponent(py.MemberExpression.Part, {
-            id: "foo-bar"
+          return [_$createComponent(py.VariableDeclaration, {
+            name: "test1",
+            refkey: rk1,
+            initializer: 1
+          }), _$createComponent(py.MemberExpression, {
+            get children() {
+              return [_$createComponent(py.MemberExpression.Part, {
+                id: "arr"
+              }), _$createComponent(py.MemberExpression.Part, {
+                key: rk1
+              }), _$createComponent(py.MemberExpression.Part, {
+                id: "foo-bar"
+              })];
+            }
           })];
         }
-      })];
+      });
     }
-  })])).toBe(d`
+  })).toRenderTo(`
     test1 = 1
     arr[test1].foo-bar
   `);
 });
 it("renders basic member expression with keys", () => {
-  expect(toSourceText([_$createComponent(py.MemberExpression, {
+  expect(_$createComponent(TestOutput, {
     get children() {
-      return [_$createComponent(py.MemberExpression.Part, {
-        id: "arr"
-      }), _$createComponent(py.MemberExpression.Part, {
-        keys: [1, 2]
-      }), _$createComponent(py.MemberExpression.Part, {
-        id: "foo-bar"
-      })];
+      return _$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "arr"
+          }), _$createComponent(py.MemberExpression.Part, {
+            keys: [1, 2]
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "foo-bar"
+          })];
+        }
+      });
     }
-  })])).toBe(d`
+  })).toRenderTo(`
     arr[1, 2].foo-bar
   `);
 });
 it("renders basic member expression with keys with references", () => {
   const rk1 = refkey();
   const rk2 = refkey();
-  expect(toSourceText([_$createComponent(py.StatementList, {
+  expect(_$createComponent(TestOutput, {
     get children() {
-      return [_$createComponent(py.VariableDeclaration, {
-        name: "test1",
-        refkey: rk1,
-        initializer: 1
-      }), _$createComponent(py.VariableDeclaration, {
-        name: "test2",
-        refkey: rk2,
-        initializer: 2
-      }), _$createComponent(py.MemberExpression, {
+      return _$createComponent(py.StatementList, {
         get children() {
-          return [_$createComponent(py.MemberExpression.Part, {
-            id: "arr"
-          }), _$createComponent(py.MemberExpression.Part, {
-            keys: [rk1, rk2]
-          }), _$createComponent(py.MemberExpression.Part, {
-            id: "foo-bar"
+          return [_$createComponent(py.VariableDeclaration, {
+            name: "test1",
+            refkey: rk1,
+            initializer: 1
+          }), _$createComponent(py.VariableDeclaration, {
+            name: "test2",
+            refkey: rk2,
+            initializer: 2
+          }), _$createComponent(py.MemberExpression, {
+            get children() {
+              return [_$createComponent(py.MemberExpression.Part, {
+                id: "arr"
+              }), _$createComponent(py.MemberExpression.Part, {
+                keys: [rk1, rk2]
+              }), _$createComponent(py.MemberExpression.Part, {
+                id: "foo-bar"
+              })];
+            }
           })];
         }
-      })];
+      });
     }
-  })])).toBe(d`
+  })).toRenderTo(`
     test1 = 1
     test2 = 2
     arr[test1, test2].foo-bar
   `);
 });
 it("renders basic member expression with slice - 1", () => {
-  expect(toSourceText([_$createComponent(py.MemberExpression, {
+  expect(_$createComponent(TestOutput, {
     get children() {
-      return [_$createComponent(py.MemberExpression.Part, {
-        id: "arr"
-      }), _$createComponent(py.MemberExpression.Part, {
-        slice: {
-          start: 1,
-          stop: 3,
-          step: 2
+      return _$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "arr"
+          }), _$createComponent(py.MemberExpression.Part, {
+            slice: {
+              start: 1,
+              stop: 3,
+              step: 2
+            }
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "foo-bar"
+          })];
         }
-      }), _$createComponent(py.MemberExpression.Part, {
-        id: "foo-bar"
-      })];
+      });
     }
-  })])).toBe(d`
+  })).toRenderTo(`
     arr[1:3:2].foo-bar
   `);
 });
@@ -127,37 +150,41 @@ it("renders basic member expression with slice - 1 with references", () => {
   const rk1 = refkey();
   const rk2 = refkey();
   const rk3 = refkey();
-  expect(toSourceText([_$createComponent(py.StatementList, {
+  expect(_$createComponent(TestOutput, {
     get children() {
-      return [_$createComponent(py.VariableDeclaration, {
-        name: "test1",
-        refkey: rk1,
-        initializer: 1
-      }), _$createComponent(py.VariableDeclaration, {
-        name: "test2",
-        refkey: rk2,
-        initializer: 3
-      }), _$createComponent(py.VariableDeclaration, {
-        name: "test3",
-        refkey: rk3,
-        initializer: 2
-      }), _$createComponent(py.MemberExpression, {
+      return _$createComponent(py.StatementList, {
         get children() {
-          return [_$createComponent(py.MemberExpression.Part, {
-            id: "arr"
-          }), _$createComponent(py.MemberExpression.Part, {
-            slice: {
-              start: rk1,
-              stop: rk2,
-              step: rk3
+          return [_$createComponent(py.VariableDeclaration, {
+            name: "test1",
+            refkey: rk1,
+            initializer: 1
+          }), _$createComponent(py.VariableDeclaration, {
+            name: "test2",
+            refkey: rk2,
+            initializer: 3
+          }), _$createComponent(py.VariableDeclaration, {
+            name: "test3",
+            refkey: rk3,
+            initializer: 2
+          }), _$createComponent(py.MemberExpression, {
+            get children() {
+              return [_$createComponent(py.MemberExpression.Part, {
+                id: "arr"
+              }), _$createComponent(py.MemberExpression.Part, {
+                slice: {
+                  start: rk1,
+                  stop: rk2,
+                  step: rk3
+                }
+              }), _$createComponent(py.MemberExpression.Part, {
+                id: "foo-bar"
+              })];
             }
-          }), _$createComponent(py.MemberExpression.Part, {
-            id: "foo-bar"
           })];
         }
-      })];
+      });
     }
-  })])).toBe(d`
+  })).toRenderTo(`
     test1 = 1
     test2 = 3
     test3 = 2
@@ -165,229 +192,281 @@ it("renders basic member expression with slice - 1 with references", () => {
   `);
 });
 it("renders basic member expression with slice - 2", () => {
-  expect(toSourceText([_$createComponent(py.MemberExpression, {
+  expect(_$createComponent(TestOutput, {
     get children() {
-      return [_$createComponent(py.MemberExpression.Part, {
-        id: "arr"
-      }), _$createComponent(py.MemberExpression.Part, {
-        slice: {
-          stop: 3
+      return _$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "arr"
+          }), _$createComponent(py.MemberExpression.Part, {
+            slice: {
+              stop: 3
+            }
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "foo-bar"
+          })];
         }
-      }), _$createComponent(py.MemberExpression.Part, {
-        id: "foo-bar"
-      })];
+      });
     }
-  })])).toBe(d`
+  })).toRenderTo(`
     arr[:3].foo-bar
   `);
 });
 it("renders basic member expression with slice - 3", () => {
-  expect(toSourceText([_$createComponent(py.MemberExpression, {
+  expect(_$createComponent(TestOutput, {
     get children() {
-      return [_$createComponent(py.MemberExpression.Part, {
-        id: "arr"
-      }), _$createComponent(py.MemberExpression.Part, {
-        slice: {
-          step: 2
+      return _$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "arr"
+          }), _$createComponent(py.MemberExpression.Part, {
+            slice: {
+              step: 2
+            }
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "foo-bar"
+          })];
         }
-      }), _$createComponent(py.MemberExpression.Part, {
-        id: "foo-bar"
-      })];
+      });
     }
-  })])).toBe(d`
+  })).toRenderTo(`
     arr[::2].foo-bar
   `);
 });
 it("renders basic member expression with slice - 4", () => {
-  expect(toSourceText([_$createComponent(py.MemberExpression, {
+  expect(_$createComponent(TestOutput, {
     get children() {
-      return [_$createComponent(py.MemberExpression.Part, {
-        id: "arr"
-      }), _$createComponent(py.MemberExpression.Part, {
-        slice: {
-          start: 1
+      return _$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "arr"
+          }), _$createComponent(py.MemberExpression.Part, {
+            slice: {
+              start: 1
+            }
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "foo-bar"
+          })];
         }
-      }), _$createComponent(py.MemberExpression.Part, {
-        id: "foo-bar"
-      })];
+      });
     }
-  })])).toBe(d`
+  })).toRenderTo(`
     arr[1:].foo-bar
   `);
 });
 it("renders basic member expression with slice - 5", () => {
-  expect(toSourceText([_$createComponent(py.MemberExpression, {
+  expect(_$createComponent(TestOutput, {
     get children() {
-      return [_$createComponent(py.MemberExpression.Part, {
-        id: "arr"
-      }), _$createComponent(py.MemberExpression.Part, {
-        slice: {
-          start: 1,
-          step: 2
+      return _$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "arr"
+          }), _$createComponent(py.MemberExpression.Part, {
+            slice: {
+              start: 1,
+              step: 2
+            }
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "foo-bar"
+          })];
         }
-      }), _$createComponent(py.MemberExpression.Part, {
-        id: "foo-bar"
-      })];
+      });
     }
-  })])).toBe(d`
+  })).toRenderTo(`
     arr[1::2].foo-bar
   `);
 });
 it("renders basic member expression with an expression index - 1", () => {
   const xRefkey = refkey();
-  expect(toSourceText([_$createComponent(py.StatementList, {
+  expect(_$createComponent(TestOutput, {
     get children() {
-      return [_$createComponent(py.VariableDeclaration, {
-        name: "x",
-        initializer: 1,
-        refkey: xRefkey
-      }), _$createComponent(py.MemberExpression, {
+      return _$createComponent(py.StatementList, {
         get children() {
-          return [_$createComponent(py.MemberExpression.Part, {
-            id: "arr"
-          }), _$createComponent(py.MemberExpression.Part, {
-            key: code`${xRefkey} + 1`
-          }), _$createComponent(py.MemberExpression.Part, {
-            id: "foo-bar"
+          return [_$createComponent(py.VariableDeclaration, {
+            name: "x",
+            initializer: 1,
+            refkey: xRefkey
+          }), _$createComponent(py.MemberExpression, {
+            get children() {
+              return [_$createComponent(py.MemberExpression.Part, {
+                id: "arr"
+              }), _$createComponent(py.MemberExpression.Part, {
+                key: code`${xRefkey} + 1`
+              }), _$createComponent(py.MemberExpression.Part, {
+                id: "foo-bar"
+              })];
+            }
           })];
         }
-      })];
+      });
     }
-  })])).toBe(d`
+  })).toRenderTo(`
     x = 1
     arr[x + 1].foo-bar
   `);
 });
 it("renders basic member expression with an expression index - 2", () => {
-  expect(toSourceText([_$createComponent(py.StatementList, {
+  expect(_$createComponent(TestOutput, {
     get children() {
-      return _$createComponent(py.MemberExpression, {
+      return _$createComponent(py.StatementList, {
         get children() {
-          return [_$createComponent(py.MemberExpression.Part, {
-            id: "arr"
-          }), _$createComponent(py.MemberExpression.Part, {
-            key: code`"foo" + 1`
-          }), _$createComponent(py.MemberExpression.Part, {
-            id: "foo-bar"
-          })];
+          return _$createComponent(py.MemberExpression, {
+            get children() {
+              return [_$createComponent(py.MemberExpression.Part, {
+                id: "arr"
+              }), _$createComponent(py.MemberExpression.Part, {
+                key: code`"foo" + 1`
+              }), _$createComponent(py.MemberExpression.Part, {
+                id: "foo-bar"
+              })];
+            }
+          });
         }
       });
     }
-  })])).toBe(d`
+  })).toRenderTo(`
     arr["foo" + 1].foo-bar
   `);
 });
 it("throws an error for invalid identifiers with quotes", () => {
-  expect(() => toSourceText([_$createComponent(py.MemberExpression, {
+  expect(() => render(_$createComponent(TestOutput, {
     get children() {
-      return [_$createComponent(py.MemberExpression.Part, {
-        id: "obj"
-      }), _$createComponent(py.MemberExpression.Part, {
-        id: `property-"name"`
-      })];
+      return _$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "obj"
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: `property-"name"`
+          })];
+        }
+      });
     }
-  })])).toThrowError(/Invalid identifier: property-"name"/);
+  }))).toThrowError(/Invalid identifier: property-"name"/);
 });
 it("supports multiple levels of nesting", () => {
-  expect(toSourceText([_$createComponent(py.MemberExpression, {
+  expect(_$createComponent(TestOutput, {
     get children() {
-      return [_$createComponent(py.MemberExpression.Part, {
-        id: "a"
-      }), _$createComponent(py.MemberExpression.Part, {
-        id: "b"
-      }), _$createComponent(py.MemberExpression.Part, {
-        id: "c"
-      }), _$createComponent(py.MemberExpression.Part, {
-        id: "d"
-      })];
+      return _$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "a"
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "b"
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "c"
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "d"
+          })];
+        }
+      });
     }
-  })])).toBe(d`
+  })).toRenderTo(`
     a.b.c.d
   `);
 });
 it("handles a mix of dot and bracket notation", () => {
-  expect(toSourceText([_$createComponent(py.MemberExpression, {
+  expect(_$createComponent(TestOutput, {
     get children() {
-      return [_$createComponent(py.MemberExpression.Part, {
-        id: "obj"
-      }), _$createComponent(py.MemberExpression.Part, {
-        id: "normalProp"
-      }), _$createComponent(py.MemberExpression.Part, {
-        key: "special-prop"
-      }), _$createComponent(py.MemberExpression.Part, {
-        key: "123"
-      })];
+      return _$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "obj"
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "normalProp"
+          }), _$createComponent(py.MemberExpression.Part, {
+            key: "special-prop"
+          }), _$createComponent(py.MemberExpression.Part, {
+            key: "123"
+          })];
+        }
+      });
     }
-  })])).toBe(d`
+  })).toRenderTo(`
     obj.normalProp["special-prop"]["123"]
   `);
 });
 it("throws an error when providing conflicting part props", () => {
-  expect(() => toSourceText([_$createComponent(py.MemberExpression, {
+  expect(() => render(_$createComponent(TestOutput, {
     get children() {
-      return [_$createComponent(py.MemberExpression.Part, {
-        id: "obj"
-      }), _$createComponent(py.MemberExpression.Part, {
-        id: "property",
-        args: [1, 2]
-      })];
+      return _$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "obj"
+          }), _$createComponent(py.MemberExpression.Part, {
+            id: "property",
+            args: [1, 2]
+          })];
+        }
+      });
     }
-  })])).toThrowError(`Only one of args, id can be used for a MemberExpression part at a time`);
+  }))).toThrowError(`Only one of args, id can be used for a MemberExpression part at a time`);
 });
 it("throws an error when providing an empty slice", () => {
-  expect(() => toSourceText([_$createComponent(py.MemberExpression, {
+  expect(() => render(_$createComponent(TestOutput, {
     get children() {
-      return [_$createComponent(py.MemberExpression.Part, {
-        id: "obj"
-      }), _$createComponent(py.MemberExpression.Part, {
-        slice: {}
-      })];
+      return _$createComponent(py.MemberExpression, {
+        get children() {
+          return [_$createComponent(py.MemberExpression.Part, {
+            id: "obj"
+          }), _$createComponent(py.MemberExpression.Part, {
+            slice: {}
+          })];
+        }
+      });
     }
-  })])).toThrowError(`MemberExpression.Part: slice object cannot be empty`);
+  }))).toThrowError(`MemberExpression.Part: slice object cannot be empty`);
 });
 it("throws an error when providing an empty array", () => {
-  expect(() => toSourceText([_$createComponent(py.MemberExpression, {
-    get children() {
-      return [_$createComponent(py.MemberExpression.Part, {
-        id: "obj"
-      }), _$createComponent(py.MemberExpression.Part, {
-        keys: []
-      })];
-    }
-  })])).toThrowError(`MemberExpression.Part: keys array cannot be empty`);
-});
-it("takes children for the id part", () => {
-  expect(toSourceText([_$createComponent(py.StatementList, {
+  expect(() => render(_$createComponent(TestOutput, {
     get children() {
-      return [_$createComponent(py.MemberExpression, {
+      return _$createComponent(py.MemberExpression, {
         get children() {
           return [_$createComponent(py.MemberExpression.Part, {
-            id: "child1"
+            id: "obj"
           }), _$createComponent(py.MemberExpression.Part, {
-            key: "child1"
+            keys: []
           })];
         }
-      }), _$createComponent(py.MemberExpression, {
+      });
+    }
+  }))).toThrowError(`MemberExpression.Part: keys array cannot be empty`);
+});
+it("takes children for the id part", () => {
+  expect(_$createComponent(TestOutput, {
+    get children() {
+      return _$createComponent(py.StatementList, {
         get children() {
-          return [_$createComponent(py.MemberExpression.Part, {
-            id: "child1"
-          }), _$createComponent(py.MemberExpression.Part, {
-            key: "child2"
-          }), _$createComponent(py.MemberExpression.Part, {
-            args: true
-          }), _$createComponent(py.MemberExpression.Part, {
-            key: "child3"
-          }), _$createComponent(py.MemberExpression.Part, {
-            args: true
-          }), _$createComponent(py.MemberExpression.Part, {
-            key: code`"foo" + 1`
-          }), _$createComponent(py.MemberExpression.Part, {
-            args: true
+          return [_$createComponent(py.MemberExpression, {
+            get children() {
+              return [_$createComponent(py.MemberExpression.Part, {
+                id: "child1"
+              }), _$createComponent(py.MemberExpression.Part, {
+                key: "child1"
+              })];
+            }
+          }), _$createComponent(py.MemberExpression, {
+            get children() {
+              return [_$createComponent(py.MemberExpression.Part, {
+                id: "child1"
+              }), _$createComponent(py.MemberExpression.Part, {
+                key: "child2"
+              }), _$createComponent(py.MemberExpression.Part, {
+                args: true
+              }), _$createComponent(py.MemberExpression.Part, {
+                key: "child3"
+              }), _$createComponent(py.MemberExpression.Part, {
+                args: true
+              }), _$createComponent(py.MemberExpression.Part, {
+                key: code`"foo" + 1`
+              }), _$createComponent(py.MemberExpression.Part, {
+                args: true
+              })];
+            }
           })];
         }
-      })];
+      });
     }
-  })])).toBe(d`
+  })).toRenderTo(`
     child1["child1"]
     child1["child2"]()["child3"]()["foo" + 1]()
   `);
@@ -396,27 +475,31 @@ describe("with refkeys", () => {
   it("handles symbols correctly", () => {
     const rk1 = refkey();
     const rk2 = refkey();
-    expect(toSourceText([_$createComponent(py.StatementList, {
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return [_$createComponent(py.VariableDeclaration, {
-          name: "test1",
-          refkey: rk1,
-          initializer: 1
-        }), _$createComponent(py.VariableDeclaration, {
-          name: "test1",
-          refkey: rk2,
-          initializer: 2
-        }), _$createComponent(py.MemberExpression, {
+        return _$createComponent(py.StatementList, {
           get children() {
-            return [_$createComponent(py.MemberExpression.Part, {
-              refkey: rk1
-            }), _$createComponent(py.MemberExpression.Part, {
-              refkey: rk2
+            return [_$createComponent(py.VariableDeclaration, {
+              name: "test1",
+              refkey: rk1,
+              initializer: 1
+            }), _$createComponent(py.VariableDeclaration, {
+              name: "test1",
+              refkey: rk2,
+              initializer: 2
+            }), _$createComponent(py.MemberExpression, {
+              get children() {
+                return [_$createComponent(py.MemberExpression.Part, {
+                  refkey: rk1
+                }), _$createComponent(py.MemberExpression.Part, {
+                  refkey: rk2
+                })];
+              }
             })];
           }
-        })];
+        });
       }
-    })])).toBe(d`
+    })).toRenderTo(`
         test1 = 1
         test1_2_test = 2
         test1.test1_2_test
@@ -488,7 +571,9 @@ describe("with refkeys", () => {
         }), [_$memo(() => memberRefkey(v1Rk, classMethod1Ref))], [_$memo(() => memberRefkey(v2Rk, classMethod2Ref))]];
       }
     });
-    expect(toSourceText([template])).toBe(d`
+    expect(_$createComponent(TestOutput, {
+      children: template
+    })).toRenderTo(`
       class Model1:
           foo: str
 
@@ -553,7 +638,9 @@ describe("with refkeys", () => {
         })];
       }
     });
-    expect(toSourceText([template])).toBe(d`
+    expect(_$createComponent(TestOutput, {
+      children: template
+    })).toRenderTo(`
       class Model:
           bar: str
 
@@ -570,61 +657,65 @@ describe("with refkeys", () => {
     const classMethodRefkey = refkey();
     const classMemberRefkey = refkey();
     const instanceRefkey = refkey();
-    expect(toSourceText([_$createComponent(py.StatementList, {
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return [_$createComponent(ClassDeclaration, {
-          name: "Bar",
-          refkey: interfaceRefkey,
-          get children() {
-            return _$createComponent(VariableDeclaration, {
-              name: "prop1",
-              refkey: interfaceMemberRefkey,
-              type: "str",
-              omitNone: true
-            });
-          }
-        }), _$createComponent(ClassDeclaration, {
-          name: "Foo",
-          refkey: classRefkey,
-          get children() {
-            return [_$createComponent(VariableDeclaration, {
-              name: "test1",
-              refkey: classMemberRefkey,
-              type: interfaceRefkey
-            }), _$createIntrinsic("br", {}), _$createComponent(FunctionDeclaration, {
-              name: "testMethod",
-              parameters: [],
-              refkey: classMethodRefkey,
-              returnType: interfaceRefkey
-            })];
-          }
-        }), _$createComponent(py.VariableDeclaration, {
-          name: "inst",
-          refkey: instanceRefkey,
-          initializer: code`${classRefkey}()`
-        }), _$createComponent(py.MemberExpression, {
-          get children() {
-            return [_$createComponent(py.MemberExpression.Part, {
-              refkey: instanceRefkey
-            }), _$createComponent(py.MemberExpression.Part, {
-              refkey: classMemberRefkey
-            }), _$createComponent(py.MemberExpression.Part, {
-              refkey: interfaceMemberRefkey
-            })];
-          }
-        }), _$createComponent(py.MemberExpression, {
+        return _$createComponent(py.StatementList, {
           get children() {
-            return [_$createComponent(py.MemberExpression.Part, {
-              refkey: instanceRefkey
-            }), _$createComponent(py.MemberExpression.Part, {
-              refkey: classMethodRefkey
-            }), _$createComponent(py.MemberExpression.Part, {
-              args: []
+            return [_$createComponent(ClassDeclaration, {
+              name: "Bar",
+              refkey: interfaceRefkey,
+              get children() {
+                return _$createComponent(VariableDeclaration, {
+                  name: "prop1",
+                  refkey: interfaceMemberRefkey,
+                  type: "str",
+                  omitNone: true
+                });
+              }
+            }), _$createComponent(ClassDeclaration, {
+              name: "Foo",
+              refkey: classRefkey,
+              get children() {
+                return [_$createComponent(VariableDeclaration, {
+                  name: "test1",
+                  refkey: classMemberRefkey,
+                  type: interfaceRefkey
+                }), _$createIntrinsic("br", {}), _$createComponent(FunctionDeclaration, {
+                  name: "testMethod",
+                  parameters: [],
+                  refkey: classMethodRefkey,
+                  returnType: interfaceRefkey
+                })];
+              }
+            }), _$createComponent(py.VariableDeclaration, {
+              name: "inst",
+              refkey: instanceRefkey,
+              initializer: code`${classRefkey}()`
+            }), _$createComponent(py.MemberExpression, {
+              get children() {
+                return [_$createComponent(py.MemberExpression.Part, {
+                  refkey: instanceRefkey
+                }), _$createComponent(py.MemberExpression.Part, {
+                  refkey: classMemberRefkey
+                }), _$createComponent(py.MemberExpression.Part, {
+                  refkey: interfaceMemberRefkey
+                })];
+              }
+            }), _$createComponent(py.MemberExpression, {
+              get children() {
+                return [_$createComponent(py.MemberExpression.Part, {
+                  refkey: instanceRefkey
+                }), _$createComponent(py.MemberExpression.Part, {
+                  refkey: classMethodRefkey
+                }), _$createComponent(py.MemberExpression.Part, {
+                  args: []
+                })];
+              }
             })];
           }
-        })];
+        });
       }
-    })])).toBe(d`
+    })).toRenderTo(`
        class Bar:
            prop1: str
 
@@ -642,27 +733,31 @@ describe("with refkeys", () => {
   it("handles late resolved refkeys correctly", () => {
     const rk1 = refkey();
     const rk2 = refkey();
-    expect(toSourceText([_$createComponent(py.StatementList, {
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return [_$createComponent(py.MemberExpression, {
+        return _$createComponent(py.StatementList, {
           get children() {
-            return [_$createComponent(py.MemberExpression.Part, {
-              refkey: rk1
-            }), _$createComponent(py.MemberExpression.Part, {
-              refkey: rk2
+            return [_$createComponent(py.MemberExpression, {
+              get children() {
+                return [_$createComponent(py.MemberExpression.Part, {
+                  refkey: rk1
+                }), _$createComponent(py.MemberExpression.Part, {
+                  refkey: rk2
+                })];
+              }
+            }), _$createComponent(py.VariableDeclaration, {
+              name: "test1",
+              refkey: rk1,
+              initializer: 1
+            }), _$createComponent(py.VariableDeclaration, {
+              name: "test1",
+              refkey: rk2,
+              initializer: 2
             })];
           }
-        }), _$createComponent(py.VariableDeclaration, {
-          name: "test1",
-          refkey: rk1,
-          initializer: 1
-        }), _$createComponent(py.VariableDeclaration, {
-          name: "test1",
-          refkey: rk2,
-          initializer: 2
-        })];
+        });
       }
-    })])).toBe(d`
+    })).toRenderTo(`
       test1.test1_2_test
       test1 = 1
       test1_2_test = 2
@@ -670,132 +765,156 @@ describe("with refkeys", () => {
   });
   it("creates a full reference to the first refkey", () => {
     const rk1 = refkey();
-    const res = toSourceTextMultiple([_$createComponent(SourceFile, {
-      path: "source.py",
-      get children() {
-        return _$createComponent(py.VariableDeclaration, {
-          name: "importMe",
-          refkey: rk1
-        });
-      }
-    }), _$createComponent(SourceFile, {
-      path: "index.py",
+    expect(_$createComponent(TestOutputDirectory, {
       get children() {
-        return _$createComponent(py.StatementList, {
+        return [_$createComponent(SourceFile, {
+          path: "source.py",
           get children() {
-            return _$createComponent(py.MemberExpression, {
+            return _$createComponent(py.VariableDeclaration, {
+              name: "importMe",
+              refkey: rk1
+            });
+          }
+        }), _$createComponent(SourceFile, {
+          path: "index.py",
+          get children() {
+            return _$createComponent(py.StatementList, {
               get children() {
-                return [_$createComponent(py.MemberExpression.Part, {
-                  refkey: rk1
-                }), _$createComponent(py.MemberExpression.Part, {
-                  id: "foo"
-                })];
+                return _$createComponent(py.MemberExpression, {
+                  get children() {
+                    return [_$createComponent(py.MemberExpression.Part, {
+                      refkey: rk1
+                    }), _$createComponent(py.MemberExpression.Part, {
+                      id: "foo"
+                    })];
+                  }
+                });
               }
             });
           }
-        });
+        })];
       }
-    })]);
-    assertFileContents(res, {
-      "index.py": d`
-        from source import import_me
+    })).toRenderTo({
+      "index.py": `
+          from source import import_me
 
-        import_me.foo
-      `
+          import_me.foo
+        `,
+      "source.py": "import_me = None"
     });
   });
 });
 describe("with function calls", () => {
   it("handles simple function calls correctly", () => {
-    expect(toSourceText([_$createComponent(py.MemberExpression, {
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return [_$createComponent(py.MemberExpression.Part, {
-          id: "myFunction"
-        }), _$createComponent(py.MemberExpression.Part, {
-          args: [1, 2]
-        })];
+        return _$createComponent(py.MemberExpression, {
+          get children() {
+            return [_$createComponent(py.MemberExpression.Part, {
+              id: "myFunction"
+            }), _$createComponent(py.MemberExpression.Part, {
+              args: [1, 2]
+            })];
+          }
+        });
       }
-    })])).toBe(d`
+    })).toRenderTo(`
       myFunction(1, 2)
     `);
   });
   it("handles function calls with references correctly", () => {
     const xRefkey = refkey();
     const yRefkey = refkey();
-    expect(toSourceText([_$createComponent(py.StatementList, {
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return [_$createComponent(py.VariableDeclaration, {
-          name: "x",
-          initializer: 1,
-          refkey: xRefkey
-        }), _$createComponent(py.VariableDeclaration, {
-          name: "y",
-          initializer: 2,
-          refkey: yRefkey
-        }), _$createComponent(py.MemberExpression, {
+        return _$createComponent(py.StatementList, {
           get children() {
-            return [_$createComponent(py.MemberExpression.Part, {
-              id: "myFunction"
-            }), _$createComponent(py.MemberExpression.Part, {
-              args: [xRefkey, yRefkey]
+            return [_$createComponent(py.VariableDeclaration, {
+              name: "x",
+              initializer: 1,
+              refkey: xRefkey
+            }), _$createComponent(py.VariableDeclaration, {
+              name: "y",
+              initializer: 2,
+              refkey: yRefkey
+            }), _$createComponent(py.MemberExpression, {
+              get children() {
+                return [_$createComponent(py.MemberExpression.Part, {
+                  id: "myFunction"
+                }), _$createComponent(py.MemberExpression.Part, {
+                  args: [xRefkey, yRefkey]
+                })];
+              }
             })];
           }
-        })];
+        });
       }
-    })])).toBe(d`
+    })).toRenderTo(`
       x = 1
       y = 2
       myFunction(x, y)
     `);
   });
   it("handles method calls correctly", () => {
-    expect(toSourceText([_$createComponent(py.MemberExpression, {
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return [_$createComponent(py.MemberExpression.Part, {
-          id: "method1"
-        }), _$createComponent(py.MemberExpression.Part, {
-          args: [1, 2]
-        }), _$createComponent(py.MemberExpression.Part, {
-          args: []
-        }), _$createComponent(py.MemberExpression.Part, {
-          id: "method2"
-        }), _$createComponent(py.MemberExpression.Part, {
-          args: []
-        }), _$createComponent(py.MemberExpression.Part, {
-          id: "prop"
-        })];
+        return _$createComponent(py.MemberExpression, {
+          get children() {
+            return [_$createComponent(py.MemberExpression.Part, {
+              id: "method1"
+            }), _$createComponent(py.MemberExpression.Part, {
+              args: [1, 2]
+            }), _$createComponent(py.MemberExpression.Part, {
+              args: []
+            }), _$createComponent(py.MemberExpression.Part, {
+              id: "method2"
+            }), _$createComponent(py.MemberExpression.Part, {
+              args: []
+            }), _$createComponent(py.MemberExpression.Part, {
+              id: "prop"
+            })];
+          }
+        });
       }
-    })])).toBe(d`
+    })).toRenderTo(`
       method1(1, 2)().method2().prop
     `);
   });
   it("handles function calls correctly", () => {
-    expect(toSourceText([_$createComponent(py.MemberExpression, {
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return [_$createComponent(py.MemberExpression.Part, {
-          id: "myFunction"
-        }), _$createComponent(py.MemberExpression.Part, {
-          args: [1, 2]
-        }), _$createComponent(py.MemberExpression.Part, {
-          id: "prop"
-        })];
+        return _$createComponent(py.MemberExpression, {
+          get children() {
+            return [_$createComponent(py.MemberExpression.Part, {
+              id: "myFunction"
+            }), _$createComponent(py.MemberExpression.Part, {
+              args: [1, 2]
+            }), _$createComponent(py.MemberExpression.Part, {
+              id: "prop"
+            })];
+          }
+        });
       }
-    })])).toBe(d`
+    })).toRenderTo(`
       myFunction(1, 2).prop
     `);
   });
   it("handles function calls correctly", () => {
-    expect(toSourceText([_$createComponent(py.MemberExpression, {
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return [_$createComponent(py.MemberExpression.Part, {
-          id: "myFunction"
-        }), _$createComponent(py.MemberExpression.Part, {
-          args: [1, 2]
-        }), _$createComponent(py.MemberExpression.Part, {
-          id: "prop"
-        })];
+        return _$createComponent(py.MemberExpression, {
+          get children() {
+            return [_$createComponent(py.MemberExpression.Part, {
+              id: "myFunction"
+            }), _$createComponent(py.MemberExpression.Part, {
+              args: [1, 2]
+            }), _$createComponent(py.MemberExpression.Part, {
+              id: "prop"
+            })];
+          }
+        });
       }
-    })])).toBe(d`
+    })).toRenderTo(`
       myFunction(1, 2).prop
     `);
   });
@@ -803,75 +922,85 @@ describe("with function calls", () => {
 describe("formatting", () => {
   describe("simple chains", () => {
     it("just dots", () => {
-      expect(toSourceText([_$createComponent(py.MemberExpression, {
+      expect(_$createComponent(TestOutput, {
         get children() {
-          return [_$createComponent(py.MemberExpression.Part, {
-            id: "four"
-          }), _$createComponent(py.MemberExpression.Part, {
-            id: "four"
-          }), _$createComponent(py.MemberExpression.Part, {
-            id: "two"
-          }), _$createComponent(py.MemberExpression.Part, {
-            id: "two"
-          }), _$createComponent(py.MemberExpression.Part, {
-            id: "two"
-          }), _$createComponent(py.MemberExpression.Part, {
-            id: "two"
-          })];
-        }
-      })], {
-        printOptions: {
-          printWidth: 14
+          return _$createComponent(py.MemberExpression, {
+            get children() {
+              return [_$createComponent(py.MemberExpression.Part, {
+                id: "four"
+              }), _$createComponent(py.MemberExpression.Part, {
+                id: "four"
+              }), _$createComponent(py.MemberExpression.Part, {
+                id: "two"
+              }), _$createComponent(py.MemberExpression.Part, {
+                id: "two"
+              }), _$createComponent(py.MemberExpression.Part, {
+                id: "two"
+              }), _$createComponent(py.MemberExpression.Part, {
+                id: "two"
+              })];
+            }
+          });
         }
-      })).toBe(d`
+      })).toRenderTo(`
         four.four \\
             .two.two \\
             .two.two
-      `);
+      `, {
+        tabWidth: 4,
+        printWidth: 14
+      });
     });
     it("bracket breaks", () => {
-      expect(toSourceText([_$createComponent(py.MemberExpression, {
+      expect(_$createComponent(TestOutput, {
         get children() {
-          return [_$createComponent(py.MemberExpression.Part, {
-            id: "obj"
-          }), _$createComponent(py.MemberExpression.Part, {
-            key: "property-name"
-          }), _$createComponent(py.MemberExpression.Part, {
-            id: "prop"
-          })];
-        }
-      })], {
-        printOptions: {
-          printWidth: 12
+          return _$createComponent(py.MemberExpression, {
+            get children() {
+              return [_$createComponent(py.MemberExpression.Part, {
+                id: "obj"
+              }), _$createComponent(py.MemberExpression.Part, {
+                key: "property-name"
+              }), _$createComponent(py.MemberExpression.Part, {
+                id: "prop"
+              })];
+            }
+          });
         }
-      })).toBe(d`
+      })).toRenderTo(`
         obj[
             "property-name"
         ].prop
-      `);
+      `, {
+        tabWidth: 4,
+        printWidth: 12
+      });
     });
   });
   describe("call chains", () => {
     it("renders multiple calls on the same line when there are no breaks and they fit", () => {
-      expect(toSourceText([_$createComponent(py.MemberExpression, {
+      expect(_$createComponent(TestOutput, {
         get children() {
-          return [_$createComponent(py.MemberExpression.Part, {
-            id: "z"
-          }), _$createComponent(py.MemberExpression.Part, {
-            id: "object"
-          }), _$createComponent(py.MemberExpression.Part, {
-            args: true
-          }), _$createComponent(py.MemberExpression.Part, {
-            id: "partial"
-          }), _$createComponent(py.MemberExpression.Part, {
-            args: true
-          }), _$createComponent(py.MemberExpression.Part, {
-            id: "optional"
-          }), _$createComponent(py.MemberExpression.Part, {
-            args: true
-          })];
+          return _$createComponent(py.MemberExpression, {
+            get children() {
+              return [_$createComponent(py.MemberExpression.Part, {
+                id: "z"
+              }), _$createComponent(py.MemberExpression.Part, {
+                id: "object"
+              }), _$createComponent(py.MemberExpression.Part, {
+                args: true
+              }), _$createComponent(py.MemberExpression.Part, {
+                id: "partial"
+              }), _$createComponent(py.MemberExpression.Part, {
+                args: true
+              }), _$createComponent(py.MemberExpression.Part, {
+                id: "optional"
+              }), _$createComponent(py.MemberExpression.Part, {
+                args: true
+              })];
+            }
+          });
         }
-      })])).toBe(d`
+      })).toRenderTo(`
         z.object().partial().optional()
       `);
     });