diff --git a/python/infinitensor/test_import_onnx.py b/python/infinitensor/test_import_onnx.py
deleted file mode 100644
index 581fb206..00000000
--- a/python/infinitensor/test_import_onnx.py
+++ /dev/null
@@ -1,12 +0,0 @@
-from pyinfinitensor import *
-from import_onnx import *
-
-import sys
-
-def main(netPath):
-    runtime = CpuRuntimeObj.getInstance()
-    graphBuilder = GraphBuilderObj(runtime)
-    import_onnx(graphBuilder, netPath)
-
-if __name__ == "__main__":
-    main(sys.argv[1])
diff --git a/python/test/run_test.py b/python/test/run_test.py
new file mode 100644
index 00000000..1521c421
--- /dev/null
+++ b/python/test/run_test.py
@@ -0,0 +1,5 @@
+import pytest
+
+
+if __name__ == "__main__":
+    retcode = pytest.main()
diff --git a/python/test/test_import_onnx.py b/python/test/test_import_onnx.py
new file mode 100644
index 00000000..de4d6914
--- /dev/null
+++ b/python/test/test_import_onnx.py
@@ -0,0 +1,9 @@
+from pyinfinitensor import *
+from infinitensor import import_onnx
+
+
+class Test_ImportOnnx:
+    def test_Netname(self):
+        runtime = CpuRuntimeObj.getInstance()
+        graphBuilder = GraphBuilderObj(runtime)
+        import_onnx(graphBuilder, '/path/to/net')