tests: skip when torch not available

DC-analysis · Jun 12, 2024 · d91976f · d91976f
1 parent 0ed4cf1
commit d91976f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/tests/test_gui.py b/tests/test_gui.py
@@ -84,6 +84,7 @@ def test_gui_correct_offset(mw, correct_offset):
 
 
 def test_gui_segm_torch_model(mw, qtbot, monkeypatch):
+    pytest.importorskip("torch")
     path = retrieve_data(
         "fmt-hdf5_cytoshot_full-features_legacy_allev_2023.zip")
     path_model = retrieve_model(

diff --git a/tests/test_gui_manager.py b/tests/test_gui_manager.py
@@ -1,5 +1,7 @@
 import shutil
 
+import pytest
+
 from dcnum.meta import ppid
 
 from chipstream.gui import manager
@@ -92,6 +94,7 @@ def test_manager_run_defaults():
 
 
 def test_manager_run_error_wrong_model():
+    pytest.importorskip("torch")
     model_file = retrieve_model(
         "segm-torch-model_unet-dcnum-test_g1_910c2.zip")