test(frontend): test TFHE-rs ML example

zama-ai · Dec 10, 2024 · 3c77fa0 · 3c77fa0
1 parent a6f5e09
commit 3c77fa0
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/frontends/concrete-python/tests/execution/test_examples.py b/frontends/concrete-python/tests/execution/test_examples.py
@@ -408,9 +408,15 @@ def test_game_of_life(implementation, dimension, sample_input_output_pairs, help
 
 
 def test_tfhers_example():
-    path_to_test_script = tfhers_utils = (
-        f"{os.path.dirname(os.path.abspath(__file__))}/../../examples/tfhers/"
-    )
+    path_to_test_script = f"{os.path.dirname(os.path.abspath(__file__))}/../../examples/tfhers/"
+    test_script_filename = "test.sh"
+    assert (
+        os.system(f"cd {path_to_test_script} && sh {test_script_filename}") == 0
+    ), "test script failed"
+
+
+def test_tfhers_ml_example():
+    path_to_test_script = f"{os.path.dirname(os.path.abspath(__file__))}/../../examples/tfhers-ml/"
     test_script_filename = "test.sh"
     assert (
         os.system(f"cd {path_to_test_script} && sh {test_script_filename}") == 0