From d319303c3aa00ec99ebd2b411c8e9e76b44cecf1 Mon Sep 17 00:00:00 2001
From: Colm Talbot <colm.talbot@ligo.org>
Date: Mon, 14 May 2018 13:25:48 +1000
Subject: [PATCH] add test of normalisation
---
test/prior_tests.py | 11 +++++++++++
1 file changed, 11 insertions(+)
diff --git a/test/prior_tests.py b/test/prior_tests.py
index 3bad18bbc..566dc3f28 100644
--- a/test/prior_tests.py
+++ b/test/prior_tests.py
@@ -169,6 +169,17 @@ class TestPriorClasses(unittest.TestCase):
surround_domain = np.linspace(prior.minimum - 1, prior.maximum + 1, 1000)
prior.prob(surround_domain)
+ def test_normalized(self):
+ """Test that each of the priors are normalised, this needs care for delta function and Gaussian priors"""
+ for prior in self.priors:
+ if isinstance(prior, tupak.prior.DeltaFunction):
+ continue
+ elif isinstance(prior, tupak.prior.Gaussian):
+ domain = np.linspace(-1e2, 1e2, 1000)
+ else:
+ domain = np.linspace(prior.minimum, prior.maximum, 1000)
+ self.assertAlmostEqual(np.trapz(prior.prob(domain), domain), 1, 3)
+
if __name__ == '__main__':
unittest.main()
--
GitLab