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

package/dist/dev/test/functiondeclaration.test.js

@@ -1,231 +1,316 @@
 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", () => {
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.FunctionDeclaration, {
+          name: "f",
+          decorators: ["@a", "@b", "@c"],
+          children: "pass"
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 11,
+          columnNumber: 9
+        });
+      }
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 10,
+      columnNumber: 7
+    })).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(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.FunctionDeclaration, {
+          name: "foo"
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 30,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 14,
-      columnNumber: 34
-    })]);
-    expect(result).toRenderTo(d`
+      lineNumber: 29,
+      columnNumber: 7
+    })).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");
+          }
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 44,
+          columnNumber: 9
+        });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 25,
+      lineNumber: 43,
       columnNumber: 7
-    })]);
-    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(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.FunctionDeclaration, {
+          name: "foo",
+          returnType: "int"
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 58,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 37,
+      lineNumber: 57,
       columnNumber: 7
-    })]);
-    expect(result).toRenderTo(d`
+    })).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
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 51,
-          columnNumber: 9
-        }), _$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
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 74,
+              columnNumber: 11
+            }), _$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: []
+                    }, {
+                      fileName: import.meta.url,
+                      lineNumber: 84,
+                      columnNumber: 17
+                    });
+                  }
                 }, {
                   fileName: import.meta.url,
-                  lineNumber: 61,
-                  columnNumber: 15
+                  lineNumber: 80,
+                  columnNumber: 13
                 });
               }
             }, {
               fileName: import.meta.url,
-              lineNumber: 57,
+              lineNumber: 79,
               columnNumber: 11
-            });
+            })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 56,
+          lineNumber: 73,
           columnNumber: 9
-        })];
+        });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 50,
+      lineNumber: 72,
       columnNumber: 7
-    })]);
-    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(_$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)"
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 105,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 80,
+      lineNumber: 104,
       columnNumber: 7
