From 9e93ba1194b021f18605d7d13288147b1f912879 Mon Sep 17 00:00:00 2001 From: Moritz Huebner <email@moritz-huebner.de> Date: Tue, 22 May 2018 16:51:31 +1000 Subject: [PATCH] Moritz Huebner: Started writing tests --- test/sampler_tests.py | 33 +++++++++++++++++++++++++++++++++ 1 file changed, 33 insertions(+) create mode 100644 test/sampler_tests.py diff --git a/test/sampler_tests.py b/test/sampler_tests.py new file mode 100644 index 000000000..494d4a18b --- /dev/null +++ b/test/sampler_tests.py @@ -0,0 +1,33 @@ +from context import tupak +from tupak import prior +from tupak import likelihood +import unittest +from mock import Mock, MagicMock +import numpy as np + + +class TestSamplerInstantiation(unittest.TestCase): + + def setUp(self): + self.likelihood = likelihood.Likelihood() + self.likelihood.parameters = MagicMock(return_value=dict(a=1, b=2, c=3)) + delta_prior = prior.DeltaFunction(0) + delta_prior.peak = MagicMock(return_value=0) + delta_prior.rescale = MagicMock(return_value=delta_prior) + delta_prior.prob = MagicMock(return_value=1) + uniform_prior = prior.Uniform(0, 1) + uniform_prior.minimum = MagicMock(return_value=0) + uniform_prior.maximum = MagicMock(return_value=1) + uniform_prior.rescale = MagicMock(return_value=uniform_prior) + uniform_prior.prob = MagicMock(return_value=1) + + self.priors = dict(a=delta_prior, b='string', c=uniform_prior) + self.likelihood.log_likelihood_ratio = MagicMock(return_value=1) + self.likelihood.log_likelihood = MagicMock(return_value=2) + + def tearDown(self): + del self.likelihood + del self.priors + + def test_default_instantiation(self): + sampler = tupak.sampler.Sampler(self.likelihood, self.priors) -- GitLab