fix a broken test

pydata · Jul 13, 2024 · 07a52ae · 07a52ae
1 parent 602c3cf
commit 07a52ae
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/xarray/tests/test_concat.py b/xarray/tests/test_concat.py
@@ -515,7 +515,7 @@ def test_concat_merge_variables_present_in_some_datasets(self, data) -> None:
         assert_identical(expected, actual)
 
     @pytest.mark.parametrize("data", [False], indirect=["data"])
-    def test_concat_2(self, data) -> None:
+    def test_concat_2(self, data: Dataset) -> None:
         dim = "dim2"
         datasets = [g.squeeze(dim) for _, g in data.groupby(dim, squeeze=False)]
         concat_over = [k for k, v in data.coords.items() if dim in v.dims and k != dim]

diff --git a/xarray/tests/test_dataset.py b/xarray/tests/test_dataset.py
@@ -581,7 +581,8 @@ def test_constructor_pandas_single(self) -> None:
             pandas_obj = a.to_pandas()
             ds_based_on_pandas = Dataset(pandas_obj)  # type: ignore  # TODO: improve typing of __init__
             for dim in ds_based_on_pandas.data_vars:
-                assert_array_equal(ds_based_on_pandas[dim], pandas_obj[str(dim)])
+                assert isinstance(dim, int)
+                assert_array_equal(ds_based_on_pandas[dim], pandas_obj[dim])
 
     def test_constructor_compat(self) -> None:
         data = {"x": DataArray(0, coords={"y": 1}), "y": ("z", [1, 1, 1])}