updated tests

juglab · Oct 10, 2024 · ff18206 · ff18206
1 parent 14af2b8
commit ff18206
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 10 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -10,12 +10,12 @@ markers = [
 
 
 [tool.black]
-line-length = 88
+line-length = 90
 target-version = ['py38', 'py39', 'py310']
 
 
 [tool.ruff]
-line-length = 88
+line-length = 90
 select = [
     "E", "F", "W", #flake8
     "UP", # pyupgrade

diff --git a/tests/model_adapter_tests/test_dino_adapter.py b/tests/model_adapter_tests/test_dino_adapter.py
@@ -61,10 +61,10 @@ def test_mock_adapter(test_patch: np.ndarray):
 @pytest.mark.parametrize(
     "test_image, expected_output_shape, expected_slices",
     [
-        (np.ones((256, 256)), (121, 28, 28, 384), 1),  # 2D
-        (np.ones((256, 256, 3)), (121, 28, 28, 384), 1),  # 2D RGB
-        (np.ones((2, 256, 256)), (121, 28, 28, 384), 2),  # 3D
-        (np.ones((2, 256, 256, 3)), (121, 28, 28, 384), 2),  # 3D RGB
+        (np.ones((256, 256)), (49, 42, 42, 384), 1),  # 2D
+        (np.ones((256, 256, 3)), (49, 42, 42, 384), 1),  # 2D RGB
+        (np.ones((2, 256, 256)), (49, 42, 42, 384), 2),  # 3D
+        (np.ones((2, 256, 256, 3)), (49, 42, 42, 384), 2),  # 3D RGB
     ],
 )
 def test_dino_embedding_extraction(

diff --git a/tests/model_adapter_tests/test_mobilesam_adapter.py b/tests/model_adapter_tests/test_mobilesam_adapter.py
@@ -68,10 +68,10 @@ def test_mock_adapter(test_patch: np.ndarray):
 @pytest.mark.parametrize(
     "test_image, expected_output_shape, expected_slices",
     [
-        (np.ones((256, 256)), (25, 64, 64, 320), 1),  # 2D
-        (np.ones((256, 256, 3)), (25, 64, 64, 320), 1),  # 2D RGB
-        (np.ones((2, 256, 256)), (25, 64, 64, 320), 2),  # 3D
-        (np.ones((2, 256, 256, 3)), (25, 64, 64, 320), 2)  # 3D RGB
+        (np.ones((256, 256)), (9, 128, 128, 320), 1),  # 2D
+        (np.ones((256, 256, 3)), (9, 128, 128, 320), 1),  # 2D RGB
+        (np.ones((2, 256, 256)), (9, 128, 128, 320), 2),  # 3D
+        (np.ones((2, 256, 256, 3)), (9, 128, 128, 320), 2)  # 3D RGB
     ],
 )
 def test_mobilesam_embedding_extraction(