From 59657bc49d0fa6edd7bd9ae3e5c3761453b24a4e Mon Sep 17 00:00:00 2001 From: Martin Bubel Date: Wed, 4 Oct 2023 21:35:09 +0200 Subject: [PATCH] migrate fitc to pytest --- GPy/testing/fitc.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/GPy/testing/fitc.py b/GPy/testing/fitc.py index d1b82cc8..f069a90d 100644 --- a/GPy/testing/fitc.py +++ b/GPy/testing/fitc.py @@ -1,13 +1,12 @@ # Copyright (c) 2014, James Hensman # Licensed under the BSD 3-clause license (see LICENSE.txt) -import unittest import numpy as np import GPy -class FITCtest(unittest.TestCase): - def setUp(self): +class FITCtest: + def setup(self): ###################################### # # 1 dimensional example @@ -27,11 +26,13 @@ class FITCtest(unittest.TestCase): ) def test_fitc_1d(self): + self.setup() m = GPy.models.SparseGPRegression(self.X1D, self.Y1D) m.inference_method = GPy.inference.latent_function_inference.FITC() - self.assertTrue(m.checkgrad()) + assert m.checkgrad(), "Gradient check failed!" def test_fitc_2d(self): + self.setup() m = GPy.models.SparseGPRegression(self.X2D, self.Y2D) m.inference_method = GPy.inference.latent_function_inference.FITC() - self.assertTrue(m.checkgrad()) + assert m.checkgrad(), "Gradient check failed!"