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

package/test/decoratorlist.test.tsx

@@ -1,7 +1,6 @@
-import { d } from "@alloy-js/core/testing";
 import { describe, expect, it } from "vitest";
 import { DecoratorList } from "../src/components/DecoratorList.jsx";
-import { toSourceText } from "./utils.js";
+import { TestOutput } from "./utils.js";
 
 /**
  * Direct contract tests for `DecoratorList`. The component is exercised
@@ -12,84 +11,104 @@ import { toSourceText } from "./utils.js";
 describe("DecoratorList", () => {
   it("renders nothing when decorators is undefined", () => {
     expect(
-      toSourceText([
-        <>
-          <DecoratorList />
-          pass
-        </>,
-      ]),
+      <TestOutput>
+        {[
+          <>
+            <DecoratorList />
+            pass
+          </>,
+        ]}
+      </TestOutput>,
     ).toRenderTo("pass");
   });
 
   it("renders nothing when decorators is an empty array", () => {
     expect(
-      toSourceText([
-        <>
-          <DecoratorList decorators={[]} />
-          pass
-        </>,
-      ]),
+      <TestOutput>
+        {[
+          <>
+            <DecoratorList decorators={[]} />
+            pass
+          </>,
+        ]}
+      </TestOutput>,
     ).toRenderTo("pass");
   });
 
   it("renders a single decorator followed by exactly one hardline", () => {
     expect(
-      toSourceText([
-        <>
-          <DecoratorList decorators={["@one"]} />
-          pass
-        </>,
-      ]),
-    ).toRenderTo(d`
-      @one
-      pass
-    `);
+      <TestOutput>
+        {[
+          <>
+            <DecoratorList decorators={["@one"]} />
+            pass
+          </>,
+        ]}
+      </TestOutput>,
+    ).toRenderTo(
+      `
+        @one
+        pass
+      `,
+    );
   });
 
   it("renders multiple decorators with no blank lines between adjacent entries", () => {
     expect(
-      toSourceText([
-        <>
-          <DecoratorList decorators={["@one", "@two", "@three"]} />
-          pass
-        </>,
-      ]),
-    ).toRenderTo(d`
-      @one
-      @two
-      @three
-      pass
-    `);
+      <TestOutput>
+        {[
+          <>
+            <DecoratorList decorators={["@one", "@two", "@three"]} />
+            pass
+          </>,
+        ]}
+      </TestOutput>,
+    ).toRenderTo(
+      `
+        @one
+        @two
+        @three
+        pass
+      `,
+    );
   });
 
   it("preserves source order: first entry is topmost (= applied last)", () => {
     expect(
-      toSourceText([
-        <>
-          <DecoratorList decorators={["@outer", "@middle", "@inner"]} />
-          pass
-        </>,
-      ]),
-    ).toRenderTo(d`
-      @outer
-      @middle
-      @inner
-      pass
-    `);
+      <TestOutput>
+        {[
+          <>
+            <DecoratorList decorators={["@outer", "@middle", "@inner"]} />
+            pass
+          </>,
+        ]}
+      </TestOutput>,
+    ).toRenderTo(
+      `
+        @outer
+        @middle
+        @inner
+        pass
+      `,
+    );
   });
 
   it("skips falsy entries without emitting blank lines", () => {
     expect(
-      toSourceText([
-        <>
-          <DecoratorList decorators={["@one", false, undefined, "@two"]} />
-          pass
-        </>,
-      ]),
-    ).toRenderTo(d`
-      @one
-      @two
-      pass
-    `);
+      <TestOutput>
+        {[
+          <>
+            <DecoratorList decorators={["@one", false, undefined, "@two"]} />
+            pass
+          </>,
+        ]}
+      </TestOutput>,
+    ).toRenderTo(
+      `
+        @one
+        @two
+        pass
+      `,
+    );
   });
 });

package/test/dundermethoddeclaration.test.tsx

@@ -1,7 +1,6 @@
-import { d } from "@alloy-js/core/testing";
 import { describe, expect, it } from "vitest";
 import * as py from "../src/index.js";
-import { toSourceText } from "./utils.js";
+import { TestOutput } from "./utils.js";
 
 describe("Dunder/Constructor Declarations", () => {
   it("renders dunder methods with parameters", () => {
@@ -19,7 +18,7 @@ describe("Dunder/Constructor Declarations", () => {
       </py.ClassDeclaration>
     );
 
-    expect(toSourceText([decl])).toBe(d`
+    expect(<TestOutput>{decl}</TestOutput>).toRenderTo(`
       class MyClass:
           def __init__(self, x: int):
               self.attribute = "value"
@@ -42,7 +41,7 @@ describe("Dunder/Constructor Declarations", () => {
       </py.ClassDeclaration>
     );
 
-    expect(toSourceText([decl])).toBe(d`
+    expect(<TestOutput>{decl}</TestOutput>).toRenderTo(`
       class MyClass:
           def __new__(cls, *args, **kwargs):
               pass

package/test/enums.test.tsx

@@ -1,14 +1,13 @@
 import { refkey } from "@alloy-js/core";
-import { d } from "@alloy-js/core/testing";
 import { describe, expect, it } from "vitest";
 import { enumModule } from "../src/builtins/python.js";
 import * as py from "../src/index.js";
-import { toSourceText } from "./utils.jsx";
+import { TestOutput } from "./utils.js";
 
 describe("Python Enum", () => {
   it("renders class-level decorators above ClassEnumDeclaration", () => {
-    const result = toSourceText(
-      [
+    expect(
+      <TestOutput externals={[enumModule]}>
         <py.ClassEnumDeclaration
           name="Color"
           baseType="IntEnum"
@@ -17,12 +16,10 @@ describe("Python Enum", () => {
             { name: "RED", value: 1 },
             { name: "GREEN", value: 2 },
           ]}
-        />,
-      ],
-      { externals: [enumModule] },
-    );
-
-    expect(result).toRenderTo(d`
+        />
+      </TestOutput>,
+    ).toRenderTo(
+      `
       from enum import IntEnum
 
 
@@ -31,13 +28,13 @@ describe("Python Enum", () => {
           RED = 1
           GREEN = 2
 
-
-    `);
+    `,
+    );
   });
 
   it("class enum with explicit values", () => {
-    const result = toSourceText(
-      [
+    expect(
+      <TestOutput externals={[enumModule]}>
         <py.ClassEnumDeclaration
           name="Color"
           baseType="IntEnum"
@@ -46,11 +43,10 @@ describe("Python Enum", () => {
             { name: "GREEN", value: "2" },
             { name: "BLUE", value: "3" },
           ]}
-        />,
-      ],
-      { externals: [enumModule] },
-    );
-    const expected = d`
+        />
+      </TestOutput>,
+    ).toRenderTo(
+      `
       from enum import IntEnum
 
 
@@ -59,14 +55,13 @@ describe("Python Enum", () => {
           GREEN = 2
           BLUE = 3
 
-        
-    `;
-    expect(result).toRenderTo(expected);
+    `,
+    );
   });
 
   it("class enum with jsValues", () => {
-    const result = toSourceText(
-      [
+    expect(
+      <TestOutput externals={[enumModule]}>
         <py.ClassEnumDeclaration
           name="Color"
           baseType="IntEnum"
@@ -75,11 +70,10 @@ describe("Python Enum", () => {
             { name: "GREEN", jsValue: 2 },
             { name: "BLUE", jsValue: "3" },
           ]}
-        />,
-      ],
-      { externals: [enumModule] },
-    );
-    const expected = d`
+        />
+      </TestOutput>,
+    ).toRenderTo(
+      `
       from enum import IntEnum
 
 
@@ -88,14 +82,13 @@ describe("Python Enum", () => {
           GREEN = 2
           BLUE = "3"
 
-
-    `;
-    expect(result).toRenderTo(expected);
+    `,
+    );
   });
 
   it("class enum with a refkey as jsValue", () => {
-    const result = toSourceText(
-      [
+    expect(
+      <TestOutput externals={[enumModule]}>
         <py.StatementList>
           <py.ClassDeclaration name="Dog" refkey={refkey("Dog")} />
           <py.ClassDeclaration name="Cat" refkey={refkey("Cat")} />
@@ -107,11 +100,10 @@ describe("Python Enum", () => {
               { name: "CAT", value: refkey("Cat") },
             ]}
           />
-        </py.StatementList>,
-      ],
-      { externals: [enumModule] },
-    );
-    const expected = d`
+        </py.StatementList>
+      </TestOutput>,
+    ).toRenderTo(
+      `
       from enum import Enum
 
 
@@ -125,23 +117,21 @@ describe("Python Enum", () => {
           DOG = Dog
           CAT = Cat
 
-
-    `;
-    expect(result).toRenderTo(expected);
+    `,
+    );
   });
 
   it("class enum with auto() values", () => {
-    const result = toSourceText(
-      [
+    expect(
+      <TestOutput externals={[enumModule]}>
         <py.ClassEnumDeclaration
           name="Animal"
           auto
           members={[{ name: "DOG" }, { name: "CAT" }, { name: "RABBIT" }]}
-        />,
-      ],
-      { externals: [enumModule] },
-    );
-    const expected = d`
+        />
+      </TestOutput>,
+    ).toRenderTo(
+      `
       from enum import auto
       from enum import Enum
 
@@ -151,14 +141,13 @@ describe("Python Enum", () => {
           CAT = auto()
           RABBIT = auto()
 
-
-    `;
-    expect(result).toRenderTo(expected);
+    `,
+    );
   });
 
   it("class enum with mixed manual and auto() values", () => {
-    const result = toSourceText(
-      [
+    expect(
+      <TestOutput externals={[enumModule]}>
         <py.ClassEnumDeclaration
           name="Permission"
           baseType="Flag"
@@ -168,12 +157,10 @@ describe("Python Enum", () => {
             { name: "WRITE" },
             { name: "EXECUTE" },
           ]}
-        />,
-      ],
-      { externals: [enumModule] },
-    );
-
-    const expected = d`
+        />
+      </TestOutput>,
+    ).toRenderTo(
+      `
       from enum import auto
       from enum import Flag
 
@@ -183,14 +170,13 @@ describe("Python Enum", () => {
           WRITE = auto()
           EXECUTE = auto()
 
-
-    `;
-    expect(result).toRenderTo(expected);
+    `,
+    );
   });
 
   it("functional enum with list", () => {
-    const result = toSourceText(
-      [
+    expect(
+      <TestOutput externals={[enumModule]}>
         <py.FunctionalEnumDeclaration
           name="Direction"
           members={[
@@ -199,22 +185,21 @@ describe("Python Enum", () => {
             { name: "EAST" },
             { name: "WEST" },
           ]}
-        />,
-      ],
-      { externals: [enumModule] },
-    );
-    const expected = d`
+        />
+      </TestOutput>,
+    ).toRenderTo(
+      `
       from enum import Enum
 
 
       Direction = Enum('Direction', ['NORTH', 'SOUTH', 'EAST', 'WEST'])
-    `;
-    expect(result).toRenderTo(expected);
+    `,
+    );
   });
 
   it("functional enum with mapping", () => {
-    const result = toSourceText(
-      [
+    expect(
+      <TestOutput externals={[enumModule]}>
         <py.FunctionalEnumDeclaration
           name="Priority"
           members={[
@@ -222,16 +207,15 @@ describe("Python Enum", () => {
             { name: "MEDIUM", value: 2 },
             { name: "LOW", value: 3 },
           ]}
-        />,
-      ],
-      { externals: [enumModule] },
-    );
-    const expected = d`
+        />
+      </TestOutput>,
+    ).toRenderTo(
+      `
       from enum import Enum
 
 
       Priority = Enum('Priority', {'HIGH' : 1, 'MEDIUM' : 2, 'LOW' : 3})
-    `;
-    expect(result).toRenderTo(expected);
+    `,
+    );
   });
 });

package/test/externals.test.tsx

@@ -1,9 +1,8 @@
 import { code, refkey } from "@alloy-js/core";
-import { d } from "@alloy-js/core/testing";
 import { expect, it } from "vitest";
 import * as py from "../src/index.js";
 import { createModule } from "../src/index.js";
-import { toSourceText } from "./utils.js";
+import { TestOutput } from "./utils.js";
 
 it("uses import from external library", () => {
   const requestsLib = createModule({
@@ -14,19 +13,18 @@ it("uses import from external library", () => {
       "models.anothermodule": ["something"],
     },
   });
-  const result = toSourceText(
-    [
+  expect(
+    <TestOutput externals={[requestsLib]}>
       <py.StatementList>
         {requestsLib["."].get}
         {requestsLib["."].post}
         {requestsLib["models"].Request}
         {requestsLib["models"].Response}
         {requestsLib["models.anothermodule"].something}
-      </py.StatementList>,
-    ],
-    { externals: [requestsLib] },
-  );
-  const expected = d`
+      </py.StatementList>
+    </TestOutput>,
+  ).toRenderTo(
+    `
     from requests import get
     from requests import post
     from requests.models import Request
@@ -38,8 +36,8 @@ it("uses import from external library", () => {
     Request
     Response
     something
-  `;
-  expect(result).toRenderTo(expected);
+  `,
+  );
 });
 
 it("uses import from external library in multiple functions", () => {
@@ -86,10 +84,12 @@ it("uses import from external library in multiple functions", () => {
     </py.FunctionDeclaration>,
   ];
 
-  const result = toSourceText(functionDeclarations, {
-    externals: [py.requestsModule],
-  });
-  const expected = d`
+  expect(
+    <TestOutput externals={[py.requestsModule]}>
+      {functionDeclarations}
+    </TestOutput>,
+  ).toRenderTo(
+    `
     from requests import get
     from requests import post
     from typing import TYPE_CHECKING
@@ -107,9 +107,8 @@ it("uses import from external library in multiple functions", () => {
         response = post(1)
         return response.json()
 
-
-  `;
-  expect(result).toRenderTo(expected);
+  `,
+  );
 });
 
 it("uses import from external library in multiple class methods", () => {
@@ -166,10 +165,12 @@ it("uses import from external library in multiple class methods", () => {
     </py.ClassDeclaration>,
   ];
 
-  const result = toSourceText(functionDeclarations, {
-    externals: [py.requestsModule],
-  });
-  const expected = d`
+  expect(
+    <TestOutput externals={[py.requestsModule]}>
+      {functionDeclarations}
+    </TestOutput>,
+  ).toRenderTo(
+    `
     from requests import get
     from requests import post
     from typing import TYPE_CHECKING
@@ -189,7 +190,6 @@ it("uses import from external library in multiple class methods", () => {
             return response.json()
 
 
-
-  `;
-  expect(result).toRenderTo(expected);
+  `,
+  );
 });