Skip to content

Commit 192b2f6

Browse files
committed
Minor test.
1 parent 7959119 commit 192b2f6

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

xarray/tests/test_groupby.py

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1472,16 +1472,16 @@ def test_groupby_fillna(self):
14721472
def test_groupby_fastpath_for_monotonic(self):
14731473
# Fixes https://github.com/pydata/xarray/issues/6220
14741474
index = [1, 2, 3, 4, 7, 9, 10]
1475-
fwd = DataArray(np.ones(len(index)), [("idx", index)]).groupby(
1476-
"idx", squeeze=False
1477-
)
1478-
rev = DataArray(np.ones(len(index)), [("idx", index[::-1])]).groupby(
1479-
"idx", squeeze=False
1480-
)
1475+
array = DataArray(np.arange(len(index)), [("idx", index)])
1476+
array_rev = array.copy().assign_coords({"idx": index[::-1]})
1477+
fwd = array.groupby("idx", squeeze=False)
1478+
rev = array_rev.groupby("idx", squeeze=False)
1479+
1480+
for gb in [fwd, rev]:
1481+
assert all([isinstance(elem, slice) for elem in gb._group_indices])
14811482

1482-
assert_array_equal(fwd._group_indices, rev._group_indices)
1483-
for da in [fwd, rev]:
1484-
assert all([isinstance(elem, slice) for elem in da._group_indices])
1483+
assert_identical(fwd.sum(), array)
1484+
assert_identical(rev.sum(), array_rev.sortby("idx"))
14851485

14861486

14871487
class TestDataArrayResample:

0 commit comments

Comments
 (0)