diff --git a/tests/unit/test_connector.py b/tests/unit/test_connector.py index 54d71e27..6afc3666 100644 --- a/tests/unit/test_connector.py +++ b/tests/unit/test_connector.py @@ -220,7 +220,9 @@ async def test_Connector_remove_cached_bad_instance( instance_uri = "projects/test-project/locations/test-region/clusters/test-cluster/instances/bad-test-instance" with Connector(credentials) as connector: connector._keys = asyncio.wrap_future( - asyncio.run_coroutine_threadsafe(generate_keys(), asyncio.get_running_loop()), + asyncio.run_coroutine_threadsafe( + generate_keys(), asyncio.get_running_loop() + ), loop=asyncio.get_running_loop(), ) with pytest.raises(ClientResponseError): @@ -240,7 +242,9 @@ async def test_Connector_remove_cached_no_ip_type(credentials: FakeCredentials) with Connector(credentials=credentials) as connector: connector._client = fake_client connector._keys = asyncio.wrap_future( - asyncio.run_coroutine_threadsafe(generate_keys(), asyncio.get_running_loop()), + asyncio.run_coroutine_threadsafe( + generate_keys(), asyncio.get_running_loop() + ), loop=asyncio.get_running_loop(), ) cache = RefreshAheadCache(instance_uri, fake_client, connector._keys)