Testing fixes

Due to Singleton Trainer widget

Author: C-Achard

…seg3d/_tests/test_supervised_training.py …cellseg3d/_tests/test_training_plugin.pynapari_cellseg3d/_tests/test_supervised_training.py renamed to napari_cellseg3d/_tests/test_training_plugin.py napari_cellseg3d/_tests/test_supervised_training.py renamed to napari_cellseg3d/_tests/test_training_plugin.py

@@ -13,9 +13,10 @@
 im_path_str = str(im_path)
 
 
-def test_create_supervised_worker_from_config(make_napari_viewer_proxy):
+def test_worker_configs(make_napari_viewer_proxy):
     viewer = make_napari_viewer_proxy()
     widget = Trainer(viewer=viewer)
+    # test supervised config and worker
     widget.device_choice.setCurrentIndex(0)
     widget.model_choice.setCurrentIndex(0)
     widget._toggle_unsupervised_mode(enabled=False)
@@ -34,6 +35,36 @@ def test_create_supervised_worker_from_config(make_napari_viewer_proxy):
             assert getattr(default_config, attr) == getattr(
                 worker.config, attr
             )
+    # test unsupervised config and worker
+    widget.model_choice.setCurrentText("WNet")
+    widget._toggle_unsupervised_mode(enabled=True)
+    default_config = config.WNetTrainingWorkerConfig()
+    worker = widget._create_worker(additional_results_description="TEST_1")
+    excluded = ["results_path_folder", "sample_size", "weights_info"]
+    for attr in dir(default_config):
+        if not attr.startswith("__") and attr not in excluded:
+            assert getattr(default_config, attr) == getattr(
+                worker.config, attr
+            )
+    widget.unsupervised_images_filewidget.text_field.setText(
+        str(im_path.parent)
+    )
+    widget.data = widget.create_dataset_dict_no_labs()
+    worker = widget._create_worker(additional_results_description="TEST_2")
+    dataloader, eval_dataloader, data_shape = worker._get_data()
+    assert eval_dataloader is None
+    assert data_shape == (6, 6, 6)
+
+    widget.images_filepaths = [str(im_path)]
+    widget.labels_filepaths = [str(im_path)]
+    # widget.unsupervised_eval_data = widget.create_train_dataset_dict()
+    worker = widget._create_worker(additional_results_description="TEST_3")
+    dataloader, eval_dataloader, data_shape = worker._get_data()
+    assert widget.unsupervised_eval_data is not None
+    assert eval_dataloader is not None
+    assert widget.unsupervised_eval_data[0]["image"] is not None
+    assert widget.unsupervised_eval_data[0]["label"] is not None
+    assert data_shape == (6, 6, 6)
 
 
 def test_update_loss_plot(make_napari_viewer_proxy):
@@ -86,8 +117,8 @@ def test_training(make_napari_viewer_proxy, qtbot):
     widget.log = LogFixture()
     viewer.window.add_dock_widget(widget)
 
-    widget.images_filepath = None
-    widget.labels_filepaths = None
+    widget.images_filepath = []
+    widget.labels_filepaths = []
 
     assert not widget.check_ready()
 

napari_cellseg3d/_tests/test_unsup_training.py

@@ -431,7 +431,11 @@ def check_ready(self):
             * False and displays a warning if not
 
         """
-        if self.images_filepaths == [] and self.labels_filepaths != []:
+        if (
+            self.images_filepaths == []
+            or self.labels_filepaths == []
+            or len(self.images_filepaths) != len(self.labels_filepaths)
+        ):
             logger.warning("Image and label paths are not correctly set")
             return False
         return True