Skip to content

Commit 3329de1

Browse files
committed
test(bigquery-magics): format and fix code coverage for deprecation unit test
1 parent 15bb190 commit 3329de1

9 files changed

Lines changed: 123 additions & 88 deletions

File tree

packages/bigquery-magics/bigquery_magics/_versions_helpers.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,8 @@
1616

1717
from typing import Any
1818

19-
from google.cloud.bigquery import exceptions
2019
import packaging.version
20+
from google.cloud.bigquery import exceptions
2121

2222
_MIN_BQ_STORAGE_VERSION = packaging.version.Version("2.0.0")
2323

packages/bigquery-magics/bigquery_magics/bigquery.py

Lines changed: 10 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -106,33 +106,33 @@
106106
from __future__ import print_function
107107

108108
import ast
109-
from concurrent import futures
110109
import copy
111110
import json
112111
import re
113112
import sys
114113
import threading
115114
import time
116-
from typing import Any, List, Tuple
117115
import warnings
116+
from concurrent import futures
117+
from typing import Any, List, Tuple
118118

119119
import IPython # type: ignore
120-
from IPython.core import magic_arguments # type: ignore
121-
from IPython.core.getipython import get_ipython
120+
import pandas
122121
from google.api_core.exceptions import NotFound
123122
from google.cloud import bigquery
124123
from google.cloud.bigquery import exceptions
125124
from google.cloud.bigquery.dataset import DatasetReference
126125
from google.cloud.bigquery.dbapi import _helpers
127126
from google.cloud.bigquery.job import QueryJobConfig
128-
import pandas
127+
from IPython.core import magic_arguments # type: ignore
128+
from IPython.core.getipython import get_ipython
129129

130-
from bigquery_magics import core
131-
from bigquery_magics import line_arg_parser as lap
132130
import bigquery_magics._versions_helpers
133131
import bigquery_magics.config
134132
import bigquery_magics.graph_server as graph_server
135133
import bigquery_magics.pyformat
134+
from bigquery_magics import core
135+
from bigquery_magics import line_arg_parser as lap
136136

137137
try:
138138
from google.cloud import bigquery_storage # type: ignore
@@ -471,8 +471,9 @@ def _parse_magic_args(line: str) -> Tuple[List[Any], Any]:
471471

472472
except lap.ParseError as exc:
473473
raise ValueError(
474-
"Unrecognized input, are option values correct? "
475-
"Error details: {}".format(exc.args[0])
474+
"Unrecognized input, are option values correct? Error details: {}".format(
475+
exc.args[0]
476+
)
476477
) from exc
477478

478479
params = []

packages/bigquery-magics/bigquery_magics/config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,9 +12,9 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15+
import warnings
1516
from dataclasses import dataclass
1617
from typing import Optional
17-
import warnings
1818

1919
import google.api_core.client_options as client_options
2020
import google.cloud.bigquery as bigquery

packages/bigquery-magics/bigquery_magics/core.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,9 +18,9 @@
1818
from google.api_core import client_info
1919
from google.cloud import bigquery
2020

21-
from bigquery_magics import environment
2221
import bigquery_magics.config
2322
import bigquery_magics.version
23+
from bigquery_magics import environment
2424

2525
context = bigquery_magics.config.context
2626

packages/bigquery-magics/bigquery_magics/line_arg_parser/lexer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,10 +12,10 @@
1212
# See the License for the specific language governing permissions and
1313
# limitations under the License.
1414

15-
from collections import OrderedDict, namedtuple
1615
import enum
1716
import itertools
1817
import re
18+
from collections import OrderedDict, namedtuple
1919

2020
Token = namedtuple("Token", ("type_", "lexeme", "pos"))
2121
StateTransition = namedtuple("StateTransition", ("new_state", "total_offset"))

packages/bigquery-magics/tests/system/test_bigquery.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,10 +16,10 @@
1616

1717
import re
1818

19-
from IPython.testing import globalipapp
20-
from IPython.utils import io
2119
import pandas
2220
import psutil
21+
from IPython.testing import globalipapp
22+
from IPython.utils import io
2323

2424

2525
def test_bigquery_magic():

0 commit comments

Comments
 (0)