add test.

This commit is contained in:
刘棒棒 2024-03-10 23:52:19 +08:00
parent b5a534d617
commit 1e04e34103

View file

@ -104,6 +104,15 @@ async def test_terminate():
time.sleep(2)
assert executor.nb_client.km is None
for _ in range(200):
executor = ExecuteNbCode()
await executor.run(code='print("This is a code!")', language="python")
is_kernel_alive = await executor.nb_client.km.is_alive()
assert is_kernel_alive
await executor.terminate()
assert executor.nb_client.km is None
assert executor.nb_client.kc is None
await executor.terminate()
@pytest.mark.asyncio