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

migrate mpi_tests to pytest

Browse source
This commit is contained in:
Martin Bubel 2023-10-10 18:49:10 +02:00
parent 1d7504fdfb
commit b0c5e137a2
1 changed files with 3 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

18
GPy/testing/mpi_tests.py
View file

@ -3,13 +3,11 @@
import unittest
import numpy as np
import GPy
try:
from mpi4py import MPI
import subprocess
class MPITests:
class TestMPI:
def test_BayesianGPLVM_MPI(self):
code = """
import numpy as np
@ -38,7 +36,7 @@ if comm.rank==0:
p = subprocess.Popen(
"mpirun -n 4 python mpi_test__.py", stdout=subprocess.PIPE, shell=True
)
(stdout, stderr) = p.communicate()
(stdout, _stderr) = p.communicate()
L1 = float(stdout.splitlines()[-2])
L2 = float(stdout.splitlines()[-1])
self.assertTrue(np.allclose(L1, L2))
@ -77,20 +75,10 @@ if comm.rank==0:
(stdout, stderr) = p.communicate()
L1 = float(stdout.splitlines()[-2])
L2 = float(stdout.splitlines()[-1])
self.assertTrue(np.allclose(L1, L2))
assert np.allclose(L1, L2)
import os
os.remove("mpi_test__.py")
except:
pass
if __name__ == "__main__":
print("Running unit tests, please be (very) patient...")
try:
import mpi4py
unittest.main()
except:
pass