add tests for regrid

tests/test_converter.py

@@ -1,4 +1,4 @@
-"""Unit test for regridding."""
+"""Unit test for converter."""
 
 from pathlib import Path
 import numpy as np

tests/test_datasets/test_eth_canopy_height.py

@@ -47,7 +47,7 @@ def test_download(self, mock_urlretrieve):
                 assert json_dict["variable_names"] == variable
 
     def ingest_dummy_data(self, temp_dir):
-        """Ïngest dummy tif data to nc for other tests."""
+        """Ingest dummy tif data to nc for other tests."""
         canopy_height_dataset = eth_canopy_height.EthCanopyHeight()
         canopy_height_dataset.ingest(
             download_dir=data_folder, ingest_dir=Path(temp_dir)
@@ -72,8 +72,7 @@ def test_ingest(self):
     def test_load(self):
         """Test load function."""
         with TemporaryDirectory() as temp_dir:
-            ds, canopy_height_dataset = self.ingest_dummy_data(temp_dir)
-            ds.close()  # make windows happy, otherwise gives PermissionError on CI/CD
+            _, canopy_height_dataset = self.ingest_dummy_data(temp_dir)
 
             times = TimeBounds(np.datetime64("2020-01-01"), np.datetime64("2020-12-31"))
             bbox = SpatialBounds(54, 6, 51, 3)
@@ -99,10 +98,7 @@ def test_convert(self):
         """Test convert function."""
         with TemporaryDirectory() as temp_dir:
             _, canopy_height_dataset = self.ingest_dummy_data(temp_dir)
-            canopy_height_dataset.convert(
-                ingest_dir=Path(temp_dir),
-                convention="ALMA"
-            )
+            canopy_height_dataset.convert(ingest_dir=Path(temp_dir), convention="ALMA")
             # TODO: finish this test when the function is complete.
 
 

tests/test_regrid.py

@@ -0,0 +1,95 @@
+"""Unit test for regridding."""
+
+from pathlib import Path
+import numpy as np
+import pytest
+from zampy.datasets.dataset_protocol import SpatialBounds
+from zampy.datasets.eth_canopy_height import parse_tiff_file
+from zampy.utils import regrid
+
+
+path_dummy_data = Path(__file__).resolve().parent / "test_data" / "eth-canopy-height"
+
+
+@pytest.fixture
+def dummy_dataset():
+    ds = parse_tiff_file(
+        path_dummy_data / "ETH_GlobalCanopyHeight_10m_2020_N51E003_Map.tif",
+    )
+    return ds
+
+
+def test_infer_resolution(dummy_dataset):
+    """Test resolution inferring function."""
+    (resolution_lat, resolution_lon) = regrid.infer_resolution(dummy_dataset)
+    expected_lat = 0.01
+    expected_lon = 0.01
+
+    assert resolution_lat == pytest.approx(expected_lat, 0.001)
+    assert resolution_lon == pytest.approx(expected_lon, 0.001)
+
+
+def test_flox_regrid_4x_courser(dummy_dataset):
+    """Test flox regridding at a 4x courser resolution
+
+    Note that the native resolution is about lat/lon=0.01.
+    """
+    bbox = SpatialBounds(54, 6, 51, 3)
+    ds = regrid.flox_regrid(data=dummy_dataset, spatial_bounds=bbox, resolution=1.0)
+    expected_lat = [51.0, 52.0, 53.0, 54.0]
+    expected_lon = [3.0, 4.0, 5.0, 6.0]
+
+    np.testing.assert_allclose(ds.latitude.values, expected_lat)
+    np.testing.assert_allclose(ds.longitude.values, expected_lon)
+
+
+def test_flox_regrid_4x_finer(dummy_dataset):
+    """Test flox regridding at a 4x finer resolution
+
+    Note that the native resolution is about lat/lon=0.01.
+    """
+    bbox = SpatialBounds(51.02, 3.02, 51, 3)
+    ds = regrid.flox_regrid(data=dummy_dataset, spatial_bounds=bbox, resolution=0.002)
+    # only compare first 5 index
+    expected_lat = [51.0, 51.002, 51.004, 51.006, 51.008]
+    expected_lon = [3.0, 3.002, 3.004, 3.006, 3.008]
+
+    np.testing.assert_allclose(ds.latitude.values[:5], expected_lat)
+    np.testing.assert_allclose(ds.longitude.values[:5], expected_lon)
+
+
+def test_flox_regrid_close(dummy_dataset):
+    """Test flox regridding at a resolution close to native.
+
+    Note that the native resolution is about lat/lon=0.01.
+    """
+    bbox = SpatialBounds(51.1, 3.1, 51, 3)
+    ds = regrid.flox_regrid(data=dummy_dataset, spatial_bounds=bbox, resolution=0.02)
+    expected_lat = [51.0, 51.02, 51.04, 51.06, 51.08, 51.1]
+    expected_lon = [3.0, 3.02, 3.04, 3.06, 3.08, 3.1]
+
+    np.testing.assert_allclose(ds.latitude.values, expected_lat)
+    np.testing.assert_allclose(ds.longitude.values, expected_lon)
+
+
+def test_regrid_data_flox(dummy_dataset):
+    bbox = SpatialBounds(54, 6, 51, 3)
+    ds = regrid.regrid_data(
+        data=dummy_dataset, spatial_bounds=bbox, resolution=1.0, method="flox"
+    )
+    expected_lat = [51.0, 52.0, 53.0, 54.0]
+    expected_lon = [3.0, 4.0, 5.0, 6.0]
+
+    np.testing.assert_allclose(ds.latitude.values, expected_lat)
+    np.testing.assert_allclose(ds.longitude.values, expected_lon)
+
+
+def test_regrid_data_unknown_method(dummy_dataset):
+    bbox = SpatialBounds(54, 6, 51, 3)
+    with pytest.raises(ValueError):
+        regrid.regrid_data(
+            data=dummy_dataset,
+            spatial_bounds=bbox,
+            resolution=1.0,
+            method="fake_method",
+        )