[Tests] Replace pytest.main with tvm.testing.main

This ensures that if you want to run a specific test script then at least it's reasonably consistent and as people copy test files they'll use the new function 😸

Author: Mousius

tests/micro/stm32/test_code_emitter.py

@@ -392,4 +392,4 @@ def test_mnist():
 
 
 if __name__ == "__main__":
-    sys.exit(pytest.main([os.path.dirname(__file__)] + sys.argv[1:]))
+    tvm.testing.main()

tests/python/contrib/test_cublas.py

@@ -378,4 +378,4 @@ def test_relay_cublas_dense(n, m, k, in_dtype, out_dtype):
 
 
 if __name__ == "__main__":
-    pytest.main([__file__])
+    tvm.testing.main()

tests/python/contrib/test_ethosu/cascader/test_ethosu_binary_elementwise_matcher.py

@@ -175,4 +175,4 @@ def test_ethosu_binary_elementwise_matcher(
 
 
 if __name__ == "__main__":
-    pytest.main([__file__])
+    tvm.testing.main()

tests/python/contrib/test_ethosu/cascader/test_ethosu_block_config.py

@@ -457,4 +457,4 @@ def test_force_block_config_elementwise(ofm_layout, block_config_str, expected_b
 
 
 if __name__ == "__main__":
-    pytest.main([__file__])
+    tvm.testing.main()

tests/python/contrib/test_ethosu/cascader/test_ethosu_conv2d_matcher.py

@@ -179,4 +179,4 @@ def test_ethosu_conv2d_block_config_from_matcher(
 
 
 if __name__ == "__main__":
-    pytest.main([__file__])
+    tvm.testing.main()

tests/python/contrib/test_ethosu/cascader/test_ethosu_depthwise2d_matcher.py

@@ -100,4 +100,4 @@ def test_ethosu_depthwise2d_matcher(kernel, stride, dilation, padding, ifm_layou
 
 
 if __name__ == "__main__":
-    pytest.main([__file__])
+    tvm.testing.main()

tests/python/contrib/test_ethosu/cascader/test_ethosu_identity_matcher.py

@@ -55,4 +55,4 @@ def test_ethosu_identity_matcher():
 
 
 if __name__ == "__main__":
-    pytest.main([__file__])
+    tvm.testing.main()

tests/python/contrib/test_ethosu/cascader/test_ethosu_inline_matcher.py

@@ -47,4 +47,4 @@ def test_ethosu_inline_matcher():
 
 
 if __name__ == "__main__":
-    pytest.main([__file__])
+    tvm.testing.main()

tests/python/contrib/test_ethosu/cascader/test_ethosu_part.py

@@ -57,4 +57,4 @@ def test_ethosu_part():
 
 
 if __name__ == "__main__":
-    pytest.main([__file__])
+    tvm.testing.main()

tests/python/contrib/test_ethosu/cascader/test_ethosu_part_performance.py

@@ -231,4 +231,4 @@ def test_conv_performance(
 
 
 if __name__ == "__main__":
-    pytest.main([__file__])
+    tvm.testing.main()