replace dims with sizes in io too

Author: SarahAlidoost

sarxarray/_io.py

@@ -45,7 +45,7 @@ def from_dataset(ds: xr.Dataset) -> xr.Dataset:
         The input dataset should have the following variables: `('real', 'imag')`.
     """
     # Check ds should have the following dimensions: (azimuth, range, time)
-    if any(dim not in ds.dims for dim in ["azimuth", "range", "time"]):
+    if any(dim not in list(ds.sizes) for dim in ["azimuth", "range", "time"]):
         raise ValueError(
             "The input dataset should have three dimensions: (azimuth, range, time)."
         )

tests/test_io.py

@@ -26,7 +26,7 @@ def test_from_dataset_normal(self):
             f"{os.path.dirname(__file__)}/data/zarrs/slcs_example.zarr"
         )
         slcs = sarxarray.from_dataset(test_ds)
-        assert all(dim in slcs.dims for dim in ["azimuth", "range", "time"])
+        assert all(dim in list(slcs.sizes) for dim in ["azimuth", "range", "time"])
         assert all(var not in slcs.variables.keys() for var in ["real", "imag"])
         assert all(
             var in slcs.variables.keys() for var in ["complex", "amplitude", "phase"]
@@ -60,7 +60,7 @@ def test_loading_vars_and_coords(self, test_slcs):
         assert set(["azimuth", "range", "time"]).issubset(
             [k for k in stack.coords.keys()]
         )
-        assert stack.dims == {"azimuth": 100, "range": 100, "time": 2}
+        assert stack.sizes == {"azimuth": 100, "range": 100, "time": 2}
 
     def test_loading_custom_var(self, test_slcs):
         """When other var names specified, do not compute amplitude and range"""
@@ -75,7 +75,7 @@ def test_loading_custom_var(self, test_slcs):
         assert not (
             set(["amplitude", "phase"]).issubset([k for k in stack.coords.keys()])
         )
-        assert stack.dims == {"azimuth": 100, "range": 100, "time": 2}
+        assert stack.sizes == {"azimuth": 100, "range": 100, "time": 2}
 
     def test_loading_chunksizes(self, test_slcs):
         stack = sarxarray.from_binary(
@@ -114,7 +114,7 @@ def test_loading_one_slc(self, test_slcs):
         assert set(["azimuth", "range", "time"]).issubset(
             [k for k in stack.coords.keys()]
         )
-        assert stack.dims == {"azimuth": 100, "range": 100, "time": 1}
+        assert stack.sizes == {"azimuth": 100, "range": 100, "time": 1}
 
 
 class TestUtils: