From 3c77fa0d7fb4cc6965d4a1339faef1f91db68916 Mon Sep 17 00:00:00 2001 From: youben11 Date: Tue, 10 Dec 2024 11:38:49 +0100 Subject: [PATCH] test(frontend): test TFHE-rs ML example --- .../concrete-python/tests/execution/test_examples.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/frontends/concrete-python/tests/execution/test_examples.py b/frontends/concrete-python/tests/execution/test_examples.py index 37e9493758..fbc0beed68 100644 --- 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