@alloy-js/python 0.0.1

package/dist/test/namepolicies.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=namepolicies.test.d.ts.map

package/dist/test/namepolicies.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"namepolicies.test.d.ts","sourceRoot":"","sources":["../../test/namepolicies.test.tsx"],"names":[],"mappings":""}

package/dist/test/namepolicies.test.js

@@ -0,0 +1,109 @@
+import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
+import { d } from "@alloy-js/core/testing";
+import { expect, it } from "vitest";
+import { enumModule } from "../src/builtins/python.js";
+import * as py from "../src/index.js";
+import { toSourceText } from "./utils.js";
+it("correct formatting of class name", () => {
+  const result = toSourceText([_$createComponent(py.ClassDeclaration, {
+    name: "a-really-WeirdClass-name"
+  })]);
+  const expected = d`
+    class AReallyWeirdClassName:
+        pass
+
+      
+  `;
+  expect(result).toRenderTo(expected);
+});
+it("correct formatting of Enum name and EnumMember names", () => {
+  const result = toSourceText([_$createComponent(py.EnumDeclaration, {
+    name: "priority",
+    style: "functional",
+    members: [{
+      name: "high",
+      value: 1
+    }, {
+      name: "Medium",
+      value: 2
+    }, {
+      name: "lowValue",
+      value: 3
+    }]
+  })], {
+    externals: [enumModule]
+  });
+  const expected = d`
+    from enum import Enum
+
+    Priority = Enum('Priority', {'HIGH' : 1, 'MEDIUM' : 2, 'LOW_VALUE' : 3})
+  `;
+  expect(result).toRenderTo(expected);
+});
+it("renders a function with parameters", () => {
+  const result = toSourceText([_$createComponent(py.FunctionDeclaration, {
+    name: "quirklyNamed-Function",
+    parameters: [{
+      name: "a-parameter",
+      type: "int"
+    }],
+    args: true,
+    kwargs: true,
+    children: "print(x, y)"
+  })]);
+  expect(result).toRenderTo(d`
+      def quirkly_named_function(a_parameter: int, *args, **kwargs):
+          print(x, y)
+
+        
+    `);
+});
+it("correct formatting of call signature parameters names", () => {
+  const result = toSourceText([_$createComponent(py.CallSignatureParameters, {
+    parameters: [{
+      name: "this-is-a-number",
+      type: "int"
+    }, {
+      name: "andThisIsADict",
+      type: "dict"
+    }]
+  })]);
+  expect(result).toRenderTo(`this_is_a_number: int, and_this_is_a_dict: dict`);
+});
+it("correct formatting of call statement vars", () => {
+  const result = toSourceText([_$createComponent(py.StatementList, {
+    get children() {
+      return _$createComponent(py.ClassInstantiation, {
+        target: "test",
+        get args() {
+          return [_$createComponent(py.VariableDeclaration, {
+            name: "this-is-a-long-name",
+            get initializer() {
+              return _$createComponent(py.Atom, {
+                jsValue: "A name"
+              });
+            },
+            callStatementVar: true
+          }), _$createComponent(py.VariableDeclaration, {
+            name: "andThisIsANumber",
+            get initializer() {
+              return _$createComponent(py.Atom, {
+                jsValue: 42
+              });
+            },
+            callStatementVar: true
+          })];
+        }
+      });
+    }
+  })]);
+  expect(result).toRenderTo(`test(this_is_a_long_name="A name", and_this_is_a_number=42)`);
+});
+it("correct formatting of variable name", () => {
+  const res = toSourceText([_$createComponent(py.VariableDeclaration, {
+    name: "myVar",
+    type: "int",
+    initializer: 42
+  })]);
+  expect(res).toBe(`my_var: int = 42`);
+});

package/dist/test/pydocs.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=pydocs.test.d.ts.map

package/dist/test/pydocs.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"pydocs.test.d.ts","sourceRoot":"","sources":["../../test/pydocs.test.tsx"],"names":[],"mappings":""}

package/dist/test/pydocs.test.js

