ran blacken

Author: daniel-sanche

tests/unit/data/_async/test__mutate_rows.py

@@ -49,7 +49,9 @@ def _make_one(self, *args, **kwargs):
             kwargs["attempt_timeout"] = kwargs.pop("attempt_timeout", 0.1)
             kwargs["retryable_exceptions"] = kwargs.pop("retryable_exceptions", ())
             kwargs["mutation_entries"] = kwargs.pop("mutation_entries", [])
-            kwargs["metric"] = kwargs.pop("metric", ActiveOperationMetric("MUTATE_ROWS"))
+            kwargs["metric"] = kwargs.pop(
+                "metric", ActiveOperationMetric("MUTATE_ROWS")
+            )
         return self._target_class()(*args, **kwargs)
 
     def _make_mutation(self, count=1, size=1):

tests/unit/data/_async/test__read_rows.py

@@ -273,7 +273,9 @@ async def mock_stream():
         table = mock.Mock()
         table._request_path = {"table_name": "table_name"}
         table.app_profile_id = "app_profile_id"
-        instance = self._make_one(query, table, 10, 10, ActiveOperationMetric("READ_ROWS"))
+        instance = self._make_one(
+            query, table, 10, 10, ActiveOperationMetric("READ_ROWS")
+        )
         assert instance._remaining_count == start_limit
         # read emit_num rows
         async for val in instance.chunk_stream(awaitable_stream()):
@@ -312,7 +314,9 @@ async def mock_stream():
         table = mock.Mock()
         table._request_path = {"table_name": "table_name"}
         table.app_profile_id = "app_profile_id"
-        instance = self._make_one(query, table, 10, 10, ActiveOperationMetric("READ_ROWS"))
+        instance = self._make_one(
+            query, table, 10, 10, ActiveOperationMetric("READ_ROWS")
+        )
         assert instance._remaining_count == start_limit
         with pytest.raises(InvalidChunk) as e:
             # read emit_num rows
@@ -338,7 +342,9 @@ async def mock_stream():
         with mock.patch.object(
             self._get_target_class(), "_read_rows_attempt"
         ) as mock_attempt:
-            instance = self._make_one(mock.Mock(), mock.Mock(), 1, 1, ActiveOperationMetric("READ_ROWS"))
+            instance = self._make_one(
+                mock.Mock(), mock.Mock(), 1, 1, ActiveOperationMetric("READ_ROWS")
+            )
             wrapped_gen = mock_stream()
             mock_attempt.return_value = wrapped_gen
             gen = instance.start_operation()

tests/unit/data/_async/test_client.py

@@ -1957,7 +1957,9 @@ async def test_read_row(self):
         async with self._make_client() as client:
             table = client.get_table("instance", "table")
             row_key = b"test_1"
-            with mock.patch.object(CrossSync, "_ReadRowsOperation") as mock_op_constructor:
+            with mock.patch.object(
+                CrossSync, "_ReadRowsOperation"
+            ) as mock_op_constructor:
                 mock_op = mock.Mock()
                 expected_result = object()
 
@@ -1996,7 +1998,9 @@ async def test_read_row_w_filter(self):
         async with self._make_client() as client:
             table = client.get_table("instance", "table")
             row_key = b"test_1"
-            with mock.patch.object(CrossSync, "_ReadRowsOperation") as mock_op_constructor:
+            with mock.patch.object(
+                CrossSync, "_ReadRowsOperation"
+            ) as mock_op_constructor:
                 mock_op = mock.Mock()
                 expected_result = object()
 
@@ -2037,7 +2041,9 @@ async def test_read_row_no_response(self):
         async with self._make_client() as client:
             table = client.get_table("instance", "table")
             row_key = b"test_1"
-            with mock.patch.object(CrossSync, "_ReadRowsOperation") as mock_op_constructor:
+            with mock.patch.object(
+                CrossSync, "_ReadRowsOperation"
+            ) as mock_op_constructor:
                 mock_op = mock.Mock()
 
                 if CrossSync.is_async:
