Skip to content

Commit

Permalink
.
Browse files Browse the repository at this point in the history
  • Loading branch information
yirutang committed Mar 19, 2024
1 parent 57d045c commit e54faeb
Showing 1 changed file with 6 additions and 6 deletions.
12 changes: 6 additions & 6 deletions tests/unit/test_writer_v1.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,14 +15,14 @@
import time
from unittest import mock

from google.api_core import exceptions
from google.protobuf import descriptor_pb2
import pytest

from google.cloud.bigquery_storage_v1 import exceptions as bqstorage_exceptions
from google.cloud.bigquery_storage_v1 import gapic_version as package_version
from google.cloud.bigquery_storage_v1 import types as gapic_types
from google.api_core import exceptions
from google.cloud.bigquery_storage_v1.services import big_query_write
from google.cloud.bigquery_storage_v1 import types as gapic_types
from google.cloud.bigquery_storage_v1 import exceptions as bqstorage_exceptions
from google.protobuf import descriptor_pb2


REQUEST_TEMPLATE = gapic_types.AppendRowsRequest()

Expand Down Expand Up @@ -97,7 +97,7 @@ def test_initial_send(background_consumer, bidi_rpc, module_under_test):
),
rows=proto_rows,
),
trace_id=f"python-writer:{package_version.__version__}",
trace_id=f'python-writer:{package_version.__version__}'
)
bidi_rpc.assert_called_once_with(
start_rpc=mock_client.append_rows,
Expand Down

0 comments on commit e54faeb

Please sign in to comment.