[tests] add utility to replace direct call to pytest.main

Author: gigiblender

python/tvm/testing/utils.py

@@ -62,6 +62,7 @@ def test_something():
 `TVM_TEST_TARGETS` environment variable in the CI.
 
 """
+import inspect
 import copy
 import copyreg
 import ctypes
@@ -1513,3 +1514,8 @@ def identity_after(x, sleep):
 def terminate_self():
     """Testing function to terminate the process."""
     sys.exit(-1)
+
+
+def main():
+    test_file = inspect.getsourcefile(sys._getframe(1))
+    sys.exit(pytest.main([test_file] + sys.argv[1:]))

tests/micro/arduino/test_arduino_error_detection.py

@@ -21,6 +21,7 @@
 from tvm.micro.project_api.server import ServerError
 
 import test_utils
+import tvm.testing
 
 # A new project and workspace dir is created for EVERY test
 @pytest.fixture
@@ -46,4 +47,4 @@ def test_bugged_project_compile_fails(workspace_dir, project):
 
 
 if __name__ == "__main__":
-    sys.exit(pytest.main([__file__] + sys.argv[1:]))
+    tvm.testing.main()

tests/micro/arduino/test_arduino_rpc_server.py

@@ -367,4 +367,6 @@ def test_tensors(sess):
 
 
 if __name__ == "__main__":
-    sys.exit(pytest.main([__file__] + sys.argv[1:]))
+    import tvm.testing
+
+    tvm.testing.main()

tests/micro/arduino/test_arduino_workflow.py

@@ -218,4 +218,6 @@ def test_project_inference_runtime(serial_output):
 
 
 if __name__ == "__main__":
-    sys.exit(pytest.main([__file__] + sys.argv[1:]))
+    import tvm.testing
+
+    tvm.testing.main()

tests/micro/common/test_tvmc.py

@@ -217,4 +217,6 @@ def test_tvmc_model_run(board, output_dir):
 
 
 if __name__ == "__main__":
-    sys.exit(pytest.main([__file__] + sys.argv[1:]))
+    import tvm.testing
+
+    tvm.testing.main()

tests/micro/zephyr/test_zephyr.py

@@ -504,4 +504,6 @@ def test_autotune_conv2d(temp_dir, board, west_cmd, tvm_debug):
 
 
 if __name__ == "__main__":
-    sys.exit(pytest.main([__file__] + sys.argv[1:]))
+    import tvm.testing
+
+    tvm.testing.main()

tests/micro/zephyr/test_zephyr_aot.py

@@ -135,4 +135,4 @@ def test_qemu_make_fail(temp_dir, board, west_cmd, tvm_debug):
 
 
 if __name__ == "__main__":
-    sys.exit(pytest.main([__file__] + sys.argv[1:]))
+    tvm.testing.main()

tests/micro/zephyr/test_zephyr_armv7m.py

@@ -187,4 +187,4 @@ def test_armv7m_intrinsic(temp_dir, board, west_cmd, tvm_debug):
 
 
 if __name__ == "__main__":
-    sys.exit(pytest.main([__file__] + sys.argv[1:]))
+    tvm.testing.main()

tests/python/ci/test_ci.py

@@ -732,4 +732,6 @@ def run(type, data, check):
 
 
 if __name__ == "__main__":
-    sys.exit(pytest.main([__file__] + sys.argv[1:]))
+    import tvm.testing
+
+    tvm.testing.main()

tests/python/contrib/test_cmsisnn/test_binary_ops.py

@@ -275,4 +275,4 @@ def test_invalid_parameters(
 
 
 if __name__ == "__main__":
-    sys.exit(pytest.main([__file__] + sys.argv[1:]))
+    tvm.testing.main()