Update statistics tests

Author: antonwolfy

dpnp/tests/third_party/cupy/statistics_tests/test_histogram.py

@@ -497,7 +497,6 @@ def test_histogramdd(self, xp, dtype):
             weights = xp.ones((x.shape[0],), dtype=self.weights_dtype)
         else:
             weights = None
-
         y, bin_edges = xp.histogramdd(
             x,
             bins=bins,
@@ -549,6 +548,13 @@ def test_histogramdd_invalid_range(self):
             with pytest.raises(ValueError):
                 y, bin_edges = xp.histogramdd(x, range=r)
 
+    @pytest.mark.skip("list of bins is allowed")
+    def test_histogramdd_disallow_arraylike_bins(self):
+        x = testing.shaped_random((16, 2), cupy, scale=100)
+        bins = [[0, 10, 20, 50, 90]] * 2  # too many dimensions
+        with pytest.raises(ValueError):
+            y, bin_edges = cupy.histogramdd(x, bins=bins)
+
 
 @testing.parameterize(
     *testing.product(
@@ -596,9 +602,10 @@ def test_histogram2d(self, xp, dtype):
 
 class TestHistogram2dErrors(unittest.TestCase):
 
+    @pytest.mark.skip("list of bins is allowed")
     def test_histogram2d_disallow_arraylike_bins(self):
         x = testing.shaped_random((16,), cupy, scale=100)
         y = testing.shaped_random((16,), cupy, scale=100)
         bins = [0, 10, 20, 50, 90]
         with pytest.raises(ValueError):
-            y, bin_edges = cupy.histogram2d(x, y, bins=bins)
+            y, _, _ = cupy.histogram2d(x, y, bins=bins)

dpnp/tests/third_party/cupy/statistics_tests/test_order.py

@@ -252,6 +252,13 @@ def test_quantile_out_of_range_q(self, dtype, method):
                 with pytest.raises(ValueError):
                     xp.quantile(a, q, axis=-1, method=method)
 
+    @testing.for_all_dtypes(no_float16=True, no_bool=True, no_complex=True)
+    @testing.numpy_cupy_allclose(rtol=1e-6)
+    def test_quantile_axis_and_keepdims(self, xp, dtype, method):
+        a = testing.shaped_random((1, 6, 3, 2), xp, dtype)
+        q = testing.shaped_random((5,), xp, scale=1)
+        return xp.quantile(a, q, axis=0, keepdims=True, method=method)
+
 
 class TestOrder: