chore: wip

Author: 34j

pyproject.toml

@@ -38,6 +38,7 @@ scripts.array-api = "array_api.cli:app"
 
 [dependency-groups]
 dev = [
+  "array-api-strict>=2.3.1",
   "pytest>=8,<9",
   "pytest-cov>=6,<7",
 ]

src/array_api/cli/_main.py

@@ -150,31 +150,6 @@ def _attributes_to_protocol(
     )
 
 
-def generate_all(
-    cache_dir: Path | str = ".cache",
-    out_path: Path | str = "src/array-api",
-    out_name: str = "_namespace.py",
-) -> None:
-    import subprocess as sp
-
-    Path(cache_dir).mkdir(exist_ok=True)
-    sp.run(["git", "clone", "https://github.com/data-apis/array-api", ".cache"])
-
-    for dir_path in (Path(cache_dir) / Path("src") / "array_api_stubs").iterdir():
-        if not dir_path.is_dir():
-            continue
-        # get module bodies
-        body_module = {
-            path.stem: ast.parse(
-                path.read_text("utf-8")
-                .replace("Dtype", "dtype")
-                .replace("Device", "device")
-            ).body
-            for path in dir_path.rglob("*.py")
-        }
-        generate(body_module, Path(out_path) / dir_path.name / out_name)
-
-
 def generate(body_module: dict[str, list[ast.stmt]], out_path: Path) -> None:
     body_typevars = body_module["_types"]
     del body_module["__init__"]
@@ -309,3 +284,27 @@ def generate(body_module: dict[str, list[ast.stmt]], out_path: Path) -> None:
     )
     out_path.parent.mkdir(parents=True, exist_ok=True)
     out_path.write_text(text, "utf-8")
+
+
+def generate_all(
+    cache_dir: Path | str = ".cache",
+    out_path: Path | str = "src/array_api",
+) -> None:
+    import subprocess as sp
+
+    Path(cache_dir).mkdir(exist_ok=True)
+    sp.run(["git", "clone", "https://github.com/data-apis/array-api", ".cache"])
+
+    for dir_path in (Path(cache_dir) / Path("src") / "array_api_stubs").iterdir():
+        if not dir_path.is_dir():
+            continue
+        # get module bodies
+        body_module = {
+            path.stem: ast.parse(
+                path.read_text("utf-8")
+                .replace("Dtype", "dtype")
+                .replace("Device", "device")
+            ).body
+            for path in dir_path.rglob("*.py")
+        }
+        generate(body_module, (Path(out_path) / dir_path.name).with_suffix(".py"))

tests/test_main.py

@@ -1,6 +1,5 @@
-from array_api.main import add
-
-
-def test_add():
-    """Adding two number works as expected."""
-    assert add(1, 1) == 2
+from array_api._2024_12 import ArrayNamespace, add
+import array_api_strict
+def test_main():
+    assert isinstance(array_api_strict.add, add)
+    assert isinstance(array_api_strict, ArrayNamespace)