tests: update tests

Author: okunator

cellseg_models_pytorch/utils/tests/test_filehandler.py

@@ -1,7 +1,7 @@
 from cellseg_models_pytorch.utils import FileHandler
 
 
-def test_filehandler(inst_map, type_map) -> None:
+def test_filehandler_get_gson(inst_map, type_map) -> None:
     split_dict = FileHandler.get_offset("x-5000_y-3000")
     assert "x" in split_dict.keys() and "y" in split_dict.keys()
     assert 3000 in split_dict.values() and 5000 in split_dict.values()
@@ -10,3 +10,9 @@ def test_filehandler(inst_map, type_map) -> None:
         inst_map, type_map, classes={"a": 0, "b": 1, "c": 2, "d": 3}
     )
     assert len(gson) == 7
+
+
+def test_filehandler_read_h5(hdf5db):
+    arrs = FileHandler.read_h5_patch(hdf5db, ix=1, return_type=True, return_sem=True)
+
+    assert arrs["image"].any()

cellseg_models_pytorch/utils/tests/test_seg_benchmark.py

@@ -1,6 +1,6 @@
 import pytest
 
-from cellseg_models_pytorch.utils.seg_benchmark import BenchMarker
+from cellseg_models_pytorch.utils.seg_benchmark import SegBenchmarker
 
 classes = {
     "0": 0,
@@ -14,10 +14,16 @@
 
 
 @pytest.mark.parametrize("how", ["binary", "multi", None])
-def test_sem_seg_bm(mask_patch_dir, how):
-    bm = BenchMarker(
-        pred_dir=mask_patch_dir,
-        true_dir=mask_patch_dir,
+@pytest.mark.parametrize("inputf", ["h5", "folder"])
+def test_sem_seg_bm(mask_patch_dir, hdf5db, how, inputf):
+    if inputf == "folder":
+        inpath = mask_patch_dir
+    else:
+        inpath = hdf5db
+
+    bm = SegBenchmarker(
+        true_path=inpath,
+        pred_path=inpath,
         type_classes=classes,
         sem_classes=classes,
     )