tests

tests/qp/test_projector_shifts_widths.py

@@ -0,0 +1,39 @@
+import qp
+import numpy as np
+import qp.projectors as proj
+
+
+def make_qp_ens(file):
+    zs = file['zs']
+    nzs = file['pzs']
+    dz = np.mean(np.diff(zs))
+    zs_edges = np.append(zs - dz/2, zs[-1] + dz/2)
+    q = qp.Ensemble(qp.hist, data={"bins":zs_edges, "pdfs":nzs})
+    return q
+
+
+def make_projector():
+    file = np.load('tests/qp/dummy.npz')
+    ens = make_qp_ens(file)
+    return proj.ProjectorShiftsWidths(ens)
+
+
+def test_prior():
+    projector = make_projector()
+    prior = projector.get_prior()
+    assert prior is not None
+
+
+def test_sample_prior():
+    projector = make_projector()
+    prior_sample = projector.sample_prior()
+    assert len([prior_sample]) == len([projector.shift, projector.width])
+
+
+def test_model():
+    projector = make_projector()
+    shift, width = projector.sample_prior()
+    input = np.array([projector.z, projector.nz_mean])
+    output = projector.evaluate_model(input, shift, width)
+    assert (projector.z == output[0]).all()
+    assert len(output[1]) == len(projector.nz_mean)