@@ -2082,7 +2088,9 @@ async def test_row_exists(self, return_value, expected_result):
         async with self._make_client() as client:
             table = client.get_table("instance", "table")
             row_key = b"test_1"
-            with mock.patch.object(CrossSync, "_ReadRowsOperation") as mock_op_constructor:
+            with mock.patch.object(
+                CrossSync, "_ReadRowsOperation"
+            ) as mock_op_constructor:
                 mock_op = mock.Mock()
                 if CrossSync.is_async:
 

tests/unit/data/_async/test_mutations_batcher.py

@@ -307,7 +307,9 @@ def _get_target_class(self):
     def _make_one(self, table=None, **kwargs):
         from google.api_core.exceptions import DeadlineExceeded
         from google.api_core.exceptions import ServiceUnavailable
-        from google.cloud.bigtable.data._metrics import BigtableClientSideMetricsController
+        from google.cloud.bigtable.data._metrics import (
+            BigtableClientSideMetricsController,
+        )
 
         if table is None:
             table = mock.Mock()
@@ -1097,7 +1099,9 @@ async def test_timeout_args_passed(self):
                 assert instance._operation_timeout == expected_operation_timeout
                 assert instance._attempt_timeout == expected_attempt_timeout
                 # make simulated gapic call
-                await instance._execute_mutate_rows([self._make_mutation()], mock.Mock())
+                await instance._execute_mutate_rows(
+                    [self._make_mutation()], mock.Mock()
+                )
                 assert mutate_rows.call_count == 1
                 kwargs = mutate_rows.call_args[1]
                 assert kwargs["operation_timeout"] == expected_operation_timeout
@@ -1196,6 +1200,7 @@ async def test_customizable_retryable_errors(
         down to the gapic layer.
         """
         from google.cloud.bigtable.data._metrics import ActiveOperationMetric
+
         with mock.patch.object(
             google.api_core.retry, "if_exception_type"
         ) as predicate_builder_mock:
@@ -1211,7 +1216,9 @@ async def test_customizable_retryable_errors(
                     predicate_builder_mock.return_value = expected_predicate
                     retry_fn_mock.side_effect = RuntimeError("stop early")
                     mutation = self._make_mutation(count=1, size=1)
-                    await instance._execute_mutate_rows([mutation], ActiveOperationMetric("MUTATE_ROWS"))
+                    await instance._execute_mutate_rows(
+                        [mutation], ActiveOperationMetric("MUTATE_ROWS")
+                    )
                     # passed in errors should be used to build the predicate
                     predicate_builder_mock.assert_called_once_with(
                         *expected_retryables, _MutateRowsIncomplete

tests/unit/data/_async/test_read_rows_acceptance.py

@@ -46,7 +46,6 @@ def _make_operation():
         op._remaining_count = None
         return op
 
-
     @staticmethod
     @CrossSync.convert
     def _get_client_class():
@@ -88,9 +87,7 @@ async def _process_chunks(self, *chunks):
         async def _row_stream():
             yield ReadRowsResponse(chunks=chunks)
 
-        chunker = self._make_operation().chunk_stream(
-            self._coro_wrapper(_row_stream())
-        )
+        chunker = self._make_operation().chunk_stream(self._coro_wrapper(_row_stream()))
         merger = self._make_operation().merge_rows(chunker)
         results = []
         async for row in merger:
@@ -200,9 +197,7 @@ async def _row_stream():
 
         op = self._make_operation()
         op._last_yielded_row_key = b"b"
-        chunker = op.chunk_stream(
-            self._coro_wrapper(_row_stream())
-        )
+        chunker = op.chunk_stream(self._coro_wrapper(_row_stream()))
         merger = self._make_operation().merge_rows(chunker)
         with pytest.raises(InvalidChunk):
             async for _ in merger:

tests/unit/data/_metrics/test_metrics_controller.py

@@ -86,4 +86,4 @@ def test_create_operation(self):
         assert op.is_streaming is expected_is_streaming
         assert op.zone is expected_zone
         assert len(op.handlers) == 1
-        assert op.handlers[0] is handler
+        assert op.handlers[0] is handler