Skip to content

Commit 70aaa9d

Browse files
committed
Update Variable and dask tests with _argmin_base, _argmax_base
The basic numpy-style argmin() and argmax() methods were renamed when adding support for handling multiple dimensions in DataArray.argmin() and DataArray.argmax(). Variable.argmin() and Variable.argmax() are therefore renamed as Variable._argmin_base() and Variable._argmax_base().
1 parent 6d9d509 commit 70aaa9d

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

xarray/tests/test_dask.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -209,11 +209,11 @@ def test_reduce(self):
209209
self.assertLazyAndAllClose(u.mean(), v.mean())
210210
self.assertLazyAndAllClose(u.std(), v.std())
211211
with raise_if_dask_computes():
212-
actual = v.argmax(dim="x")
213-
self.assertLazyAndAllClose(u.argmax(dim="x"), actual)
212+
actual = v._argmax_base(dim="x")
213+
self.assertLazyAndAllClose(u._argmax_base(dim="x"), actual)
214214
with raise_if_dask_computes():
215-
actual = v.argmin(dim="x")
216-
self.assertLazyAndAllClose(u.argmin(dim="x"), actual)
215+
actual = v._argmin_base(dim="x")
216+
self.assertLazyAndAllClose(u._argmin_base(dim="x"), actual)
217217
self.assertLazyAndAllClose((u > 1).any(), (v > 1).any())
218218
self.assertLazyAndAllClose((u < 1).all("x"), (v < 1).all("x"))
219219
with raises_regex(NotImplementedError, "only works along an axis"):

xarray/tests/test_variable.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -1657,7 +1657,7 @@ def test_reduce_funcs(self):
16571657
assert_identical(v.all(dim="x"), Variable([], False))
16581658

16591659
v = Variable("t", pd.date_range("2000-01-01", periods=3))
1660-
assert v.argmax(skipna=True) == 2
1660+
assert v._argmax_base(skipna=True) == 2
16611661

16621662
assert_identical(v.max(), Variable([], pd.Timestamp("2000-01-03")))
16631663

0 commit comments

Comments
 (0)