Added test

steven-murray · Apr 5, 2024 · 2026d6c · 2026d6c
1 parent 2b6e5a7
commit 2026d6c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/src/powerbox/__init__.py b/src/powerbox/__init__.py
@@ -16,4 +16,4 @@
 
 from .dft_backend import FFTW, NumpyFFT, get_fft_backend
 from .powerbox import LogNormalPowerBox, PowerBox
-from .tools import angular_average, angular_average_nd, get_power
+from .tools import angular_average, angular_average_nd, get_power, power2delta
diff --git a/tests/test_power.py b/tests/test_power.py
@@ -1,6 +1,6 @@
 import numpy as np
 
-from powerbox import PowerBox, get_power
+from powerbox import PowerBox, get_power, power2delta
 
 
 def test_power1d():
@@ -105,3 +105,12 @@ def test_k_weights():
 
     assert np.all(k1 == k0)
     assert not np.allclose(p1, p0)
+
+
+def test_prefactor_fnc():
+    pb = PowerBox(50, dim=3, pk=lambda k: 1.0 * k**-2.0, boxlength=1.0, b=1)
+    pdelta, kdelta = get_power(pb.delta_x(), pb.boxlength, prefactor_fnc=power2delta)
+    p, k = get_power(pb.delta_x(), pb.boxlength)
+
+    assert np.all(k == kdelta)
+    assert np.any(p != pdelta)