precommit

Author: DHRUVA KUMAR KAUSHAL

xarray/tests/test_dataset.py

@@ -1141,29 +1141,28 @@ def test_chunks_does_not_load_data(self) -> None:
         [(True, "standard"), (False, "standard"), (True, "noleap"), (True, "360_day")],
     )
     def test_chunk_by_season_resampler(self, use_cftime: bool, calendar: str) -> None:
-        # With 2 years of data starting Jan 1, we get 9 seasonal chunks:
-        # partial DJF (Jan-Feb), MAM, JJA, SON, DJF, MAM, JJA, SON, partial DJF (Dec)
         ds = xr.Dataset(
             {"foo": (("x", "time"), np.ones((10, 365 * 2)))},
             coords={
                 "x": np.arange(10),
                 "time": pd.date_range("2000-01-01", periods=365 * 2),
             },
         )
-        rechunked = ds.chunk({"x": 2, "time": SeasonResampler()})
+
+        # Standard seasons
+        rechunked = ds.chunk(
+            {"x": 2, "time": SeasonResampler(["DJF", "MAM", "JJA", "SON"])}
+        )
         assert len(rechunked.chunksizes["time"]) == 9
         assert rechunked.chunksizes["x"] == (2,) * 5
-        # Write out the actual chunks tuple for clarity
         assert rechunked.chunksizes["time"] == (31, 92, 92, 92, 31, 92, 92, 92, 31)
 
-        # Test custom seasons
+        # Custom seasons
         rechunked = ds.chunk(
             {"x": 2, "time": SeasonResampler(["DJFM", "AM", "JJA", "SON"])}
         )
-        # Custom seasons also produce boundary chunks
         assert len(rechunked.chunksizes["time"]) == 9
         assert rechunked.chunksizes["x"] == (2,) * 5
-        # Write out the actual chunks tuple for clarity
         assert rechunked.chunksizes["time"] == (120, 61, 92, 92, 120, 61, 92, 92, 120)
         """Test chunking using SeasonResampler."""
         import dask.array