Build a list of bin counts based on available memory

Author: antonwolfy

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(12):
+    _bin_counts += [10**6]
+
 
 class TestDigitize:
     @pytest.mark.parametrize("dtype", get_integer_float_dtypes())
@@ -488,7 +494,7 @@ def test_weights_another_sycl_queue(self):
 
     @pytest.mark.parametrize(
         "bins_count",
-        [10, 10**2, 10**3, 10**4, 10**5, 10**6],
+        _bin_counts,
     )
     def test_different_bins_amount(self, bins_count):
         v = numpy.linspace(0, bins_count, bins_count, dtype=numpy.float32)
@@ -586,7 +592,7 @@ def test_weights_unsupported_dtype(self, xp, dt):
 
     @pytest.mark.parametrize(
         "bins_count",
-        [10, 10**2, 10**3, 10**4, 10**5, 10**6],
+        _bin_counts,
     )
     def test_different_bins_amount(self, bins_count):
         v = numpy.arange(0, bins_count, dtype=int)
@@ -865,7 +871,7 @@ def test_weights_another_sycl_queue(self):
 
     @pytest.mark.parametrize(
         "bins_count",
-        [10, 10**2, 10**3, 10**4, 10**5, 10**6],
+        _bin_counts,
     )
     def test_different_bins_amount(self, bins_count):
         v = numpy.linspace(0, bins_count, bins_count, dtype=numpy.float32)