Do not test captcha recognition by default

jgriss · Jun 22, 2024 · 6f62f7e · 6f62f7e
1 parent 81e6fc3
commit 6f62f7e
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -5,4 +5,5 @@ tests/credentials.json
 dist
 *.egg-info
 __pycache__
-NEW FEATURES.md
+NEW FEATURES.md
+build
diff --git a/tests/test_onnx.py b/tests/test_onnx.py
@@ -13,6 +13,12 @@
 
 class TestInference(unittest.TestCase):
     def test_inference(self):
+        # only run if set
+        run_tests = os.getenv("TEST_ONNX", False)
+
+        if not run_tests:
+            return
+
         """Test if the inference works on the test image"""
         solver = Solver(os.path.join(currentdir, "../models/captcha_huawei.onnx"), device=["CPUExecutionProvider"])
         result = solver.solve_captcha(cv2.imread(os.path.join(currentdir, "test_img.png")))