Commit 0b9c3814 authored by Dion Häfner's avatar Dion Häfner

more disconnects

parent 2883a7fd
...@@ -56,6 +56,8 @@ def test_gather(backend): ...@@ -56,6 +56,8 @@ def test_gather(backend):
[1., 1., 1., 1., 3., 3., 3., 3.], [1., 1., 1., 1., 3., 3., 3., 3.],
[1., 1., 1., 1., 3., 3., 3., 3.]] [1., 1., 1., 1., 3., 3., 3., 3.]]
)) ))
comm.Disconnect()
else: else:
rs.num_proc = (2, 2) rs.num_proc = (2, 2)
...@@ -128,6 +130,8 @@ def test_scatter(backend): ...@@ -128,6 +130,8 @@ def test_scatter(backend):
for proc, idx in enumerate(proc_slices): for proc, idx in enumerate(proc_slices):
comm.Recv(res, proc) comm.Recv(res, proc)
assert np.array_equal(res, global_arr[idx]) assert np.array_equal(res, global_arr[idx])
comm.Disconnect()
else: else:
rs.num_proc = (2, 2) rs.num_proc = (2, 2)
...@@ -203,6 +207,8 @@ def test_acc(backend): ...@@ -203,6 +207,8 @@ def test_acc(backend):
) )
np.testing.assert_allclose(sim.state.psi / scale, other_psi / scale, rtol=0, atol=1e-5) np.testing.assert_allclose(sim.state.psi / scale, other_psi / scale, rtol=0, atol=1e-5)
comm.Disconnect()
else: else:
rs.num_proc = (2, 2) rs.num_proc = (2, 2)
......
...@@ -106,6 +106,7 @@ def cli(run): ...@@ -106,6 +106,7 @@ def cli(run):
time.sleep(0.1) time.sleep(0.1)
comm.Disconnect()
return return
kwargs['override'] = dict(kwargs['override']) kwargs['override'] = dict(kwargs['override'])
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment