Build a list of bin counts based on available memory

Author: antonwolfy

dpnp/tests/helper.py

@@ -550,4 +550,4 @@ def requires_memory(no_of_gbs, device=None):
     """
 
     free_mem = get_dev_info(device).get("free_memory", 0)
-    return free_mem > no_of_gbs * (1024**3)
+    return free_mem >= no_of_gbs * (1024**3)

dpnp/tests/test_histogram.py

@@ -22,8 +22,14 @@
     get_integer_float_dtypes,
     has_support_aspect64,
     numpy_version,
+    requires_memory,
 )
 
+# Build a list of bin sizes to check the histogram
+_bin_counts = [10, 10**2, 10**3, 10**4, 10**5]
+if requires_memory(16):
+    _bin_counts += [10**6]
+
 
 class TestDigitize:
     @pytest.mark.parametrize("dtype", get_integer_float_dtypes())
@@ -486,10 +492,7 @@ def test_weights_another_sycl_queue(self):
         with assert_raises(ValueError):
             dpnp.histogram(v, weights=w)
 
-    @pytest.mark.parametrize(
-        "bins_count",
-        [10, 10**2, 10**3, 10**4, 10**5, 10**6],
-    )
+    @pytest.mark.parametrize("bins_count", _bin_counts)
     def test_different_bins_amount(self, bins_count):
         v = numpy.linspace(0, bins_count, bins_count, dtype=numpy.float32)
         iv = dpnp.array(v)
@@ -584,10 +587,7 @@ def test_weights_unsupported_dtype(self, xp, dt):
         w = xp.arange(5, dtype=dt)
         assert_raises((TypeError, ValueError), xp.bincount, v, weights=w)
 
-    @pytest.mark.parametrize(
-        "bins_count",
-        [10, 10**2, 10**3, 10**4, 10**5, 10**6],
-    )
+    @pytest.mark.parametrize("bins_count", _bin_counts)
     def test_different_bins_amount(self, bins_count):
         v = numpy.arange(0, bins_count, dtype=int)
         iv = dpnp.array(v)
@@ -863,10 +863,7 @@ def test_weights_another_sycl_queue(self):
         with assert_raises(ValueError):
             dpnp.histogramdd(v, weights=w)
 
-    @pytest.mark.parametrize(
-        "bins_count",
-        [10, 10**2, 10**3, 10**4, 10**5, 10**6],
-    )
+    @pytest.mark.parametrize("bins_count", _bin_counts)
     def test_different_bins_amount(self, bins_count):
         v = numpy.linspace(0, bins_count, bins_count, dtype=numpy.float32)
         iv = dpnp.array(v)