index ordering tests

Author: keewis

xarray_array_testing/reduction.py

@@ -53,3 +53,17 @@ def test_variable_order_reduce(self, op, data):
 
         assert isinstance(actual, self.array_type)
         self.assert_equal(actual, expected)
+
+    @pytest.mark.parametrize("op", ["argmax", "argmin"])
+    @given(st.data())
+    def test_variable_order_reduce_index(self, op, data):
+        variable = data.draw(xrst.variables(array_strategy_fn=self.array_strategy_fn))
+
+        with self.expected_errors(op, variable=variable):
+            # compute using xr.Variable.<OP>()
+            actual = getattr(variable, op)().data
+            # compute using xp.<OP>(array)
+            expected = getattr(self.xp, op)(variable.data)
+
+        assert isinstance(actual, self.array_type)
+        self.assert_equal(actual, expected)