Merge pull request #749 from InfiniTensor/issue/723

issue/723 - Expose test framework to other packages

Author: wooway777

setup.py

@@ -1,15 +1,34 @@
 import subprocess
-
-from setuptools import setup
+from setuptools import setup, find_packages
 from setuptools.command.build import build
 
+def run_xmake_build():
+    print("Running xmake build...")
+    subprocess.run(["xmake", "build"], check=True)
+    subprocess.run(["xmake", "install"], check=True)
+    subprocess.run(["xmake", "build", "-y", "_infinicore"], check=True)
+    subprocess.run(["xmake", "install", "_infinicore"], check=True)
 
 class Build(build):
     def run(self):
-        subprocess.run(["xmake", "build"])
-        subprocess.run(["xmake", "install"])
-        subprocess.run(["xmake", "build", "-y", "_infinicore"])
-        subprocess.run(["xmake", "install", "_infinicore"])
-
+        run_xmake_build()
+        super().run()
 
-setup(package_dir={"": "python"}, cmdclass={"build": Build})
+setup(
+    # 1. Find main packages and manually add test/framework packages
+    packages=find_packages(where="python") + [
+        "infinicore.test", 
+        "infinicore.test.framework"
+    ],
+    
+    # 2. Directory mappings
+    package_dir={
+        "": "python",  # Root package is under python/ directory
+        "infinicore.test": "test/infinicore"  # Intermediate package mapping
+    },
+    
+    # 3. Register commands
+    cmdclass={
+        "build": Build
+    }
+)