-    })]);
-    expect(result).toRenderTo(d`
+    })).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"
+              }]
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 131,
+              columnNumber: 11
+            });
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 106,
+          lineNumber: 130,
           columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 105,
+      lineNumber: 129,
       columnNumber: 7
-    })]);
-    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"
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.FunctionDeclaration, {
+          async: true,
+          name: "foo"
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 151,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 123,
-      columnNumber: 26
-    })])).toBe(d`
+      lineNumber: 150,
+      columnNumber: 7
+    })).toRenderTo(`
       async def foo():
           pass
 
     `);
   });
   it("can be an async function with returnType", () => {
-    expect(toSourceText([_$createComponent(py.FunctionDeclaration, {
-      async: true,
-      name: "foo",
-      returnType: "Foo"
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.FunctionDeclaration, {
+          async: true,
+          name: "foo",
+          returnType: "Foo"
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 165,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 133,
-      columnNumber: 9
-    })])).toBe(d`
+      lineNumber: 164,
+      columnNumber: 7
+    })).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");
-          }
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 146,
-          columnNumber: 11
-        }), _$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");
+              }
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 180,
+              columnNumber: 11
+            }), _$createComponent(py.FunctionDeclaration, {
+              async: true,
+              name: "foo",
+              get returnType() {
+                return refkey("Foo");
+              }
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 181,
+              columnNumber: 11
+            })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 147,
-          columnNumber: 11
-        })];
+          lineNumber: 179,
+          columnNumber: 9
+        });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 145,
-      columnNumber: 9
-    })])).toBe(d`
+      lineNumber: 178,
+      columnNumber: 7
+    })).toRenderTo(`
       class Foo:
           pass
 
@@ -235,34 +320,42 @@ 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");
-          }
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 164,
-          columnNumber: 11
-        }), _$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");
+              }
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 200,
+              columnNumber: 11
+            }), _$createComponent(py.FunctionDeclaration, {
+              async: true,
+              name: "foo",
+              get returnType() {
+                return code`list[${refkey("Foo")}]`;
+              }
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 201,
+              columnNumber: 11
+            })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 165,
-          columnNumber: 11
-        })];
+          lineNumber: 199,
+          columnNumber: 9
+        });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 163,
-      columnNumber: 9
-    })])).toBe(d`
+      lineNumber: 198,
+      columnNumber: 7
+    })).toRenderTo(`
       class Foo:
           pass
 
@@ -286,28 +379,34 @@ describe("Function Declaration", () => {
                   children: "return self"
                 }, {
                   fileName: import.meta.url,
-                  lineNumber: 187,
+                  lineNumber: 225,
                   columnNumber: 13
                 });
               }
             }, {
               fileName: import.meta.url,
-              lineNumber: 186,
+              lineNumber: 224,
               columnNumber: 11
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 185,
+          lineNumber: 223,
           columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 184,
+      lineNumber: 222,
       columnNumber: 7
     });
-    expect(toSourceText([decl])).toBe(d`
+    expect(_$createComponent(TestOutput, {
+      children: decl
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 237,
+      columnNumber: 12
+    })).toRenderTo(`
       class MyClass:
           async def __aenter__(self) -> MyClass:
               return self
@@ -329,28 +428,34 @@ describe("Function Declaration", () => {
                   children: "return super().__new__(cls)"
                 }, {
                   fileName: import.meta.url,
-                  lineNumber: 213,
+                  lineNumber: 253,
                   columnNumber: 13
                 });
               }
             }, {
               fileName: import.meta.url,
-              lineNumber: 212,
+              lineNumber: 252,
               columnNumber: 11
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 211,
+          lineNumber: 251,
           columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 210,
+      lineNumber: 250,
       columnNumber: 7
     });
-    expect(toSourceText([decl])).toBe(d`
+    expect(_$createComponent(TestOutput, {
+      children: decl
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 261,
+      columnNumber: 12
+    })).toRenderTo(`
       class MyClass:
           async def __new__(cls) -> MyClass:
               return super().__new__(cls)
@@ -369,10 +474,16 @@ describe("Function Declaration", () => {
       children: "return a + b"
     }, {
       fileName: import.meta.url,
-      lineNumber: 232,
+      lineNumber: 274,
       columnNumber: 7
     });
-    expect(toSourceText([decl])).toBe(d`
+    expect(_$createComponent(TestOutput, {
+      children: decl
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 282,
+      columnNumber: 12
+    })).toRenderTo(`
       def foo(a, b):
           return a + b
 
@@ -390,10 +501,16 @@ describe("Function Declaration", () => {
       children: "return a + b"
     }, {
       fileName: import.meta.url,
-      lineNumber: 248,
+      lineNumber: 293,
       columnNumber: 7
     });
-    expect(toSourceText([decl])).toBe(d`
+    expect(_$createComponent(TestOutput, {
+      children: decl
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 302,
+      columnNumber: 12
+    })).toRenderTo(`
       def foo[T, U](a, b):
           return a + b
 
@@ -417,7 +534,7 @@ describe("Function Declaration", () => {
                   abstract: true
                 }, {
                   fileName: import.meta.url,
-                  lineNumber: 269,
+                  lineNumber: 317,
                   columnNumber: 13
                 }), _$createComponent(py.ClassMethodDeclaration, {
                   name: "classdef",
@@ -425,7 +542,7 @@ describe("Function Declaration", () => {
                   abstract: true
                 }, {
                   fileName: import.meta.url,
-                  lineNumber: 274,
+                  lineNumber: 322,
                   columnNumber: 13
                 }), _$createComponent(py.StaticMethodDeclaration, {
                   name: "staticdef",
@@ -433,30 +550,35 @@ describe("Function Declaration", () => {
                   abstract: true
                 }, {
                   fileName: import.meta.url,
-                  lineNumber: 279,
+                  lineNumber: 327,
                   columnNumber: 13
                 })];
               }
             }, {
               fileName: import.meta.url,
-              lineNumber: 268,
+              lineNumber: 316,
               columnNumber: 11
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 267,
+          lineNumber: 315,
           columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 266,
+      lineNumber: 314,
       columnNumber: 7
     });
-    expect(toSourceText([decl], {
-      externals: [abcModule]
-    })).toBe(d`
+    expect(_$createComponent(TestOutput, {
+      externals: [abcModule],
+      children: decl
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 337,
+      columnNumber: 12
+    })).toRenderTo(`
       from abc import abstractmethod
 
 
@@ -494,7 +616,7 @@ describe("Function Declaration", () => {
               children: "self.attribute = \"value\""
             }, {
               fileName: import.meta.url,
-              lineNumber: 316,
+              lineNumber: 367,
               columnNumber: 11
             }), _$createComponent(py.DunderMethodDeclaration, {
               name: "__repr__",
@@ -502,22 +624,28 @@ describe("Function Declaration", () => {
               children: "return \"MyClass\""
             }, {
               fileName: import.meta.url,
-              lineNumber: 319,
+              lineNumber: 370,
               columnNumber: 11
             })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 315,
+          lineNumber: 366,
           columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 314,
+      lineNumber: 365,
       columnNumber: 7
     });
-    expect(toSourceText([decl])).toBe(d`
+    expect(_$createComponent(TestOutput, {
+      children: decl
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 377,
+      columnNumber: 12
+    })).toRenderTo(`
       class MyClass:
           def __init__(self, x: int):
               self.attribute = "value"
@@ -540,22 +668,28 @@ describe("Function Declaration", () => {
               children: "pass"
             }, {
               fileName: import.meta.url,
-              lineNumber: 342,
+              lineNumber: 395,
               columnNumber: 11
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 341,
+          lineNumber: 394,
           columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 340,
+      lineNumber: 393,
       columnNumber: 7
     });
-    expect(toSourceText([decl])).toBe(d`
+    expect(_$createComponent(TestOutput, {
+      children: decl
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 402,
+      columnNumber: 12
+    })).toRenderTo(`
       class MyClass:
           def __new__(cls, *args, **kwargs):
               pass
@@ -596,7 +730,7 @@ describe("Function Declaration", () => {
               children: "return z * 2"
             }, {
               fileName: import.meta.url,
-              lineNumber: 376,
+              lineNumber: 431,
               columnNumber: 11
             }), "return", " ", _$createComponent(py.FunctionCallExpression, {
               target: foobarRef,
@@ -605,19 +739,19 @@ describe("Function Declaration", () => {
                   jsValue: 2
                 }, {
                   fileName: import.meta.url,
-                  lineNumber: 386,
+                  lineNumber: 441,
                   columnNumber: 20
                 })];
               }
             }, {
               fileName: import.meta.url,
-              lineNumber: 384,
+              lineNumber: 439,
               columnNumber: 11
             })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 371,
+          lineNumber: 426,
           columnNumber: 9
         }), "return", " ", _$createComponent(py.FunctionCallExpression, {
           target: barRef,
@@ -626,22 +760,28 @@ describe("Function Declaration", () => {
               jsValue: 3
             }, {
               fileName: import.meta.url,
-              lineNumber: 392,
+              lineNumber: 447,
               columnNumber: 18
             })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 390,
+          lineNumber: 445,
           columnNumber: 9
         })];
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 366,
+      lineNumber: 421,
       columnNumber: 7
     });
-    expect(toSourceText([decl])).toBe(d`
+    expect(_$createComponent(TestOutput, {
+      children: decl
+    }, {
+      fileName: import.meta.url,
+      lineNumber: 452,
+      columnNumber: 12
+    })).toRenderTo(`
       def foo(x: int):
           def bar(y: int):
               def foobar(z: int):
@@ -652,83 +792,90 @@ 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");
-          }
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 410,
-          columnNumber: 9
-        });
-      }
-    }, {
-      fileName: import.meta.url,
-      lineNumber: 409,
-      columnNumber: 7
-    }), _$createComponent(py.SourceFile, {
-      path: "mod2.py",
-      get children() {
-        return [_$createComponent(py.ClassDeclaration, {
-          name: "A",
-          get refkey() {
-            return refkey("A");
+        return [_$createComponent(py.SourceFile, {
+          path: "mod1.py",
+          get children() {
+            return _$createComponent(py.ClassDeclaration, {
+              name: "Foo",
+              get refkey() {
+                return refkey("Foo");
+              }
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 469,
+              columnNumber: 11
+            });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 413,
+          lineNumber: 468,
           columnNumber: 9
-        }), _$createComponent(py.ClassDeclaration, {
-          name: "B",
-          get refkey() {
-            return refkey("B");
+        }), _$createComponent(py.SourceFile, {
+          path: "mod2.py",
+          get children() {
+            return [_$createComponent(py.ClassDeclaration, {
+              name: "A",
+              get refkey() {
+                return refkey("A");
+              }
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 472,
+              columnNumber: 11
+            }), _$createComponent(py.ClassDeclaration, {
+              name: "B",
+              get refkey() {
+                return refkey("B");
+              }
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 473,
+              columnNumber: 11
+            })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 414,
+          lineNumber: 471,
           columnNumber: 9
-        })];
-      }
-    }, {
-      fileName: import.meta.url,
-      lineNumber: 412,
-      columnNumber: 7
-    }), _$createComponent(py.SourceFile, {
-      path: "usage.py",
-      get children() {
-        return _$createComponent(py.FunctionDeclaration, {
-          async: true,
-          name: "foo",
-          get parameters() {
-            return [{
-              name: "x",
-              type: refkey("A")
+        }), _$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");
+              }
             }, {
-              name: "y",
-              type: refkey("B")
-            }];
-          },
-          args: true,
-          kwargs: true,
-          get returnType() {
-            return refkey("Foo");
+              fileName: import.meta.url,
+              lineNumber: 476,
+              columnNumber: 11
+            });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 417,
+          lineNumber: 475,
           columnNumber: 9
-        });
+        })];
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 416,
+      lineNumber: 467,
       columnNumber: 7
-    })]);
-    assertFileContents(res, {
+    })).toRenderTo({
       "mod1.py": `
             class Foo:
                 pass
@@ -760,57 +907,97 @@ 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"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 530,
+            columnNumber: 11
+          });
+        }
       }, {
         fileName: import.meta.url,
-        lineNumber: 470,
-        columnNumber: 21
-      })]);
+        lineNumber: 529,
+        columnNumber: 9
+      }));
     }).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"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 540,
+            columnNumber: 11
+          });
+        }
       }, {
         fileName: import.meta.url,
-        lineNumber: 476,
-        columnNumber: 21
-      })]);
+        lineNumber: 539,
+        columnNumber: 9
+      }));
     }).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"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 552,
+            columnNumber: 11
+          });
+        }
       }, {
         fileName: import.meta.url,
-        lineNumber: 484,
-        columnNumber: 21
-      })]);
+        lineNumber: 551,
+        columnNumber: 9
+      }));
     }).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"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 564,
+            columnNumber: 11
+          });
+        }
       }, {
         fileName: import.meta.url,
-        lineNumber: 492,
-        columnNumber: 21
-      })]);
+        lineNumber: 563,
+        columnNumber: 9
+      }));
     }).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__"
+          }, {
+            fileName: import.meta.url,
+            lineNumber: 576,
+            columnNumber: 11
+          });
+        }
       }, {
         fileName: import.meta.url,
-        lineNumber: 500,
-        columnNumber: 21
-      })]);
+        lineNumber: 575,
+        columnNumber: 9
+      }));
     }).toThrow('Method "__init__" must be declared inside a class (member scope)');
   });
 });