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

package/dist/dev/test/variables.test.js

@@ -1,280 +1,384 @@
 import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
 import { namekey, refkey } from "@alloy-js/core";
-import { d } from "@alloy-js/core/testing";
 import { describe, expect, it } from "vitest";
 import * as py from "../src/index.js";
-import { assertFileContents, toSourceText, toSourceTextMultiple } from "./utils.js";
+import { TestOutput, TestOutputDirectory } from "./utils.js";
 describe("Python Variable", () => {
   it("declares a python variable", () => {
-    const res = toSourceText([_$createComponent(py.VariableDeclaration, {
-      name: "myVar",
-      type: "int",
-      initializer: 42
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.VariableDeclaration, {
+          name: "myVar",
+          type: "int",
+          initializer: 42
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 10,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 14,
+      lineNumber: 9,
       columnNumber: 7
-    })]);
-    expect(res).toBe(`my_var: int = 42`);
+    })).toRenderTo("my_var: int = 42");
   });
   it("takes a namekey", () => {
-    const res = toSourceText([_$createComponent(py.VariableDeclaration, {
-      get name() {
-        return namekey("my-var");
-      },
-      type: "int",
-      initializer: 42
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.VariableDeclaration, {
+          get name() {
+            return namekey("my-var");
+          },
+          type: "int",
+          initializer: 42
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 18,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 21,
+      lineNumber: 17,
       columnNumber: 7
-    })]);
-    expect(res).toBe(`my_var: int = 42`);
+    })).toRenderTo("my_var: int = 42");
   });
   it("takes a namekey", () => {
-    const res = toSourceText([_$createComponent(py.VariableDeclaration, {
-      get name() {
-        return namekey("my-var");
-      },
-      type: "int",
-      initializer: 42
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.VariableDeclaration, {
+          get name() {
+            return namekey("my-var");
+          },
+          type: "int",
+          initializer: 42
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 30,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 32,
+      lineNumber: 29,
       columnNumber: 7
-    })]);
-    expect(res).toBe(`my_var: int = 42`);
+    })).toRenderTo("my_var: int = 42");
   });
   it("declares a python variable without value", () => {
-    const res = toSourceText([_$createComponent(py.VariableDeclaration, {
-      name: "myVar",
-      type: "int",
-      omitNone: true
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.VariableDeclaration, {
+          name: "myVar",
+          type: "int",
+          omitNone: true
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 42,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 43,
+      lineNumber: 41,
       columnNumber: 7
-    })]);
-    expect(res).toBe(`my_var: int`);
+    })).toRenderTo("my_var: int");
   });
   it("declares a python variable without typeAnnotations", () => {
-    const res = toSourceText([_$createComponent(py.VariableDeclaration, {
-      name: "myVar",
-      initializer: 42
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.VariableDeclaration, {
+          name: "myVar",
+          initializer: 42
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 50,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 50,
+      lineNumber: 49,
       columnNumber: 7
-    })]);
-    expect(res).toBe(`my_var = 42`);
+    })).toRenderTo("my_var = 42");
   });
   it("declares a python variable as None when undefined", () => {
-    const res = toSourceText([_$createComponent(py.VariableDeclaration, {
-      name: "myVar"
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.VariableDeclaration, {
+          name: "myVar"
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 58,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 56,
-      columnNumber: 31
-    })]);
-    expect(res).toBe(`my_var = None`);
+      lineNumber: 57,
+      columnNumber: 7
+    })).toRenderTo("my_var = None");
   });
   it("declares a python variable as None when null", () => {
-    const res = toSourceText([_$createComponent(py.VariableDeclaration, {
-      name: "myVar",
-      get initializer() {
-        return _$createComponent(py.Atom, {
-          jsValue: null
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.VariableDeclaration, {
+          name: "myVar",
+          get initializer() {
+            return _$createComponent(py.Atom, {
+              jsValue: null
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 68,
+              columnNumber: 24
+            });
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 64,
-          columnNumber: 22
+          lineNumber: 66,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 62,
+      lineNumber: 65,
       columnNumber: 7
-    })]);
-    expect(res).toBe(`my_var = None`);
+    })).toRenderTo("my_var = None");
   });
   it("declares a python variable that's an array", () => {
-    const res = toSourceText([_$createComponent(py.VariableDeclaration, {
-      name: "numbers",
-      get type() {
-        return _$createComponent(py.TypeReference, {
-          name: "list",
-          typeArgs: ["int"]
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 74,
-          columnNumber: 15
-        });
-      },
-      get initializer() {
-        return _$createComponent(py.Atom, {
-          jsValue: [1, 2, 3]
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.VariableDeclaration, {
+          name: "numbers",
+          get type() {
+            return _$createComponent(py.TypeReference, {
+              name: "list",
+              typeArgs: ["int"]
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 79,
+              columnNumber: 17
+            });
+          },
+          get initializer() {
+            return _$createComponent(py.Atom, {
+              jsValue: [1, 2, 3]
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 80,
+              columnNumber: 24
+            });
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 75,
-          columnNumber: 22
+          lineNumber: 77,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 72,
+      lineNumber: 76,
       columnNumber: 7
-    })]);
-    expect(res).toBe(`numbers: list[int] = [1, 2, 3]`);
+    })).toRenderTo("numbers: list[int] = [1, 2, 3]");
   });
   it("declares a python variable with a python value", () => {
-    const res = toSourceText([_$createComponent(py.VariableDeclaration, {
-      name: "nameIdPairs",
-      get initializer() {
-        return _$createComponent(py.Atom, {
-          jsValue: {
-            John: 123,
-            Doe: 234
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.VariableDeclaration, {
+          name: "nameIdPairs",
+          get initializer() {
+            return _$createComponent(py.Atom, {
+              jsValue: {
+                John: 123,
+                Doe: 234
+              }
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 91,
+              columnNumber: 24
+            });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 85,
-          columnNumber: 22
+          lineNumber: 89,
+          columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 83,
+      lineNumber: 88,
       columnNumber: 7
-    })]);
-    expect(res).toBe(`name_id_pairs = {"John": 123, "Doe": 234}`);
+    })).toRenderTo(`name_id_pairs = {"John": 123, "Doe": 234}`);
   });
   it("declares a python variable with omitNone", () => {
-    const res = toSourceText([_$createComponent(py.VariableDeclaration, {
-      name: "omitNoneVar",
-      type: "int",
-      omitNone: true
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.VariableDeclaration, {
+          name: "omitNoneVar",
+          type: "int",
+          omitNone: true
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 100,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 93,
+      lineNumber: 99,
       columnNumber: 7
-    })]);
-    expect(res).toBe(`omit_none_var: int`);
+    })).toRenderTo("omit_none_var: int");
   });
   it("declares a call statement python variable", () => {
-    const res = toSourceText([_$createComponent(py.VariableDeclaration, {
-      name: "callStmtVar",
-      initializer: 12,
-      callStatementVar: true
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.VariableDeclaration, {
+          name: "callStmtVar",
+          initializer: 12,
+          callStatementVar: true
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 108,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 100,
+      lineNumber: 107,
       columnNumber: 7
-    })]);
-    expect(res).toBe(`call_stmt_var=12`);
+    })).toRenderTo("call_stmt_var=12");
   });
   it("declares a call statement python variable without name", () => {
-    const res = toSourceText([_$createComponent(py.VariableDeclaration, {
-      name: "",
-      initializer: 12,
-      callStatementVar: true
+    expect(_$createComponent(TestOutput, {
+      get children() {
+        return _$createComponent(py.VariableDeclaration, {
+          name: "",
+          initializer: 12,
+          callStatementVar: true
+        }, {
+          fileName: import.meta.url,
+          lineNumber: 120,
+          columnNumber: 9
+        });
+      }
     }, {
       fileName: import.meta.url,
-      lineNumber: 111,
+      lineNumber: 119,
       columnNumber: 7
-    })]);
-    expect(res).toBe(`12`);
+    })).toRenderTo("12");
   });
   it("declares a python variable with an optional type", () => {
-    const res = toSourceText([_$createComponent(py.StatementList, {
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return _$createComponent(py.VariableDeclaration, {
-          name: "my_var",
-          get type() {
-            return _$createComponent(py.UnionTypeExpression, {
-              children: ["int", "None"]
+        return _$createComponent(py.StatementList, {
+          get children() {
+            return _$createComponent(py.VariableDeclaration, {
+              name: "my_var",
+              get type() {
+                return _$createComponent(py.UnionTypeExpression, {
+                  children: ["int", "None"]
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 136,
+                  columnNumber: 15
+                });
+              }
             }, {
               fileName: import.meta.url,
-              lineNumber: 126,
-              columnNumber: 13
+              lineNumber: 133,
+              columnNumber: 11
             });
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 123,
+          lineNumber: 132,
           columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 122,
+      lineNumber: 131,
       columnNumber: 7
-    })]);
-    expect(res).toBe(d`
+    })).toRenderTo(`
       my_var: int | None = None`);
   });
   it("declares a python variable with an optional type omitting none", () => {
-    const res = toSourceText([_$createComponent(py.StatementList, {
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return _$createComponent(py.VariableDeclaration, {
-          name: "my_var",
-          get type() {
-            return _$createComponent(py.UnionTypeExpression, {
-              children: ["int", "None"]
+        return _$createComponent(py.StatementList, {
+          get children() {
+            return _$createComponent(py.VariableDeclaration, {
+              name: "my_var",
+              get type() {
+                return _$createComponent(py.UnionTypeExpression, {
+                  children: ["int", "None"]
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 154,
+                  columnNumber: 15
+                });
+              },
+              omitNone: true
             }, {
               fileName: import.meta.url,
-              lineNumber: 141,
-              columnNumber: 13
+              lineNumber: 151,
+              columnNumber: 11
             });
-          },
-          omitNone: true
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 138,
+          lineNumber: 150,
           columnNumber: 9
         });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 137,
+      lineNumber: 149,
       columnNumber: 7
-    })]);
-    expect(res).toBe(d`
+    })).toRenderTo(`
       my_var: int | None`);
   });
   it("declares a python variable with a class type", () => {
     const classKey = refkey();
-    const res = toSourceText([_$createComponent(py.StatementList, {
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return [_$createComponent(py.ClassDeclaration, {
-          name: "MyClass",
-          refkey: classKey
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 155,
-          columnNumber: 9
-        }), _$createComponent(py.VariableDeclaration, {
-          name: "my_var",
-          get type() {
-            return _$createComponent(py.Reference, {
+        return _$createComponent(py.StatementList, {
+          get children() {
+            return [_$createComponent(py.ClassDeclaration, {
+              name: "MyClass",
               refkey: classKey
             }, {
               fileName: import.meta.url,
-              lineNumber: 158,
-              columnNumber: 17
-            });
+              lineNumber: 171,
+              columnNumber: 11
+            }), _$createComponent(py.VariableDeclaration, {
+              name: "my_var",
+              get type() {
+                return _$createComponent(py.Reference, {
+                  refkey: classKey
+                }, {
+                  fileName: import.meta.url,
+                  lineNumber: 174,
+                  columnNumber: 19
+                });
+              }
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 172,
+              columnNumber: 11
+            })];
           }
         }, {
           fileName: import.meta.url,
-          lineNumber: 156,
+          lineNumber: 170,
           columnNumber: 9
-        })];
+        });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 154,
+      lineNumber: 169,
       columnNumber: 7
-    })]);
-    expect(res).toBe(d`
+    })).toRenderTo(`
       class MyClass:
           pass
 
@@ -282,82 +386,96 @@ describe("Python Variable", () => {
   });
   it("declares a python variable with a class type from a different module", () => {
     const classKey = refkey();
-    const res = toSourceTextMultiple([_$createComponent(py.SourceFile, {
-      path: "classes.py",
+    expect(_$createComponent(TestOutputDirectory, {
       get children() {
-        return _$createComponent(py.ClassDeclaration, {
-          name: "MyClass",
-          refkey: classKey
+        return [_$createComponent(py.SourceFile, {
+          path: "classes.py",
+          get children() {
+            return _$createComponent(py.ClassDeclaration, {
+              name: "MyClass",
+              refkey: classKey
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 192,
+              columnNumber: 11
+            });
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 173,
+          lineNumber: 191,
           columnNumber: 9
-        });
-      }
-    }, {
-      fileName: import.meta.url,
-      lineNumber: 172,
-      columnNumber: 7
-    }), _$createComponent(py.SourceFile, {
-      path: "usage.py",
-      get children() {
-        return _$createComponent(py.VariableDeclaration, {
-          name: "my_var",
-          type: classKey
+        }), _$createComponent(py.SourceFile, {
+          path: "usage.py",
+          get children() {
+            return _$createComponent(py.VariableDeclaration, {
+              name: "my_var",
+              type: classKey
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 195,
+              columnNumber: 11
+            });
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 176,
+          lineNumber: 194,
           columnNumber: 9
-        });
+        })];
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 175,
+      lineNumber: 190,
       columnNumber: 7
-    })]);
-    assertFileContents(res, {
+    })).toRenderTo({
       "classes.py": `
-        class MyClass:
-            pass
+          class MyClass:
+              pass
 
-      `,
+        `,
       "usage.py": `
-        from typing import TYPE_CHECKING
+          from typing import TYPE_CHECKING
 
-        if TYPE_CHECKING:
-            from classes import MyClass
+          if TYPE_CHECKING:
+              from classes import MyClass
 
-        my_var: MyClass = None
-      `
+          my_var: MyClass = None
+        `
     });
   });
   it("declares a python variable receiving other variable as value", () => {
     const varKey = refkey();
-    const res = toSourceText([_$createComponent(py.StatementList, {
+    expect(_$createComponent(TestOutput, {
       get children() {
-        return [_$createComponent(py.VariableDeclaration, {
-          name: "my_var",
-          refkey: varKey,
-          initializer: 42
-        }, {
-          fileName: import.meta.url,
-          lineNumber: 200,
-          columnNumber: 9
-        }), _$createComponent(py.VariableDeclaration, {
-          name: "my_other_var",
-          initializer: varKey
+        return _$createComponent(py.StatementList, {
+          get children() {
+            return [_$createComponent(py.VariableDeclaration, {
+              name: "my_var",
+              refkey: varKey,
+              initializer: 42
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 220,
+              columnNumber: 11
+            }), _$createComponent(py.VariableDeclaration, {
+              name: "my_other_var",
+              initializer: varKey
+            }, {
+              fileName: import.meta.url,
+              lineNumber: 225,
+              columnNumber: 11
+            })];
+          }
         }, {
           fileName: import.meta.url,
-          lineNumber: 205,
+          lineNumber: 219,
           columnNumber: 9
-        })];
+        });
       }
     }, {
       fileName: import.meta.url,
-      lineNumber: 199,
+      lineNumber: 218,
       columnNumber: 7
-    })]);
-    expect(res).toBe(`my_var = 42\nmy_other_var = my_var`);
+    })).toRenderTo("my_var = 42\nmy_other_var = my_var");
   });
 });
 //# sourceMappingURL=variables.test.js.map