optimize imports

Author: sakthivelmanii

tests/mockserver_tests/mock_server_test_base.py

@@ -24,24 +24,31 @@
 
 from google.cloud.spanner_v1 import PartialResultSet
 from google.protobuf.duration_pb2 import Duration
-from google.rpc import code_pb2
-from google.rpc import status_pb2
+from google.rpc import (
+    code_pb2,
+    status_pb2
+)
+
 from google.rpc.error_details_pb2 import RetryInfo
 from grpc_status._common import code_to_grpc_status_code
 from grpc_status.rpc_status import _Status
 
 import google.cloud.spanner_v1.types.result_set as result_set
 import google.cloud.spanner_v1.types.type as spanner_type
 from google.cloud.spanner_dbapi.parsed_statement import AutocommitDmlMode
-from google.cloud.spanner_v1 import Client
-from google.cloud.spanner_v1 import FixedSizePool
-from google.cloud.spanner_v1 import ResultSetMetadata
-from google.cloud.spanner_v1 import TypeCode
+from google.cloud.spanner_v1 import (
+    Client,
+    FixedSizePool,
+    ResultSetMetadata,
+    TypeCode
+)
 from google.cloud.spanner_v1.database import Database
 from google.cloud.spanner_v1.instance import Instance
 from google.cloud.spanner_v1.testing.mock_database_admin import DatabaseAdminServicer
-from google.cloud.spanner_v1.testing.mock_spanner import SpannerServicer
-from google.cloud.spanner_v1.testing.mock_spanner import start_mock_server
+from google.cloud.spanner_v1.testing.mock_spanner import (
+    SpannerServicer,
+    start_mock_server
+)
 
 
 # Creates an aborted status with the smallest possible retry delay.

tests/mockserver_tests/test_basics.py

@@ -15,22 +15,24 @@
 from google.cloud.spanner_admin_database_v1.types import spanner_database_admin
 from google.cloud.spanner_dbapi import Connection
 from google.cloud.spanner_dbapi.parsed_statement import AutocommitDmlMode
-from google.cloud.spanner_v1 import BatchCreateSessionsRequest
-from google.cloud.spanner_v1 import BeginTransactionRequest
-from google.cloud.spanner_v1 import ExecuteBatchDmlRequest
-from google.cloud.spanner_v1 import ExecuteSqlRequest
-from google.cloud.spanner_v1 import TransactionOptions
-from google.cloud.spanner_v1 import TypeCode
+from google.cloud.spanner_v1 import (
+    BatchCreateSessionsRequest,
+    BeginTransactionRequest,
+    ExecuteBatchDmlRequest,
+    ExecuteSqlRequest,
+    TransactionOptions,
+    TypeCode
+)
 from google.cloud.spanner_v1.testing.mock_spanner import SpannerServicer
 from google.cloud.spanner_v1.transaction import Transaction
 
 from tests.mockserver_tests.mock_server_test_base import (
     MockServerTestBase,
     _make_partial_result_sets,
-    add_error,
     add_select1_result,
     add_single_result,
     add_update_count,
+    add_error,
     unavailable_status,
     add_execute_streaming_sql_results,
 )