From c20a361179e0339cfdad5d3f4e6918a05ed390c6 Mon Sep 17 00:00:00 2001 From: Martin Bubel Date: Fri, 6 Oct 2023 18:58:00 +0200 Subject: [PATCH] migrate mapping_tests to pytest --- GPy/testing/mapping_tests.py | 19 +++++++------------ 1 file changed, 7 insertions(+), 12 deletions(-) diff --git a/GPy/testing/mapping_tests.py b/GPy/testing/mapping_tests.py index 0c10e53b..6b829f06 100644 --- a/GPy/testing/mapping_tests.py +++ b/GPy/testing/mapping_tests.py @@ -30,22 +30,22 @@ class MappingGradChecker(GPy.core.Model): self.mapping.update_gradients(self.dL_dY, self.X) -class MappingTests(unittest.TestCase): +class TestMapping: def test_kernelmapping(self): X = np.random.randn(100, 3) Z = np.random.randn(10, 3) mapping = GPy.mappings.Kernel(3, 2, Z, GPy.kern.RBF(3)) - self.assertTrue(MappingGradChecker(mapping, X).checkgrad()) + assert MappingGradChecker(mapping, X).checkgrad() def test_linearmapping(self): mapping = GPy.mappings.Linear(3, 2) X = np.random.randn(100, 3) - self.assertTrue(MappingGradChecker(mapping, X).checkgrad()) + assert MappingGradChecker(mapping, X).checkgrad() def test_mlpmapping(self): mapping = GPy.mappings.MLP(input_dim=3, hidden_dim=5, output_dim=2) X = np.random.randn(100, 3) - self.assertTrue(MappingGradChecker(mapping, X).checkgrad()) + assert MappingGradChecker(mapping, X).checkgrad() def test_mlpextmapping(self): np.random.seed(42) @@ -54,14 +54,14 @@ class MappingTests(unittest.TestCase): mapping = GPy.mappings.MLPext( input_dim=3, hidden_dims=[5, 5], output_dim=2, activation=activation ) - self.assertTrue(MappingGradChecker(mapping, X).checkgrad()) + assert MappingGradChecker(mapping, X).checkgrad() def test_addmapping(self): m1 = GPy.mappings.MLP(input_dim=3, hidden_dim=5, output_dim=2) m2 = GPy.mappings.Linear(input_dim=3, output_dim=2) mapping = GPy.mappings.Additive(m1, m2) X = np.random.randn(100, 3) - self.assertTrue(MappingGradChecker(mapping, X).checkgrad()) + assert MappingGradChecker(mapping, X).checkgrad() def test_compoundmapping(self): m1 = GPy.mappings.MLP(input_dim=3, hidden_dim=5, output_dim=2) @@ -69,9 +69,4 @@ class MappingTests(unittest.TestCase): m2 = GPy.mappings.Kernel(2, 4, Z, GPy.kern.RBF(2)) mapping = GPy.mappings.Compound(m1, m2) X = np.random.randn(100, 3) - self.assertTrue(MappingGradChecker(mapping, X).checkgrad()) - - -if __name__ == "__main__": - print("Running unit tests, please be (very) patient...") - unittest.main() + assert MappingGradChecker(mapping, X).checkgrad()