From a800f4b7ed7eca2fa0733fa4cc29f4119a7e59d2 Mon Sep 17 00:00:00 2001 From: Max Zwiessele Date: Fri, 14 Mar 2014 12:31:28 +0000 Subject: [PATCH] prior tests renewed --- GPy/testing/prior_tests.py | 13 +++---------- 1 file changed, 3 insertions(+), 10 deletions(-) diff --git a/GPy/testing/prior_tests.py b/GPy/testing/prior_tests.py index c16057db..db6cc685 100644 --- a/GPy/testing/prior_tests.py +++ b/GPy/testing/prior_tests.py @@ -15,7 +15,7 @@ class PriorTests(unittest.TestCase): X, y = X[:, None], y[:, None] m = GPy.models.GPRegression(X, y) lognormal = GPy.priors.LogGaussian(1, 2) - m.set_prior('rbf', lognormal) + m.rbf.set_prior(lognormal) m.randomize() self.assertTrue(m.checkgrad()) @@ -28,7 +28,7 @@ class PriorTests(unittest.TestCase): X, y = X[:, None], y[:, None] m = GPy.models.GPRegression(X, y) Gamma = GPy.priors.Gamma(1, 1) - m.set_prior('rbf', Gamma) + m.rbf.set_prior(Gamma) m.randomize() self.assertTrue(m.checkgrad()) @@ -41,16 +41,9 @@ class PriorTests(unittest.TestCase): X, y = X[:, None], y[:, None] m = GPy.models.GPRegression(X, y) gaussian = GPy.priors.Gaussian(1, 1) - success = False - # setting a Gaussian prior on non-negative parameters # should raise an assertionerror. - try: - m.set_prior('rbf', gaussian) - except AssertionError: - success = True - - self.assertTrue(success) + self.assertRaises(AssertionError, m.rbf.set_prior, gaussian) if __name__ == "__main__":