@@ -0,0 +1,500 @@
+import { createComponent as _$createComponent, createIntrinsic as _$createIntrinsic } from "@alloy-js/core/jsx-runtime";
+import { Prose } 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.js";
+describe("PyDoc", () => {
+  it("formats properly", () => {
+    const res = toSourceText([_$createComponent(py.PyDoc, {
+      get children() {
+        return [_$createComponent(Prose, {
+          children: "This is an example of a long docstring that will be broken in lines. We will also render another paragraph after this one."
+        }), _$createComponent(Prose, {
+          children: "This is another paragraph, and there's a line break before it."
+        })];
+      }
+    })], {
+      printOptions: {
+        printWidth: 40
+      }
+    });
+    expect(res).toRenderTo(d`
+        """
+        This is an example of a long docstring
+        that will be broken in lines. We will
+        also render another paragraph after this
+        one.
+        
+        This is another paragraph, and there's a
+        line break before it.
+        """
+
+        
+      `);
+  });
+});
+describe("PyDocExample", () => {
+  it("creates docstring with a code sample", () => {
+    const res = toSourceText([_$createComponent(py.PyDoc, {
+      get children() {
+        return [_$createComponent(Prose, {
+          children: "This is an example of a docstring with a code sample."
+        }), _$createComponent(py.PyDocExample, {
+          children: "print(\"Hello world!\")"
+        })];
+      }
+    })], {
+      printOptions: {
+        printWidth: 40
+      }
+    });
+    expect(res).toRenderTo(d`
+      """
+      This is an example of a docstring with a
+      code sample.
+      
+      >> print("Hello world!")
+      """
+
+
+      `);
+  });
+  it("creates docstring with more than one code sample", () => {
+    const res = toSourceText([_$createComponent(py.PyDoc, {
+      get children() {
+        return [_$createComponent(Prose, {
+          children: "This is an example of a docstring with a code sample."
+        }), _$createComponent(py.PyDocExample, {
+          children: "print(\"Hello world!\")"
+        }), _$createComponent(py.PyDocExample, {
+          children: "print(\"Hello world again!\")"
+        })];
+      }
+    })], {
+      printOptions: {
+        printWidth: 40
+      }
+    });
+    expect(res).toRenderTo(d`
+      """
+      This is an example of a docstring with a
+      code sample.
+      
+      >> print("Hello world!")
+      
+      >> print("Hello world again!")
+      """
+
+
+      `);
+  });
+  it("creates docstring with a multiline code sample", () => {
+    const res = toSourceText([_$createComponent(py.PyDoc, {
+      get children() {
+        return [_$createComponent(Prose, {
+          children: "This is an example of a docstring with a code sample."
+        }), _$createComponent(py.PyDocExample, {
+          get children() {
+            return ["print(\"Hello world!\")", _$createIntrinsic("br", {}), "x = \"Hello\"", _$createIntrinsic("br", {}), "print(x)"];
+          }
+        })];
+      }
+    })], {
+      printOptions: {
+        printWidth: 40
+      }
+    });
+    expect(res).toRenderTo(d`
+      """
+      This is an example of a docstring with a
+      code sample.
+      
+      >> print("Hello world!")
+      >> x = "Hello"
+      >> print(x)
+      """
+
+
+      `);
+  });
+});
+describe("GoogleStyleDocParam", () => {
+  it("name only", () => {
+    const res = toSourceText([_$createComponent(py.PyDoc, {
+      get children() {
+        return _$createComponent(py.GoogleStyleDocParam, {
+          name: "somebody"
+        });
+      }
+    })]);
+    expect(res).toRenderTo(d`
+          """
+          somebody
+          """
+
+
+          `);
+  });
+  it("name and type", () => {
+    const res = toSourceText([_$createComponent(py.PyDoc, {
+      get children() {
+        return _$createComponent(py.GoogleStyleDocParam, {
+          name: "somebody",
+          type: "str"
+        });
+      }
+    })]);
+    expect(res).toRenderTo(d`
+          """
+          somebody (str)
+          """
+
+
+          `);
+  });
+  it("name, type and description", () => {
+    const res = toSourceText([_$createComponent(py.PyDoc, {
+      get children() {
+        return _$createComponent(py.GoogleStyleDocParam, {
+          name: "somebody",
+          type: "str",
+          children: "Somebody's name."
+        });
+      }
+    })]);
+    expect(res).toRenderTo(d`
+          """
+          somebody (str): Somebody's name.
+          """
+
+
+          `);
+  });
+  it("name, type, description, and optional", () => {
+    const res = toSourceText([_$createComponent(py.PyDoc, {
+      get children() {
+        return _$createComponent(py.GoogleStyleDocParam, {
+          name: "somebody",
+          type: "str",
+          optional: true,
+          children: "Somebody's name."
+        });
+      }
+    })]);
+    expect(res).toRenderTo(d`
+          """
+          somebody (str, optional): Somebody's name.
+          """
+
+
+          `);
+  });
+  it("name, type, description, and optional with default value", () => {
+    const res = toSourceText([_$createComponent(py.PyDoc, {
+      get children() {
+        return _$createComponent(py.GoogleStyleDocParam, {
+          name: "somebody",
+          type: "str",
+          optional: true,
+          defaultValue: "John Doe",
+          children: "Somebody's name."
+        });
+      }
+    })]);
+    expect(res).toRenderTo(d`
+        """
+        somebody (str, optional): Somebody's name. Defaults to \"John Doe\".
+        """
+
+
+        `);
+  });
+  it("name, type, description, and optional with default value with a very long description", () => {
+    const res = toSourceText([_$createComponent(py.PyDoc, {
+      get children() {
+        return [_$createComponent(py.GoogleStyleDocParam, {
+          name: "somebody",
+          type: "str",
+          optional: true,
+          defaultValue: "John Doe",
+          children: "Somebody's name. This can be any string representing a person, whether it's a first name, full name, nickname, or even a codename (e.g., \"Agent X\"). It's used primarily for display purposes, logging, or greeting messages and is not required to be unique or validated unless specified by the caller."
+        }), _$createComponent(py.GoogleStyleDocParam, {
+          name: "somebody2",
+          type: "str",
+          children: "Somebody's name. This can be any string representing a person, whether it's a first name, full name, nickname, or even a codename (e.g., \"Agent X\"). It's used primarily for display purposes, logging, or greeting messages and is not required to be unique or validated unless specified by the caller."
+        })];
+      }
+    })], {
+      printOptions: {
+        printWidth: 80
+      }
+    });
+    expect(res).toRenderTo(d`
+          """
+          somebody (str, optional): Somebody's name. This can be any string representing a
+              person, whether it's a first name, full name, nickname, or even a codename
+              (e.g., "Agent X"). It's used primarily for display purposes, logging, or
+              greeting messages and is not required to be unique or validated unless
+              specified by the caller. Defaults to \"John Doe\".
+          
+          somebody2 (str): Somebody's name. This can be any string representing a person,
+              whether it's a first name, full name, nickname, or even a codename (e.g.,
+              "Agent X"). It's used primarily for display purposes, logging, or greeting
+              messages and is not required to be unique or validated unless specified by
+              the caller.
+          """
+
+
+          `);
+  });
+  it("name, type, description, and optional with default value with a description containing a linebreak", () => {
+    const res = toSourceText([_$createComponent(py.PyDoc, {
+      get children() {
+        return _$createComponent(py.GoogleStyleDocParam, {
+          name: "somebody",
+          type: "str",
+          optional: true,
+          defaultValue: "John Doe",
+          get children() {
+            return ["Somebody's name. This is one line", _$createIntrinsic("hbr", {}), "This is another line."];
+          }
+        });
+      }
+    })], {
+      printOptions: {
+        printWidth: 80
+      }
+    });
+    expect(res).toRenderTo(d`
+          """
+          somebody (str, optional): Somebody's name. This is one line
+              This is another line. Defaults to \"John Doe\".
+          """
+
+
+          `);
+  });
+});
+describe("Full example", () => {
+  it("renders correctly in a Class", () => {
+    const doc = _$createComponent(py.ClassDoc, {
+      get description() {
+        return [_$createComponent(Prose, {
+          children: "This is an example of a long docstring that will be broken in lines. We will also render another paragraph after this one."
+        }), _$createComponent(py.PyDocExample, {
+          get children() {
+            return ["print(\"Hello world!\")", _$createIntrinsic("br", {}), "x = \"Hello\"", _$createIntrinsic("br", {}), "print(x)"];
+          }
+        })];
+      },
+      parameters: [{
+        name: "somebody",
+        type: "str",
+        optional: true,
+        default: "John Doe",
+        doc: "Somebody's name. This can be any string representing a person, whether it's a first name, full name, nickname, or even a codename (e.g., 'Agent X'). It's used primarily for display purposes, logging, or greeting messages and is not required to be unique or validated unless specified by the caller. Defaults to \"John Doe\"."
+      }, {
+        name: "somebody2",
+        type: "str",
+        doc: "Somebody's name. This can be any string representing a person, whether it's a first name, full name, nickname, or even a codename (e.g., 'Agent X'). It's used primarily for display purposes, logging, or greeting messages and is not required to be unique or validated unless specified by the caller."
+      }],
+      style: "google"
+    });
+    const res = toSourceText([_$createComponent(py.ClassDeclaration, {
+      name: "A",
+      doc: doc,
+      get children() {
+        return _$createComponent(py.StatementList, {
+          get children() {
+            return [_$createComponent(py.VariableDeclaration, {
+              name: "just_name"
+            }), _$createComponent(py.VariableDeclaration, {
+              name: "name_and_type",
+              type: "number"
+            }), _$createComponent(py.VariableDeclaration, {
+              name: "name_type_and_value",
+              type: "number",
+              initializer: 12
+            })];
+          }
+        });
+      }
+    })], {
+      printOptions: {
+        printWidth: 80,
+        tabWidth: 4
+      }
+    });
+    expect(res).toRenderTo(d`
+          class A:
+              """
+              This is an example of a long docstring that will be broken in lines. We will
+              also render another paragraph after this one.
+
+              >> print("Hello world!")
+              >> x = "Hello"
+              >> print(x)
+
+              Args:
+                  somebody (str, optional): Somebody's name. This can be any string
+                      representing a person, whether it's a first name, full name,
+                      nickname, or even a codename (e.g., 'Agent X'). It's used primarily
+                      for display purposes, logging, or greeting messages and is not
+                      required to be unique or validated unless specified by the caller.
+                      Defaults to \"John Doe\".
+
+                  somebody2 (str): Somebody's name. This can be any string representing a
+                      person, whether it's a first name, full name, nickname, or even a
+                      codename (e.g., 'Agent X'). It's used primarily for display
+                      purposes, logging, or greeting messages and is not required to be
+                      unique or validated unless specified by the caller.
+              """
+              just_name = None
+              name_and_type: number = None
+              name_type_and_value: number = 12
+
+
+          `);
+  });
+  it("renders correctly in a Function", () => {
+    const doc = _$createComponent(py.FunctionDoc, {
+      get description() {
+        return [_$createComponent(Prose, {
+          children: "This is an example of a long docstring that will be broken in lines. We will also render another paragraph after this one."
+        }), _$createComponent(py.PyDocExample, {
+          get children() {
+            return ["print(\"Hello world!\")", _$createIntrinsic("br", {}), "x = \"Hello\"", _$createIntrinsic("br", {}), "print(x)"];
+          }
+        })];
+      },
+      parameters: [{
+        name: "somebody",
+        type: "str",
+        optional: true,
+        default: "John Doe",
+        doc: "Somebody's name. This can be any string representing a person, whether it's a first name, full name, nickname, or even a codename (e.g., 'Agent X'). It's used primarily for display purposes, logging, or greeting messages and is not required to be unique or validated unless specified by the caller. Defaults to \"John Doe\"."
+      }, {
+        name: "somebody2",
+        type: "str",
+        doc: "Somebody's name. This can be any string representing a person, whether it's a first name, full name, nickname, or even a codename (e.g., 'Agent X'). It's used primarily for display purposes, logging, or greeting messages and is not required to be unique or validated unless specified by the caller."
+      }],
+      returns: "The return value. True for success, False otherwise.",
+      raises: ["ValueError: If somebody2 is equal to somebody."],
+      style: "google"
+    });
+    const res = toSourceText([_$createComponent(py.FunctionDeclaration, {
+      name: "some_function",
+      doc: doc,
+      get children() {
+        return _$createComponent(py.StatementList, {
+          get children() {
+            return [_$createComponent(py.VariableDeclaration, {
+              name: "just_name"
+            }), _$createComponent(py.VariableDeclaration, {
+              name: "name_and_type",
+              type: "number"
+            }), _$createComponent(py.VariableDeclaration, {
+              name: "name_type_and_value",
+              type: "number",
+              initializer: 12
+            })];
+          }
+        });
+      }
+    })], {
+      printOptions: {
+        printWidth: 80,
+        tabWidth: 4
+      }
+    });
+    expect(res).toRenderTo(d`
+          def some_function():
+              """
+              This is an example of a long docstring that will be broken in lines. We will
+              also render another paragraph after this one.
+
+              >> print("Hello world!")
+              >> x = "Hello"
+              >> print(x)
+
+              Args:
+                  somebody (str, optional): Somebody's name. This can be any string
+                      representing a person, whether it's a first name, full name,
+                      nickname, or even a codename (e.g., 'Agent X'). It's used primarily
+                      for display purposes, logging, or greeting messages and is not
+                      required to be unique or validated unless specified by the caller.
+                      Defaults to \"John Doe\".
+
+                  somebody2 (str): Somebody's name. This can be any string representing a
+                      person, whether it's a first name, full name, nickname, or even a
+                      codename (e.g., 'Agent X'). It's used primarily for display
+                      purposes, logging, or greeting messages and is not required to be
+                      unique or validated unless specified by the caller.
+
+              Returns:
+                  The return value. True for success, False otherwise.
+
+              Raises:
+                  ValueError: If somebody2 is equal to somebody.
+              """
+              just_name = None
+              name_and_type: number = None
+              name_type_and_value: number = 12
+
+
+          `);
+  });
+  it("renders correctly in a Variable", () => {
+    const res = toSourceText([_$createComponent(py.VariableDeclaration, {
+      name: "myVar",
+      initializer: 42,
+      doc: "This is a very long docstring that will be broken in two lines when rendered. This part of the docstring will be in the second line."
+    })], {
+      printOptions: {
+        printWidth: 80,
+        tabWidth: 4
+      }
+    });
+    expect(res).toRenderTo(d`
+            # This is a very long docstring that will be broken in two lines when rendered.
+            # This part of the docstring will be in the second line.
+            my_var = 42
+            `);
+  });
+  it("classic enum with explicit values", () => {
+    const result = toSourceText([_$createComponent(py.EnumDeclaration, {
+      name: "Color",
+      baseType: "IntEnum",
+      members: [{
+        name: "RED",
+        value: "1",
+        doc: "The color red."
+      }, {
+        name: "GREEN",
+        value: "2",
+        doc: "The color green."
+      }, {
+        name: "BLUE",
+        value: "3",
+        doc: "The color blue."
+      }],
+      doc: "An enum representing colors."
+    })], {
+      externals: [enumModule]
+    });
+    const expected = d`
+      from enum import IntEnum
+
+      # An enum representing colors.
+      class Color(IntEnum):
+          RED = 1  # The color red.
+          GREEN = 2  # The color green.
+          BLUE = 3  # The color blue.
+
+
+    `;
+    expect(result).toRenderTo(expected);
+  });
+});

