test all params upd

CAREamics · Jan 22, 2025 · 0346fff · 0346fff
1 parent 6de8e56
commit 0346fff
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/test_careamist.py b/tests/test_careamist.py
@@ -1144,10 +1144,12 @@ def test_all_parameters_used(tmp_path, minimum_n2v_configuration):
     config.training_config.num_epochs = 2
     config.data_config.batch_size = 2
     config.data_config.patch_size = (16, 16)
+    config.data_config.dataloader_params = {"num_workers": 2}
 
     # TODO add other params ?
     careamist = CAREamist(config, work_dir=tmp_path)
 
     assert careamist.cfg.training_config.num_epochs == 2
     assert careamist.cfg.data_config.batch_size == 2
     assert all(i == 16 for i in careamist.cfg.data_config.patch_size)
+    assert careamist.cfg.data_config.dataloader_params["num_workers"] == 2