Unit Tests for PointSourceResponse

tests/response/test_point_source_response.py

@@ -0,0 +1,63 @@
+import numpy as np
+from numpy import array_equal as arr_eq
+from histpy import Histogram, Axes, Axis
+from scoords import SpacecraftFrame
+from astropy.coordinates import SkyCoord
+import astropy.units as u
+import h5py as h5
+from astropy.time import Time
+from mhealpy import HealpixBase, HealpixMap
+
+from cosipy import test_data
+from cosipy.response.FullDetectorResponse import cosi_response
+from cosipy.response import PointSourceResponse, FullDetectorResponse
+
+from threeML import DiracDelta, Constant, Line, Quadratic, Cubic, Quartic, Model 
+from threeML import StepFunction, StepFunctionUpper, Cosine_Prior, Uniform_prior, PhAbs, Gaussian, TbAbs
+from cosipy.threeml.custom_functions import SpecFromDat
+
+import pytest
+
+# init/load
+response_path = test_data.path/"test_full_detector_response.h5"
+
+with FullDetectorResponse.open(response_path) as response:
+    exposure_map = HealpixMap(base=response,
+                                    unit=u.s,
+                                    coordsys=SpacecraftFrame())
+
+    ti = Time('1999-01-01T00:00:00.123456789')
+    tf = Time('2010-01-01T00:00:00')
+    dt = (tf-ti).to(u.s)
+
+    exposure_map[:4] = dt/4
+
+    psr = response.get_point_source_response(exposure_map = exposure_map)
+
+def test_photon_energy_axis():
+    assert psr.photon_energy_axis == psr.axes['Ei']
+
+def test_get_expectation():
+    # supported spectral functions
+    ## see astromodels.functions.function.Function1D.[function_name]()
+    ## 3 sig-figs
+    assert 631 < np.sum(psr.get_expectation(Constant(k=1e-1)).contents) < 632
+    assert 75.2 < np.sum(psr.get_expectation(Line(a=1e-1, b=-4e-5)).contents) < 75.3
+    assert 116 < np.sum(psr.get_expectation(Quadratic(a=1e-1, b=-4e-5, c=1e-9)).contents) < 117
+    assert 59.4 < np.sum(psr.get_expectation(Cubic(a=1e-1, b=-4e-5, c=1e-9, d=-4e-13)).contents) < 59.5
+    assert 64.8 < np.sum(psr.get_expectation(Quartic(a=1e-1, b=-4e-5, c=1e-9, d=-4e-13, e=1e-17)).contents) < 64.9
+    assert 39.9 < np.sum(psr.get_expectation(StepFunction(upper_bound=3e2, lower_bound=0, value=1)).contents) < 40.0
+    assert 39.9 < np.sum(psr.get_expectation(StepFunctionUpper(upper_bound=3e2, lower_bound=0, value=1)).contents) < 40.0
+    assert 6310 < np.sum(psr.get_expectation(PhAbs(NH=1e-2, redshift=1)).contents) < 6320
+    assert 1570 < np.sum(psr.get_expectation(DiracDelta(zero_point=511, value=1000)).contents) < 1580
+    assert 2160 < np.sum(psr.get_expectation(Gaussian(F=1000, mu=1e3, sigma=1e1)).contents) < 2170
+    assert 2.220 < np.sum(psr.get_expectation(Cosine_Prior(upper_bound=1e3, lower_bound=0, value=1))) < 2.230
+    assert 1280 < np.sum(psr.get_expectation(Uniform_prior(upper_bound=1e3, lower_bound=0, value=1)).contents) < 1290
+
+    ## example implicitly supported :py:class:`threeML.Model` with units
+    assert 6290 < np.sum(psr.get_expectation(TbAbs(NH=3, redshift=15)).contents) < 6300
+
+    ## generic unsupported :py:class:`threeML.Model` without units
+    with pytest.raises(RuntimeError) as pytest_wrapped_exp:
+        expectation = psr.get_expectation(Model())
+    assert pytest_wrapped_exp.type == RuntimeError