Fix bug with cross-power (#6)

Author: steven-murray

CHANGELOG.rst

@@ -1,8 +1,14 @@
 Changelog
 =========
 
-dev
----
+0.6.1
+-----
+**Bugfixes**
+
+- Fix error when doing cross-power [Issue #5].
+
+0.6.0
+-----
 **Features**
 
 - New ``left_edge`` argument in fft/ifft which gives the ability to specify where

powerbox/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.6.0"
+__version__ = "0.6.1"
 
 from .powerbox import PowerBox, LogNormalPowerBox
 from .tools import get_power, angular_average, angular_average_nd

powerbox/tools.py

@@ -460,7 +460,7 @@ def get_power(deltax, boxlength, deltax2=None, N=None, a=1., b=1., remove_shotno
         if not np.iterable(boxlength):
             boxlength = [boxlength] * dim
 
-        if deltax2 is not None and deltax.shape != len(deltax2.shape):
+        if deltax2 is not None and deltax.shape != deltax2.shape:
             raise ValueError("deltax and deltax2 must have the same shape!")
 
         N = deltax.shape

tests/test_tools.py

@@ -1,5 +1,6 @@
 import numpy as np
-from powerbox.tools import angular_average_nd, angular_average
+from powerbox.tools import angular_average_nd, angular_average, get_power
+from powerbox.powerbox import PowerBox
 import pytest
 
 def test_angular_avg_nd_3():
@@ -109,6 +110,13 @@ def test_logbins():
     assert np.all(np.isclose(np.diff(coord[1:]/coord[:-1]), 0))
 
 
+def test_cross_power_identity():
+    pb = PowerBox(200, dim=2, pk=lambda k: 1.0 * k ** -2., boxlength=1.0, b=1)
+    dx = pb.delta_x()
+    p, k = get_power(dx, pb.boxlength, b=1)
+    p_cross, k = get_power(dx, pb.boxlength, b=1, deltax2=dx)
+    assert np.all(np.isclose(p, p_cross))
+
 
 @pytest.mark.skip()
 def test_against_multirealisation():