Major fixes and new features

venv/lib/python3.12/site-packages/mypyc/test/test_cheader.py

@@ -0,0 +1,45 @@
+"""Test that C functions used in primitives are declared in a header such as CPy.h."""
+
+from __future__ import annotations
+
+import glob
+import os
+import re
+import unittest
+
+from mypyc.primitives import registry
+from mypyc.primitives.registry import CFunctionDescription
+
+
+class TestHeaderInclusion(unittest.TestCase):
+    def test_primitives_included_in_header(self) -> None:
+        base_dir = os.path.join(os.path.dirname(__file__), "..", "lib-rt")
+        with open(os.path.join(base_dir, "CPy.h")) as f:
+            header = f.read()
+        with open(os.path.join(base_dir, "pythonsupport.h")) as f:
+            header += f.read()
+
+        def check_name(name: str) -> None:
+            if name.startswith("CPy"):
+                assert re.search(
+                    rf"\b{name}\b", header
+                ), f'"{name}" is used in mypyc.primitives but not declared in CPy.h'
+
+        for values in [
+            registry.method_call_ops.values(),
+            registry.function_ops.values(),
+            registry.binary_ops.values(),
+            registry.unary_ops.values(),
+        ]:
+            for ops in values:
+                if isinstance(ops, CFunctionDescription):
+                    ops = [ops]
+                for op in ops:
+                    check_name(op.c_function_name)
+
+        primitives_path = os.path.join(os.path.dirname(__file__), "..", "primitives")
+        for fnam in glob.glob(f"{primitives_path}/*.py"):
+            with open(fnam) as f:
+                content = f.read()
+            for name in re.findall(r'c_function_name=["\'](CPy[A-Z_a-z0-9]+)', content):
+                check_name(name)