Invoke tests without exec

tests/test_builder.py

@@ -103,9 +103,7 @@ def index(expr):
     assert index(Reads(a0)) == 2
 
 
-if __name__ == "__main__":
-    if len(sys.argv) > 1:
-        exec(sys.argv[1])
-    else:
-        from pytest import main
-        main([__file__])
+if __name__ == '__main__':
+    test_saxpy_builder()
+    test_query()
+    test_build_with_queries()

tests/test_linalg.py

@@ -167,5 +167,15 @@ def test_matvec():
 }""")
 
 
-from pytest import main
-    main([__file__])
+if __name__ == '__main__':
+    test_batch_matmul()
+    test_conv()
+    test_copy()
+    test_dot()
+    test_fill()
+    test_generic()
+    test_indexed_generic()
+    test_reduce()
+    test_view()
+    test_matmul()
+    test_matvec()

tests/test_scf.py

@@ -50,5 +50,7 @@ def test_scf_while():
 }""")
 
 
-from pytest import main
-    main([__file__])
+if __name__ == '__main__':
+    test_scf_for()
+    test_scf_if()
+    test_scf_while()