apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-05-10 20:42:39 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

migrate mapping_tests to pytest

Browse source
This commit is contained in:
Martin Bubel 2023-10-06 18:58:00 +02:00
parent 618f35531d
commit c20a361179
1 changed files with 7 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

19
GPy/testing/mapping_tests.py
View file

@ -30,22 +30,22 @@ class MappingGradChecker(GPy.core.Model):
self.mapping.update_gradients(self.dL_dY, self.X) self.mapping.update_gradients(self.dL_dY, self.X)
class MappingTests(unittest.TestCase): class TestMapping:
def test_kernelmapping(self): def test_kernelmapping(self):
X = np.random.randn(100, 3) X = np.random.randn(100, 3)
Z = np.random.randn(10, 3) Z = np.random.randn(10, 3)
mapping = GPy.mappings.Kernel(3, 2, Z, GPy.kern.RBF(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): def test_linearmapping(self):
mapping = GPy.mappings.Linear(3, 2) mapping = GPy.mappings.Linear(3, 2)
X = np.random.randn(100, 3) X = np.random.randn(100, 3)
self.assertTrue(MappingGradChecker(mapping, X).checkgrad()) assert MappingGradChecker(mapping, X).checkgrad()
def test_mlpmapping(self): def test_mlpmapping(self):
mapping = GPy.mappings.MLP(input_dim=3, hidden_dim=5, output_dim=2) mapping = GPy.mappings.MLP(input_dim=3, hidden_dim=5, output_dim=2)
X = np.random.randn(100, 3) X = np.random.randn(100, 3)
self.assertTrue(MappingGradChecker(mapping, X).checkgrad()) assert MappingGradChecker(mapping, X).checkgrad()
def test_mlpextmapping(self): def test_mlpextmapping(self):
np.random.seed(42) np.random.seed(42)
@ -54,14 +54,14 @@ class MappingTests(unittest.TestCase):
mapping = GPy.mappings.MLPext( mapping = GPy.mappings.MLPext(
input_dim=3, hidden_dims=[5, 5], output_dim=2, activation=activation 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): def test_addmapping(self):
m1 = GPy.mappings.MLP(input_dim=3, hidden_dim=5, output_dim=2) m1 = GPy.mappings.MLP(input_dim=3, hidden_dim=5, output_dim=2)
m2 = GPy.mappings.Linear(input_dim=3, output_dim=2) m2 = GPy.mappings.Linear(input_dim=3, output_dim=2)
mapping = GPy.mappings.Additive(m1, m2) mapping = GPy.mappings.Additive(m1, m2)
X = np.random.randn(100, 3) X = np.random.randn(100, 3)
self.assertTrue(MappingGradChecker(mapping, X).checkgrad()) assert MappingGradChecker(mapping, X).checkgrad()
def test_compoundmapping(self): def test_compoundmapping(self):
m1 = GPy.mappings.MLP(input_dim=3, hidden_dim=5, output_dim=2) 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)) m2 = GPy.mappings.Kernel(2, 4, Z, GPy.kern.RBF(2))
mapping = GPy.mappings.Compound(m1, m2) mapping = GPy.mappings.Compound(m1, m2)
X = np.random.randn(100, 3) X = np.random.randn(100, 3)
self.assertTrue(MappingGradChecker(mapping, X).checkgrad()) assert MappingGradChecker(mapping, X).checkgrad()
if __name__ == "__main__":
print("Running unit tests, please be (very) patient...")
unittest.main()