linting

Author: gkevinzheng

google/cloud/bigtable/admin_v2/overlay/services/bigtable_table_admin/async_client.py

@@ -50,14 +50,18 @@
 from google.cloud.bigtable.admin_v2.services.bigtable_table_admin.transports.base import (
     BigtableTableAdminTransport,
 )
-from google.cloud.bigtable.admin_v2.overlay.types import async_consistency, wait_for_consistency_request
+from google.cloud.bigtable.admin_v2.overlay.types import (
+    async_consistency,
+    wait_for_consistency_request,
+)
 
 from google.cloud.bigtable.gapic_version import __version__ as bigtable_version
 
 
 DEFAULT_CLIENT_INFO = copy.copy(base_client.DEFAULT_CLIENT_INFO)
 DEFAULT_CLIENT_INFO.client_library_version = f"{bigtable_version}-admin-overlay-async"
 
+
 class BigtableTableAdminAsyncClient(base_client.BaseBigtableTableAdminAsyncClient):
     def __init__(
         self,
@@ -129,7 +133,6 @@ def __init__(
             client_info=client_info,
         )
 
-
     async def wait_for_consistency(
         self,
         request: Optional[
@@ -269,8 +272,9 @@ def sample_wait_for_consistency():
         )
 
         # Block and wait until the polling harness returns True.
-        check_consistency_future = async_consistency._AsyncCheckConsistencyPollingFuture(
-            check_consistency_call
+        check_consistency_future = (
+            async_consistency._AsyncCheckConsistencyPollingFuture(
+                check_consistency_call
+            )
         )
-        return (await check_consistency_future.result())
-
+        return await check_consistency_future.result()

tests/unit/admin_overlay/test_async_client.py

@@ -55,7 +55,9 @@
         ),
     ],
 )
-def test_bigtable_table_admin_async_client_client_version(transport_class, transport_name):
+def test_bigtable_table_admin_async_client_client_version(
+    transport_class, transport_name
+):
     with mock.patch.object(transport_class, "__init__") as patched:
         patched.return_value = None
         BigtableTableAdminAsyncClient(transport=transport_name)
@@ -143,8 +145,12 @@ async def test_bigtable_table_admin_async_client_wait_for_consistency(
     check_mock_side_effect = [FALSE_CONSISTENCY_RESPONSE] * (poll_count - 1)
     check_mock_side_effect.append(TRUE_CONSISTENCY_RESPONSE)
 
-    with mock.patch.object(client, "generate_consistency_token", new_callable=mock.AsyncMock) as generate_mock:
-        with mock.patch.object(client, "check_consistency", new_callable=mock.AsyncMock) as check_mock:
+    with mock.patch.object(
+        client, "generate_consistency_token", new_callable=mock.AsyncMock
+    ) as generate_mock:
+        with mock.patch.object(
+            client, "check_consistency", new_callable=mock.AsyncMock
+        ) as check_mock:
             generate_mock.return_value = (
                 bigtable_table_admin.GenerateConsistencyTokenResponse(
                     consistency_token=CONSISTENCY_TOKEN,
@@ -189,15 +195,21 @@ async def test_bigtable_table_admin_async_client_wait_for_consistency_error_in_c
     )
 
     with pytest.raises(exceptions.GoogleAPICallError):
-        with mock.patch.object(client, "generate_consistency_token", new_callable=mock.AsyncMock) as generate_mock:
+        with mock.patch.object(
+            client, "generate_consistency_token", new_callable=mock.AsyncMock
+        ) as generate_mock:
             generate_mock.side_effect = exceptions.DeadlineExceeded(
                 "Deadline Exceeded."
             )
             await client.wait_for_consistency(request)
 
     with pytest.raises(exceptions.GoogleAPICallError):
-        with mock.patch.object(client, "generate_consistency_token", new_callable=mock.AsyncMock) as generate_mock:
-            with mock.patch.object(client, "check_consistency", new_callable=mock.AsyncMock) as check_mock:
+        with mock.patch.object(
+            client, "generate_consistency_token", new_callable=mock.AsyncMock
+        ) as generate_mock:
+            with mock.patch.object(
+                client, "check_consistency", new_callable=mock.AsyncMock
+            ) as check_mock:
                 generate_mock.return_value = (
                     bigtable_table_admin.GenerateConsistencyTokenResponse(
                         consistency_token=CONSISTENCY_TOKEN,

tests/unit/admin_overlay/test_async_consistency.py

@@ -45,7 +45,9 @@ def async_mock_check_consistency_callable(max_poll_count=1):
 
 def test_check_consistency_future_cancel():
     check_consistency_call = async_mock_check_consistency_callable()
-    future = async_consistency._AsyncCheckConsistencyPollingFuture(check_consistency_call)
+    future = async_consistency._AsyncCheckConsistencyPollingFuture(
+        check_consistency_call
+    )
     with pytest.raises(NotImplementedError):
         future.cancel()
 
@@ -57,7 +59,9 @@ def test_check_consistency_future_cancel():
 async def test_check_consistency_future_result():
     times = 5
     check_consistency_call = async_mock_check_consistency_callable(times)
-    future = async_consistency._AsyncCheckConsistencyPollingFuture(check_consistency_call)
+    future = async_consistency._AsyncCheckConsistencyPollingFuture(
+        check_consistency_call
+    )
     is_consistent = await future.result()
 
     assert is_consistent