update percentile tests

Author: rcomer

lib/iris/tests/unit/analysis/test_PERCENTILE.py

@@ -11,6 +11,7 @@
 
 from unittest import mock
 
+import dask.array as da
 import numpy as np
 import numpy.ma as ma
 
@@ -147,23 +148,35 @@ def test_masked_2d_multi(self):
             data, axis, percent, expected, mdtol=mdtol, approx=True
         )
 
-    @mock.patch("scipy.stats.mstats.mquantiles")
+    @mock.patch("scipy.stats.mstats.mquantiles", return_value=[2, 4])
     def test_default_kwargs_passed(self, mocked_mquantiles):
         data = np.arange(5)
-        percent = 50
+        percent = [42, 75]
         axis = 0
+        if self.lazy:
+            data = as_lazy_data(data)
+
         self.agg_method(data, axis=axis, percent=percent)
+
+        # Trigger calculation for lazy case.
+        as_concrete_data(data)
         for key in ["alphap", "betap"]:
             self.assertEqual(mocked_mquantiles.call_args.kwargs[key], 1)
 
     @mock.patch("scipy.stats.mstats.mquantiles")
     def test_chosen_kwargs_passed(self, mocked_mquantiles):
         data = np.arange(5)
-        percent = 50
+        percent = [42, 75]
         axis = 0
+        if self.lazy:
+            data = as_lazy_data(data)
+
         self.agg_method(
             data, axis=axis, percent=percent, alphap=0.6, betap=0.5
         )
+
+        # Trigger calculation for lazy case.
+        as_concrete_data(data)
         for key, val in zip(["alphap", "betap"], [0.6, 0.5]):
             self.assertEqual(mocked_mquantiles.call_args.kwargs[key], val)
 
@@ -200,11 +213,18 @@ def test_masked(self):
                 data, axis=0, percent=50, fast_percentile_method=True
             )
 
+    @mock.patch("numpy.percentile")
+    def test_numpy_percentile_called(self, mocked_percentile):
+        # Basic check that numpy.percentile is called.
+        data = np.arange(5)
+        self.agg_method(data, axis=0, percent=42, fast_percentile_method=True)
+        mocked_percentile.assert_called_once()
+
 
 class MultiAxisMixin:
     """
     Tests for axis passed as a tuple.  Only relevant for lazy aggregation since
-    axis is always specified as int for real aggregation.
+    axis is always specified asax int for real aggregation.
 
     """
 
@@ -269,6 +289,18 @@ def test_masked(self):
         with self.assertRaisesRegex(TypeError, emsg):
             as_concrete_data(actual)
 
+    @mock.patch("numpy.percentile", return_value=np.array([2, 4]))
+    def test_numpy_percentile_called(self, mocked_percentile):
+        # Basic check that numpy.percentile is called.
+        data = da.arange(5)
+        result = self.agg_method(
+            data, axis=0, percent=[42, 75], fast_percentile_method=True
+        )
+
+        self.assertTrue(is_lazy_data(result))
+        as_concrete_data(result)
+        mocked_percentile.assert_called()
+
 
 class Test_lazy_aggregate(
     tests.IrisTest, AggregateMixin, MaskedAggregateMixin, MultiAxisMixin