diff --git a/tests/test_spatial.py b/tests/test_spatial.py index eb1c6522..db89704e 100644 --- a/tests/test_spatial.py +++ b/tests/test_spatial.py @@ -35,8 +35,8 @@ class TestCreepFill: [(1, "clip"), (2, "clip"), (3, "clip"), (1, "wrap"), (2, "wrap"), (3, "wrap")], ) def test_n(self, n, mode): - w = xs.creep_weights(self.ds["mask"], n=n, mode=mode) - out = xs.creep_fill(self.ds["tas"], w) + w = xs.spatial.creep_weights(self.ds["mask"], n=n, mode=mode) + out = xs.spatial.creep_fill(self.ds["tas"], w) if mode == "clip": neighbours_0 = { @@ -135,11 +135,11 @@ def test_n(self, n, mode): def test_wrong_mode(self): with pytest.raises(ValueError, match="mode must be either"): - xs.creep_weights(self.ds["mask"], n=1, mode="wrong") + xs.spatial.creep_weights(self.ds["mask"], n=1, mode="wrong") def test_n0(self): - w = xs.creep_weights(self.ds["mask"], n=0, mode="clip") - out = xs.creep_fill(self.ds["tas"], w) + w = xs.spatial.creep_weights(self.ds["mask"], n=0, mode="clip") + out = xs.spatial.creep_fill(self.ds["tas"], w) np.testing.assert_equal(out.isel(lat=0, lon=0), np.tile(np.nan, 3)) np.testing.assert_equal(out.isel(lat=3, lon=3), np.tile(np.nan, 3)) @@ -167,7 +167,9 @@ class TestSubset: ) def test_subset_gridpoint(self, kwargs, name): with pytest.warns(UserWarning, match="tile_buffer is not used"): - out = xs.subset(self.ds, "gridpoint", name=name, tile_buffer=5, **kwargs) + out = xs.spatial.subset( + self.ds, "gridpoint", name=name, tile_buffer=5, **kwargs + ) if isinstance(kwargs["lon"], list): expected = { @@ -212,9 +214,9 @@ def test_subset_bboxshape(self, kwargs, tile_buffer, method): with pytest.raises( ValueError, match="Both tile_buffer and clisops' buffer were requested." ): - xs.subset(self.ds, method, tile_buffer=tile_buffer, **kwargs) + xs.spatial.subset(self.ds, method, tile_buffer=tile_buffer, **kwargs) else: - out = xs.subset(self.ds, method, tile_buffer=tile_buffer, **kwargs) + out = xs.spatial.subset(self.ds, method, tile_buffer=tile_buffer, **kwargs) assert f"{method} spatial subsetting" in out.attrs["history"] if tile_buffer: @@ -254,8 +256,8 @@ def test_subset_sel(self): ) with pytest.raises(KeyError): - xs.subset(ds, "sel", lon=[-75, -70], lat=[-5, 0]) - out = xs.subset(ds, "sel", rlon=[-5, 5], rlat=[0, 3]) + xs.spatial.subset(ds, "sel", lon=[-75, -70], lat=[-5, 0]) + out = xs.spatial.subset(ds, "sel", rlon=[-5, 5], rlat=[0, 3]) assert "sel subsetting" in out.attrs["history"] np.testing.assert_array_equal(out["rlon"], np.arange(-5, 6)) @@ -264,14 +266,14 @@ def test_subset_sel(self): def test_history(self): ds = self.ds.copy(deep=True) ds.attrs["history"] = "this is previous history" - out = xs.subset(ds, "gridpoint", lon=-70, lat=45) + out = xs.spatial.subset(ds, "gridpoint", lon=-70, lat=45) assert "this is previous history" in out.attrs["history"].split("\n")[1] assert "gridpoint spatial subsetting" in out.attrs["history"].split("\n")[0] def test_subset_wrong_method(self): with pytest.raises(ValueError, match="Subsetting type not recognized"): - xs.subset(self.ds, "wrong", lon=-70, lat=45) + xs.spatial.subset(self.ds, "wrong", lon=-70, lat=45) def test_dask_coords():