rebalance halo sampler test buffer size

21cmfast · Nov 13, 2024 · 76d594b · 76d594b
1 parent 4e01ba6
commit 76d594b
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/py21cmfast/wrapper/cfuncs.py b/src/py21cmfast/wrapper/cfuncs.py
@@ -1203,7 +1203,7 @@ def halo_sample_test(
     if from_cat:
         z_prev = (1 + redshift) / global_params.ZPRIME_STEP_FACTOR - 1
 
-    buffer_size = int(2e7)
+    buffer_size = int(3e7)  # about 500MB total 2e7 * 4 (float) * 4 (mass + 3crd)
     nhalo_out = np.zeros(1).astype("i4")
     N_out = np.zeros(n_cond).astype("i4")
     M_out = np.zeros(n_cond).astype("f8")

diff --git a/tests/test_halo_sampler.py b/tests/test_halo_sampler.py
@@ -41,7 +41,7 @@ def test_sampler(name, cond, from_cat, plt):
 
     from_cat = "cat" in from_cat
 
-    n_cond = 25000
+    n_cond = 15000
     if from_cat:
         mass = 10 ** options_log10mass[cond]
         cond = mass