apunkt/GPy
1
0
Fork 0
mirror of https://github.com/SheffieldML/GPy.git synced 2026-06-11 15:15:15 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix pytesting test_cython.py

Browse source
This commit is contained in:
Martin Bubel 2023-10-17 08:11:58 +02:00
parent 65af6ee35e
commit 3a8b093c65
1 changed files with 28 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

40
GPy/testing/test_cython.py
View file

@ -24,30 +24,30 @@ These tests make sure that the pure python and cython codes work the same
"""
@pytest.mark.skipif(
not choleskies_cython_working,
"Cython cholesky module has not been built on this machine",
)
class CythonTestChols:
def setup(self):
self.flat = np.random.randn(45, 5)
self.triang = np.array([np.eye(20) for i in range(3)])
@pytest.mark.skipif(
not choleskies_cython_working,
"Cython cholesky module has not been built on this machine",
)
def test_flat_to_triang(self):
L1 = choleskies._flat_to_triang_pure(self.flat)
L2 = choleskies._flat_to_triang_cython(self.flat)
assert np.allclose(L1, L2), "Triang mismatch!"
@pytest.mark.skipif(
not choleskies_cython_working,
"Cython cholesky module has not been built on this machine",
)
def test_triang_to_flat(self):
A1 = choleskies._triang_to_flat_pure(self.triang)
A2 = choleskies._triang_to_flat_cython(self.triang)
assert np.allclose(A1, A2), "Flat mismatch!"
@pytest.mark.skipif(
not stationary_cython_working,
"Cython stationary module has not been built on this machine",
)
class TestStationary:
def setup(self):
self.k = GPy.kern.RBF(10)
@ -57,24 +57,40 @@ class TestStationary:
self.dKzz = np.random.randn(20, 20)
self.dKxz = np.random.randn(300, 20)
@pytest.mark.skipif(
not stationary_cython_working,
reason="Cython stationary module has not been built on this machine",
)
def test_square_gradX(self):
self.setup()
g1 = self.k._gradients_X_cython(self.dKxx, self.X)
g2 = self.k._gradients_X_pure(self.dKxx, self.X)
assert np.allclose(g1, g2), "Gradient mismatch on square X!"
@pytest.mark.skipif(
not stationary_cython_working,
reason="Cython stationary module has not been built on this machine",
)
def test_rect_gradx(self):
self.setup()
g1 = self.k._gradients_X_cython(self.dKxz, self.X, self.Z)
g2 = self.k._gradients_X_pure(self.dKxz, self.X, self.Z)
assert np.allclose(g1, g2), "Gradient mismatch on rect X!"
@pytest.mark.skipif(
not stationary_cython_working,
reason="Cython stationary module has not been built on this machine",
)
def test_square_lengthscales(self):
self.setup()
g1 = self.k._lengthscale_grads_pure(self.dKxx, self.X, self.X)
g2 = self.k._lengthscale_grads_cython(self.dKxx, self.X, self.X)
assert np.allclose(g1, g2), "Gradient mismatch on square lengthscale!"
@pytest.mark.skipif(
not stationary_cython_working,
reason="Cython stationary module has not been built on this machine",
)
def test_rect_lengthscales(self):
self.setup()
g1 = self.k._lengthscale_grads_pure(self.dKxz, self.X, self.Z)
@ -82,10 +98,6 @@ class TestStationary:
assert np.allclose(g1, g2), "Gradient mismatch on rect lengthscale!"
@pytest.mark.skipif(
not choleskies_cython_working,
"Cython cholesky module has not been built on this machine",
)
class TestCholeskiesBackprop:
def setup(self):
a = np.random.randn(10, 12)
@ -93,6 +105,10 @@ class TestCholeskiesBackprop:
self.L = GPy.util.linalg.jitchol(A)
self.dL = np.random.randn(10, 10)
@pytest.mark.skipif(
not choleskies_cython_working,
reason="Cython cholesky module has not been built on this machine",
)
def test_backprop(self):
self.setup()
r1 = choleskies._backprop_gradient_pure(self.dL, self.L)