More test more betterer.

icecube · Aug 4, 2020 · 2c916a7 · 2c916a7
1 parent bc7dea8
commit 2c916a7
Showing 1 changed file with 13 additions and 5 deletions.
diff --git a/tests/test_cosmo_rates.py b/tests/test_cosmo_rates.py
@@ -1,5 +1,6 @@
 import logging
 import unittest
+import numpy as np
 from astropy import units as u
 from flarestack.cosmo import get_rate
 from flarestack.cosmo.rates import source_maps
@@ -9,6 +10,8 @@
 from flarestack.cosmo.rates.grb_rates import grb_evolutions, local_grb_rates
 from flarestack.cosmo.rates.fbot_rates import local_fbot_rates
 
+zrange = np.linspace(0.0, 8.0, 5)
+
 class TestCosmoRates(unittest.TestCase):
 
     def setUp(self):
@@ -27,7 +30,8 @@ def test_tde_rates(self):
 
         for evolution in tde_evolutions.keys():
             for rate in local_tde_rates.keys():
-                get_rate("tde", evolution_name=evolution, rate_name=rate)
+                f = get_rate("tde", evolution_name=evolution, rate_name=rate)
+                f(zrange)
 
         f = get_rate("tde", evolution_name="biehl_18_jetted", m=-2)
         true = 2.e-07 / (u.Mpc**3 * u.yr)
@@ -38,7 +42,8 @@ def test_sfr_rates(self):
 
         for evolution in sfr_evolutions.keys():
             for rate in local_sfr_rates.keys():
-                get_rate("sfr", evolution_name=evolution, rate_name=rate)
+                f = get_rate("sfr", evolution_name=evolution, rate_name=rate)
+                f(zrange)
 
         f = get_rate("sfr")
         true = 0.08687592762508031 * u.solMass / (u.Mpc**3 * u.yr)
@@ -50,12 +55,13 @@ def test_ccsn_rates(self):
         for kcc_name in kcc_rates.keys():
             for (subclass_fractions_name, (sn_type, _)) in sn_subclass_rates.items():
                 for sn_subclass in sn_type.keys():
-                    get_rate(
+                    f = get_rate(
                         "ccsn",
                         kcc_name=kcc_name,
                         sn_subclass=sn_subclass,
                         subclass_fractions_name=subclass_fractions_name
                     )
+                    f(zrange)
 
         f = get_rate("ccsn", sn_subclass="Ibc", fraction=0.5)
 
@@ -67,7 +73,8 @@ def test_grn_rates(self):
 
         for evolution in grb_evolutions.keys():
             for rate in local_grb_rates.keys():
-                get_rate("grb", evolution_name=evolution, rate_name=rate)
+                f = get_rate("grb", evolution_name=evolution, rate_name=rate)
+                f(zrange)
 
         f = get_rate("grb", evolution_name="lien_14")
         true = 1.7635240284867526e-09 / (u.Mpc**3 * u.yr)
@@ -78,7 +85,8 @@ def test_fbot_rates(self):
 
         for evolution in sfr_evolutions.keys():
             for rate in local_fbot_rates.keys():
-                get_rate("fbot", evolution_name=evolution, rate_name=rate)
+                f = get_rate("fbot", evolution_name=evolution, rate_name=rate)
+                f(zrange)
 
         f = get_rate("fbot")
         true = 4.054209955837081e-06/ (u.Mpc**3 * u.yr)