fix tests

Author: Jiaqi-Lv

tests/models/test_arch_mapde.py

@@ -49,7 +49,7 @@ def test_functionality(remote_sample: Callable) -> None:
     batch = torch.from_numpy(patch)[None]
     output = model.infer_batch(model, batch, device=select_device(on_gpu=ON_GPU))
     output = model.postproc(output[0])
-    xs, ys, _, _ = NucleusDetector._centroid_maps_to_detection_arrays(output, None)
+    xs, ys, _, _ = NucleusDetector._extract_detection_arrays_from_block(output, None)
 
     np.testing.assert_array_equal(xs[0:2], np.array([242, 192]))
     np.testing.assert_array_equal(ys[0:2], np.array([10, 13]))
@@ -73,7 +73,7 @@ def test_functionality(remote_sample: Callable) -> None:
         }
     }
     output = model.postproc(output[0], block_info=block_info)
-    xs, ys, _, _ = NucleusDetector._centroid_maps_to_detection_arrays(output, None)
+    xs, ys, _, _ = NucleusDetector._extract_detection_arrays_from_block(output, None)
     np.testing.assert_array_equal(xs, np.array([]))
     np.testing.assert_array_equal(ys, np.array([]))
 

tests/models/test_arch_sccnn.py

@@ -49,7 +49,7 @@ def test_functionality(remote_sample: Callable) -> None:
         device=select_device(on_gpu=env_detection.has_gpu()),
     )
     output = model.postproc(output[0])
-    xs, ys, _, _ = NucleusDetector._centroid_maps_to_detection_arrays(output, None)
+    xs, ys, _, _ = NucleusDetector._extract_detection_arrays_from_block(output, None)
 
     np.testing.assert_array_equal(xs, np.array([8]))
     np.testing.assert_array_equal(ys, np.array([7]))
@@ -66,7 +66,7 @@ def test_functionality(remote_sample: Callable) -> None:
         }
     }
     output = model.postproc(output[0], block_info=block_info)
-    xs, ys, _, _ = NucleusDetector._centroid_maps_to_detection_arrays(output, None)
+    xs, ys, _, _ = NucleusDetector._extract_detection_arrays_from_block(output, None)
     np.testing.assert_array_equal(xs, np.array([7]))
     np.testing.assert_array_equal(ys, np.array([8]))
 
@@ -85,6 +85,6 @@ def test_functionality(remote_sample: Callable) -> None:
         }
     }
     output = model.postproc(output[0], block_info=block_info)
-    xs, ys, _, _ = NucleusDetector._centroid_maps_to_detection_arrays(output, None)
+    xs, ys, _, _ = NucleusDetector._extract_detection_arrays_from_block(output, None)
     np.testing.assert_array_equal(xs, np.array([]))
     np.testing.assert_array_equal(ys, np.array([]))