test(CoordinatesFilter): mock coordinates filters (#28)

geospaitial-lab · Apr 29, 2024 · ee7c839 · ee7c839
1 parent ca0b5c1
commit ee7c839
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 12 deletions.
diff --git a/src/data/tests/conftest.py b/src/data/tests/conftest.py
@@ -1,12 +1,13 @@
 from pathlib import Path
-from unittest.mock import patch
+from unittest.mock import MagicMock, patch
 
 import geopandas as gpd
 import numpy as np
 import pytest
 
 from src.data.coordinates_filter import (
     CompositeFilter,
+    CoordinatesFilter,
     DuplicatesFilter,
     GeospatialFilter,
     MaskFilter,
@@ -25,12 +26,10 @@
 
 @pytest.fixture(scope='session')
 def composite_filter() -> CompositeFilter:
-    mask = np.array([0, 1, 0, 1], dtype=np.bool_)
     coordinates_filters = [
-        DuplicatesFilter(),
-        MaskFilter(
-            mask=mask,
-        ),
+        MagicMock(spec=CoordinatesFilter),
+        MagicMock(spec=CoordinatesFilter),
+        MagicMock(spec=CoordinatesFilter),
     ]
     return CompositeFilter(
         coordinates_filters=coordinates_filters,

diff --git a/src/data/tests/test_coordinates_filter.py b/src/data/tests/test_coordinates_filter.py
@@ -1,11 +1,12 @@
-from unittest.mock import patch
+from unittest.mock import MagicMock, patch
 
 import geopandas as gpd
 import geopandas.testing
 import numpy as np
 
 from src.data.coordinates_filter import (
     CompositeFilter,
+    CoordinatesFilter,
     DuplicatesFilter,
     GeospatialFilter,
     MaskFilter,
@@ -18,12 +19,10 @@
 
 
 def test_init_composite_filter() -> None:
-    mask = np.array([0, 1, 0, 1], dtype=np.bool_)
     coordinates_filters = [
-        DuplicatesFilter(),
-        MaskFilter(
-            mask=mask,
-        ),
+        MagicMock(spec=CoordinatesFilter),
+        MagicMock(spec=CoordinatesFilter),
+        MagicMock(spec=CoordinatesFilter),
     ]
     composite_filter = CompositeFilter(
         coordinates_filters=coordinates_filters,