package/dist/test/references.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=references.test.d.ts.map

package/dist/test/references.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"references.test.d.ts","sourceRoot":"","sources":["../../test/references.test.tsx"],"names":[],"mappings":""}

package/dist/test/references.test.js

@@ -0,0 +1,49 @@
+import { createComponent as _$createComponent } from "@alloy-js/core/jsx-runtime";
+import { refkey } from "@alloy-js/core";
+import { describe, it } from "vitest";
+import * as py from "../src/index.js";
+import { assertFileContents, toSourceTextMultiple } from "./utils.js";
+describe("Reference", () => {
+  it("Verifies that a reference is correctly resolved", () => {
+    const rk1 = refkey();
+    const res = toSourceTextMultiple([_$createComponent(py.SourceFile, {
+      path: "models.py",
+      get children() {
+        return _$createComponent(py.ClassDeclaration, {
+          name: "User",
+          refkey: rk1
+        });
+      }
+    }), _$createComponent(py.SourceFile, {
+      path: "services.py",
+      get children() {
+        return _$createComponent(py.VariableDeclaration, {
+          name: "current_user",
+          get type() {
+            return _$createComponent(py.Reference, {
+              refkey: rk1
+            });
+          },
+          get initializer() {
+            return _$createComponent(py.ClassInstantiation, {
+              target: "User",
+              args: ['"Marvin"']
+            });
+          }
+        });
+      }
+    })]);
+    assertFileContents(res, {
+      "models.py": `
+        class User:
+            pass
+
+        `,
+      "services.py": `
+        from models import User
+
+        current_user: User = User("Marvin")
+        `
+    });
+  });
+});

package/dist/test/sourcefiles.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=sourcefiles.test.d.ts.map

package/dist/test/sourcefiles.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"sourcefiles.test.d.ts","sourceRoot":"","sources":["../../test/sourcefiles.test.tsx"],"names":[],"mappings":""}