From c4f87cdccd50d998d6dc3fbb981a902bfd73f384 Mon Sep 17 00:00:00 2001 From: Nuno Campos Date: Wed, 7 Feb 2024 20:02:36 -0800 Subject: [PATCH] Close the connection --- tests/test_pregel_async.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/test_pregel_async.py b/tests/test_pregel_async.py index 37a2f667f..db281e7c6 100644 --- a/tests/test_pregel_async.py +++ b/tests/test_pregel_async.py @@ -527,6 +527,8 @@ async def test_invoke_checkpoint_sqlite(mocker: MockerFixture) -> None: assert checkpoint is not None assert checkpoint["channel_values"].get("total") == 5 + await memory.conn.close() + async def test_invoke_two_processes_two_in_join_two_out(mocker: MockerFixture) -> None: add_one = mocker.Mock(side_effect=lambda x: x + 1)