refactor: update black version to fix lint CI (#130)

Author: tswast

environment.yml

@@ -10,7 +10,7 @@ dependencies:
 - sqlalchemy=1.3.23
 
 # dev
-- black=19.10b0 # Same as ibis
+- black=22.3.0 # Same as ibis
 - pytest
 - pytest-cov
 - pytest-mock

ibis_bigquery/udf/__init__.py

@@ -23,7 +23,7 @@ def validate_output_type(*args):
 
 __all__ = ("udf",)
 
-_udf_name_cache: Dict[str, Iterable[int]] = (collections.defaultdict(itertools.count))
+_udf_name_cache: Dict[str, Iterable[int]] = collections.defaultdict(itertools.count)
 
 
 def create_udf_node(name, fields):

ibis_bigquery/udf/core.py

@@ -70,7 +70,9 @@ def wrapper(*args, **kwargs):
 def rewrite_print(node):
     return ast.Call(
         func=ast.Attribute(
-            value=ast.Name(id="console", ctx=ast.Load()), attr="log", ctx=ast.Load(),
+            value=ast.Name(id="console", ctx=ast.Load()),
+            attr="log",
+            ctx=ast.Load(),
         ),
         args=node.args,
         keywords=node.keywords,
@@ -395,7 +397,9 @@ def visit_Compare(self, node):
     @semicolon
     def visit_AugAssign(self, node):
         return "{} {}= {}".format(
-            self.visit(node.target), self.visit(node.op), self.visit(node.value),
+            self.visit(node.target),
+            self.visit(node.op),
+            self.visit(node.value),
         )
 
     def visit_Module(self, node):
@@ -420,8 +424,7 @@ def visit_Lambda(self, node):
 
     @contextlib.contextmanager
     def local_scope(self):
-        """Assign symbols to local variables.
-        """
+        """Assign symbols to local variables."""
         self.scope = self.scope.new_child()
         try:
             yield self.scope
@@ -444,7 +447,9 @@ def visit_If(self, node):
 
     def visit_IfExp(self, node):
         return "({} ? {} : {})".format(
-            self.visit(node.test), self.visit(node.body), self.visit(node.orelse),
+            self.visit(node.test),
+            self.visit(node.body),
+            self.visit(node.orelse),
         )
 
     def visit_Index(self, node):
@@ -604,6 +609,6 @@ def range(n):
         z = (x if y else b) + 2 + foobar
         foo = Rectangle(1, 2)
         nnn = len(values)
-        return [sum(values) - a + b * y ** -x, z, foo.width, nnn]
+        return [sum(values) - a + b * y**-x, z, foo.width, nnn]
 
     print(my_func.js)

ibis_bigquery/udf/find.py

@@ -4,8 +4,7 @@
 
 
 class NameFinder:
-    """Helper class to find the unique names in an AST.
-    """
+    """Helper class to find the unique names in an AST."""
 
     __slots__ = ()
 

requirements.txt

@@ -5,7 +5,7 @@ google-cloud-bigquery
 pydata-google-auth
 
 # dev
-black==19.10b0
+black==22.3.0
 pytest
 pytest-cov
 pytest-mock

tests/system/conftest.py

@@ -72,14 +72,18 @@ def credentials(default_credentials):
 @pytest.fixture(scope="session")
 def client(credentials, project_id):
     return bq.connect(
-        project_id=project_id, dataset_id=DATASET_ID, credentials=credentials,
+        project_id=project_id,
+        dataset_id=DATASET_ID,
+        credentials=credentials,
     )
 
 
 @pytest.fixture(scope="session")
 def client2(credentials, project_id):
     return bq.connect(
-        project_id=project_id, dataset_id=DATASET_ID, credentials=credentials,
+        project_id=project_id,
+        dataset_id=DATASET_ID,
+        credentials=credentials,
     )
 
 
@@ -184,7 +188,9 @@ def load_functional_alltypes_data(request, bqclient, create_functional_alltypes_
     filepath = download_file("{}/functional_alltypes.csv".format(TESTING_DATA_URI))
     with open(filepath.name, "rb") as csvfile:
         job = bqclient.load_table_from_file(
-            csvfile, table, job_config=load_config,
+            csvfile,
+            table,
+            job_config=load_config,
         ).result()
     if job.error_result:
         print("error")
@@ -238,7 +244,9 @@ def load_functional_alltypes_parted_data(
     filepath = download_file("{}/functional_alltypes.csv".format(TESTING_DATA_URI))
     with open(filepath.name, "rb") as csvfile:
         job = bqclient.load_table_from_file(
-            csvfile, table, job_config=load_config,
+            csvfile,
+            table,
+            job_config=load_config,
         ).result()
     if job.error_result:
         print("error")
@@ -261,7 +269,9 @@ def load_struct_table_data(request, bqclient, struct_bq_table):
     filepath = download_file("{}/struct_table.avro".format(TESTING_DATA_URI))
     with open(filepath.name, "rb") as avrofile:
         job = bqclient.load_table_from_file(
-            avrofile, struct_bq_table, job_config=load_config,
+            avrofile,
+            struct_bq_table,
+            job_config=load_config,
         ).result()
     if job.error_result:
         print("error")

tests/system/test_client.py

@@ -615,7 +615,8 @@ def test_string_to_timestamp(client):
     assert result == timestamp
 
     timestamp_tz = pd.Timestamp(
-        datetime.datetime(year=2017, month=2, day=6, hour=5), tz=pytz.timezone("UTC"),
+        datetime.datetime(year=2017, month=2, day=6, hour=5),
+        tz=pytz.timezone("UTC"),
     )
     expr_tz = ibis.literal("2017-02-06").to_timestamp("%F", "America/New_York")
     result_tz = client.execute(expr_tz)

tests/system/test_connect.py

@@ -27,12 +27,17 @@ def mock_credentials(*args, **kwargs):
         return creds, "default-project-id"
 
     monkeypatch.setattr(pydata_google_auth, "default", mock_credentials)
-    con = ibis_bigquery.connect(project_id="explicit-project-id",)
+    con = ibis_bigquery.connect(
+        project_id="explicit-project-id",
+    )
     assert con.billing_project == "explicit-project-id"
 
 
 def test_without_dataset(project_id, credentials):
-    con = ibis_bigquery.connect(project_id=project_id, credentials=credentials,)
+    con = ibis_bigquery.connect(
+        project_id=project_id,
+        credentials=credentials,
+    )
     with pytest.raises(ValueError, match="Unable to determine BigQuery"):
         con.list_tables()
 
@@ -62,7 +67,8 @@ def mock_default(*args, **kwargs):
     monkeypatch.setattr(pydata_google_auth, "default", mock_default)
 
     ibis_bigquery.connect(
-        project_id=project_id, dataset_id="bigquery-public-data.stackoverflow",
+        project_id=project_id,
+        dataset_id="bigquery-public-data.stackoverflow",
     )
 
     assert len(mock_calls) == 1
@@ -73,7 +79,10 @@ def mock_default(*args, **kwargs):
     auth_local_webserver = kwargs["use_local_webserver"]
     auth_cache = kwargs["credentials_cache"]
     assert not auth_local_webserver
-    assert isinstance(auth_cache, pydata_google_auth.cache.ReadWriteCredentialsCache,)
+    assert isinstance(
+        auth_cache,
+        pydata_google_auth.cache.ReadWriteCredentialsCache,
+    )
 
 
 def test_auth_local_webserver(project_id, credentials, monkeypatch):
@@ -137,7 +146,10 @@ def mock_default(*args, **kwargs):
     assert len(mock_calls) == 1
     _, kwargs = mock_calls[0]
     auth_cache = kwargs["credentials_cache"]
-    assert isinstance(auth_cache, pydata_google_auth.cache.WriteOnlyCredentialsCache,)
+    assert isinstance(
+        auth_cache,
+        pydata_google_auth.cache.WriteOnlyCredentialsCache,
+    )
 
 
 def test_auth_cache_none(project_id, credentials, monkeypatch):

tests/system/udf/test_udf_execute.py

@@ -39,7 +39,8 @@ def my_add(a, b):
 
     expected = (df.double_col + df.double_col).rename("tmp")
     tm.assert_series_equal(
-        result.value_counts().sort_index(), expected.value_counts().sort_index(),
+        result.value_counts().sort_index(),
+        expected.value_counts().sort_index(),
     )
 
 
@@ -216,11 +217,15 @@ def my_len(s):
         param(dt.float64, dt.int64, marks=pytest.mark.xfail(raises=TypeError)),
         # complex argument type, valid return type
         param(
-            dt.Array(dt.int64), dt.float64, marks=pytest.mark.xfail(raises=TypeError),
+            dt.Array(dt.int64),
+            dt.float64,
+            marks=pytest.mark.xfail(raises=TypeError),
         ),
         # valid argument type, complex invalid return type
         param(
-            dt.float64, dt.Array(dt.int64), marks=pytest.mark.xfail(raises=TypeError),
+            dt.float64,
+            dt.Array(dt.int64),
+            marks=pytest.mark.xfail(raises=TypeError),
         ),
         # both invalid
         param(

tests/unit/test_client.py

@@ -8,7 +8,11 @@
     ["project", "dataset", "expected"],
     [
         ("my-project", "", ("my-project", "my-project", "")),
-        ("my-project", "my_dataset", ("my-project", "my-project", "my_dataset"),),
+        (
+            "my-project",
+            "my_dataset",
+            ("my-project", "my-project", "my_dataset"),
+        ),
         (
             "billing-project",
             "data-project.my_dataset",