@@ -195,9 +195,9 @@ async def test_state_lookup(self, mock_appendable_writer):
195195 writer ._is_stream_open = True
196196 writer .write_obj_stream = mock_appendable_writer ["mock_stream" ]
197197
198- mock_appendable_writer [
199- "mock_stream"
200- ]. recv . return_value = storage_type . BidiWriteObjectResponse ( persisted_size = 100 )
198+ mock_appendable_writer ["mock_stream" ]. recv . return_value = (
199+ storage_type . BidiWriteObjectResponse ( persisted_size = 100 )
200+ )
201201
202202 size = await writer .state_lookup ()
203203
@@ -431,9 +431,9 @@ async def test_flush_resets_counters(self, mock_appendable_writer):
431431 writer .write_obj_stream = mock_appendable_writer ["mock_stream" ]
432432 writer .bytes_appended_since_last_flush = 100
433433
434- mock_appendable_writer [
435- "mock_stream"
436- ]. recv . return_value = storage_type . BidiWriteObjectResponse ( persisted_size = 200 )
434+ mock_appendable_writer ["mock_stream" ]. recv . return_value = (
435+ storage_type . BidiWriteObjectResponse ( persisted_size = 200 )
436+ )
437437
438438 await writer .flush ()
439439
@@ -474,9 +474,9 @@ async def test_finalize_lifecycle(self, mock_appendable_writer):
474474 writer .write_obj_stream = mock_appendable_writer ["mock_stream" ]
475475
476476 resource = storage_type .Object (size = 999 )
477- mock_appendable_writer [
478- "mock_stream"
479- ]. recv . return_value = storage_type . BidiWriteObjectResponse ( resource = resource )
477+ mock_appendable_writer ["mock_stream" ]. recv . return_value = (
478+ storage_type . BidiWriteObjectResponse ( resource = resource )
479+ )
480480
481481 res = await writer .finalize ()
482482
@@ -534,9 +534,9 @@ async def test_finalize_with_checksum(self, mock_appendable_writer):
534534 writer .write_obj_stream = mock_appendable_writer ["mock_stream" ]
535535
536536 resource = storage_type .Object (size = 999 )
537- mock_appendable_writer [
538- "mock_stream"
539- ]. recv . return_value = storage_type . BidiWriteObjectResponse ( resource = resource )
537+ mock_appendable_writer ["mock_stream" ]. recv . return_value = (
538+ storage_type . BidiWriteObjectResponse ( resource = resource )
539+ )
540540
541541 checksum = 12345678
542542 res = await writer .finalize (full_object_checksum = checksum )
@@ -655,7 +655,9 @@ async def test_finalize_custom_retry_policy(self, mock_appendable_writer):
655655 writer ._is_stream_open = True
656656 writer .write_obj_stream = mock_appendable_writer ["mock_stream" ]
657657
658- custom_policy = AsyncRetry (predicate = lambda exc : isinstance (exc , exceptions .InternalServerError ))
658+ custom_policy = AsyncRetry (
659+ predicate = lambda exc : isinstance (exc , exceptions .InternalServerError )
660+ )
659661 resource = storage_type .Object (size = 999 )
660662 mock_appendable_writer ["mock_stream" ].recv .return_value = (
661663 storage_type .BidiWriteObjectResponse (resource = resource )
@@ -665,7 +667,9 @@ async def test_finalize_custom_retry_policy(self, mock_appendable_writer):
665667 assert res == resource
666668
667669 @pytest .mark .asyncio
668- async def test_close_with_finalize_and_custom_retry_policy (self , mock_appendable_writer ):
670+ async def test_close_with_finalize_and_custom_retry_policy (
671+ self , mock_appendable_writer
672+ ):
669673 from google .api_core .retry_async import AsyncRetry
670674
671675 writer = self ._make_one (mock_appendable_writer ["mock_client" ])
@@ -697,6 +701,3 @@ async def test_close_retry_on_transient_error(self, mock_appendable_writer):
697701 assert writer .persisted_size == 999
698702 assert mock_appendable_writer ["mock_stream" ].send .await_count == 2
699703 assert not writer ._is_stream_open
700-
701-
702-
0 commit comments