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

more disconnects

parent 2883a7fd
......@@ -56,6 +56,8 @@ def test_gather(backend):
[1., 1., 1., 1., 3., 3., 3., 3.],
[1., 1., 1., 1., 3., 3., 3., 3.]]
))
comm.Disconnect()
else:
rs.num_proc = (2, 2)
......@@ -128,6 +130,8 @@ def test_scatter(backend):
for proc, idx in enumerate(proc_slices):
comm.Recv(res, proc)
assert np.array_equal(res, global_arr[idx])
comm.Disconnect()
else:
rs.num_proc = (2, 2)
......@@ -203,6 +207,8 @@ def test_acc(backend):
)
np.testing.assert_allclose(sim.state.psi / scale, other_psi / scale, rtol=0, atol=1e-5)
comm.Disconnect()
else:
rs.num_proc = (2, 2)
......
......@@ -106,6 +106,7 @@ def cli(run):
time.sleep(0.1)
comm.Disconnect()
return
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