Skip tests if tensorflow not available

CCSI-Toolset · Sep 6, 2023 · a6061f1 · a6061f1
1 parent bc5c7df
commit a6061f1
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/foqus_lib/gui/tests/test_surrogate.py b/foqus_lib/gui/tests/test_surrogate.py
@@ -18,6 +18,14 @@
 import pytest
 
 
+try:
+    import tensorflow
+except ModuleNotFoundError:
+    tensorflow_available = False
+else:
+    tensorflow_available = True
+
+
 pytestmark = pytest.mark.gui
 
 
@@ -52,6 +60,7 @@ class TestFrame:
     #         # "ACOSSO",
     #     ]
     # )
+    @pytest.mark.skipif(not tensorflow_available, reason="tensorflow not available")
     def test_run_surrogate(
         self, qtbot, frame, main_window: mainWindow, name: str = "keras_nn"
     ):