File tree 1 file changed +5
-4
lines changed
1 file changed +5
-4
lines changed Original file line number Diff line number Diff line change @@ -1475,9 +1475,10 @@ def test_groupby_restore_coord_dims(self):
1475
1475
("a" , ("a" , "y" )),
1476
1476
("b" , ("x" , "b" )),
1477
1477
]:
1478
- result = array .groupby (by , squeeze = False , restore_coord_dims = True ).map (
1479
- lambda x : x .squeeze ()
1480
- )["c" ]
1478
+ with pytest .warns (UserWarning , match = "The `squeeze` kwarg" ):
1479
+ result = array .groupby (by , restore_coord_dims = True ).map (
1480
+ lambda x : x .squeeze ()
1481
+ )["c" ]
1481
1482
assert result .dims == expected_dims
1482
1483
1483
1484
def test_groupby_first_and_last (self ):
@@ -1585,7 +1586,7 @@ def test_groupby_bins_ellipsis(self):
1585
1586
da = xr .DataArray (np .ones ((2 , 3 , 4 )))
1586
1587
bins = [- 1 , 0 , 1 , 2 ]
1587
1588
with xr .set_options (use_flox = False ):
1588
- actual = da .groupby_bins ("dim_0" , bins , squeeze = False ).mean (...)
1589
+ actual = da .groupby_bins ("dim_0" , bins ).mean (...)
1589
1590
with xr .set_options (use_flox = True ):
1590
1591
expected = da .groupby_bins ("dim_0" , bins ).mean (...)
1591
1592
assert_allclose (actual , expected )
You can’t perform that action at this time.
0 commit comments