tests for glass.grf._solver

Author: ntessore

tests/grf/test_solver.py

@@ -0,0 +1,52 @@
+import numpy as np
+import numpy.testing as npt
+
+import glass.grf
+
+
+def test_lognormal(rng):
+    t1 = glass.grf.Lognormal()
+    t2 = glass.grf.Lognormal()
+
+    gl0 = rng.random()
+
+    lmax = 100
+    ell = np.arange(lmax + 1)
+    cl = 1e-2 / (2 * ell + 1) ** 2
+
+    cltol = 1e-7
+
+    gl, cl_, info = glass.grf.solve(cl, t1, t2, monopole=gl0, cltol=cltol)
+
+    assert info > 0
+
+    npt.assert_allclose(cl_[1 : cl.size], cl[1:], atol=0.0, rtol=cltol)
+
+    gl_ = glass.grf.compute(cl_, t1, t2)
+
+    assert gl[0] == gl0
+    npt.assert_allclose(gl_[1 : gl.size], gl[1:])
+
+
+def test_monopole(rng):
+    t = glass.grf.Normal()
+    x = rng.random(100)
+
+    y, x_, _ = glass.grf.solve(x, t, monopole=None)
+
+    npt.assert_allclose(y, x)
+    npt.assert_allclose(x_, x)
+
+    y, x_, _ = glass.grf.solve(x, t, monopole=0.0)
+
+    assert y[0] == 0.0
+
+    npt.assert_allclose(y[1:], x[1:])
+    npt.assert_allclose(x_[1:], x[1:])
+
+    y, x_, _ = glass.grf.solve(x, t, monopole=1.0)
+
+    assert y[0] == 1.0
+
+    npt.assert_allclose(y[1:], x[1:])
+    npt.assert_allclose(x_[1:], x[1:])