Repurpose lib-rt/setup.py

Author: ilevkivskyi

mypyc/build.py

@@ -544,6 +544,7 @@ def mypycify(
                    mypyc_trace.txt (derived from executed operations). This is
                    useful for performance analysis, such as analyzing which
                    primitive ops are used the most and on which lines.
+        depends_on_native_internal: This is True only for mypy itself.
         install_native_libs: If True, also build the native extension modules. Normally,
                              those are build and published on PyPI separately, but during
                              tests, we want to use their development versions (i.e. from

mypyc/lib-rt/setup.py

@@ -44,28 +44,52 @@ def run(self):
         return build_ext.run(self)
 
 
-setup(
-    name="test_capi",
-    version="0.1",
-    ext_modules=[
-        Extension(
-            "test_capi",
-            [
-                "test_capi.cc",
-                "init.c",
-                "int_ops.c",
-                "float_ops.c",
-                "list_ops.c",
-                "exc_ops.c",
-                "generic_ops.c",
-                "pythonsupport.c",
-            ],
-            depends=["CPy.h", "mypyc_util.h", "pythonsupport.h"],
-            extra_compile_args=["-Wno-unused-function", "-Wno-sign-compare"] + compile_args,
-            libraries=["gtest"],
-            include_dirs=["../external/googletest", "../external/googletest/include"],
-            **kwargs,
-        )
-    ],
-    cmdclass={"build_ext": build_ext_custom},
-)
+if "--run-capi-tests" in sys.argv:
+    sys.argv.pop()
+    setup(
+        name="test_capi",
+        version="0.1",
+        ext_modules=[
+            Extension(
+                "test_capi",
+                [
+                    "test_capi.cc",
+                    "init.c",
+                    "int_ops.c",
+                    "float_ops.c",
+                    "list_ops.c",
+                    "exc_ops.c",
+                    "generic_ops.c",
+                    "pythonsupport.c",
+                ],
+                depends=["CPy.h", "mypyc_util.h", "pythonsupport.h"],
+                extra_compile_args=["-Wno-unused-function", "-Wno-sign-compare"] + compile_args,
+                libraries=["gtest"],
+                include_dirs=["../external/googletest", "../external/googletest/include"],
+                **kwargs,
+            )
+        ],
+        cmdclass={"build_ext": build_ext_custom},
+    )
+else:
+    # TODO: this is a stub, we need a way to share common build logic with
+    # mypyc/build.py without code duplication.
+    setup(
+        name="mypy-native",
+        version="0.0.1",
+        ext_modules=[
+            Extension(
+                "native_internal",
+                [
+                    "native_internal.c",
+                    "init.c",
+                    "int_ops.c",
+                    "exc_ops.c",
+                    "pythonsupport.c",
+                ],
+                include_dirs=["."],
+                **kwargs,
+            )
+        ],
+        cmdclass={"build_ext": build_ext_custom},
+    )

mypyc/test/test_external.py

@@ -34,6 +34,7 @@ def test_c_unit_test(self) -> None:
                     "build_ext",
                     f"--build-lib={tmpdir}",
                     f"--build-temp={tmpdir}",
+                    "--run-capi-tests",
                 ],
                 env=env,
                 cwd=os.path.join(base_dir, "mypyc", "lib-rt"),