Skip to content

Commit 89f66ad

Browse files
committed
cluster test: close clients
1 parent e8739e9 commit 89f66ad

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

ipyparallel/tests/test_cluster.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,6 +63,7 @@ async def test_start_stop_controller(Cluster):
6363
# TODO: wait for connection
6464
client = cluster.connect_client()
6565
assert client.queue_status() == {'unassigned': 0}
66+
client.close()
6667
await cluster.stop_controller()
6768
assert not proc.poll() is not None
6869
assert cluster._controller is None
@@ -105,11 +106,12 @@ async def test_signal_engines(Cluster, engine_launcher_class):
105106
# wait for it to be running
106107
await asyncio.sleep(0.5)
107108
# send signal
108-
await cluster.signal_engines(engine_set_id, signal.SIGINT)
109+
await cluster.signal_engines(signal.SIGINT, engine_set_id)
109110

110111
# wait for result, which should raise KeyboardInterrupt
111112
with raises_remote(KeyboardInterrupt) as e:
112113
ar.get(timeout=10)
114+
rc.close()
113115

114116
await cluster.stop_engines()
115117
await cluster.stop_controller()

0 commit comments

Comments
 (0)