Update black and change some code so is sees it as a call chain. (#3645)

py-polars/build.requirements.txt

@@ -13,7 +13,7 @@ types-pytz
 maturin==0.12.19
 pytest==7.1.2
 pytest-cov[toml]==3.0.0
-black==21.6b0
+black==22.3.0
 blackdoc==0.3.4
 isort~=5.10.1
 mypy==0.961

py-polars/polars/internals/expr.py

@@ -241,7 +241,7 @@ def sqrt(self) -> "Expr":
         """
         Compute the square root of the elements
         """
-        return self ** 0.5
+        return self**0.5
 
     def log10(self) -> "Expr":
         """

py-polars/polars/internals/series.py

@@ -548,7 +548,7 @@ def sqrt(self) -> "Series":
         ]
 
         """
-        return self ** 0.5
+        return self**0.5
 
     def any(self) -> bool:
         """

py-polars/tests/io/test_lazy_csv.py

@@ -64,5 +64,4 @@ def test_scan_csv_schema_overwrite_and_dtypes_overwrite(foods_csv: str) -> None:
         )
         .collect()
         .dtypes
-        == [pl.Utf8, pl.Utf8, pl.Float32, pl.Int64]
-    )
+    ) == [pl.Utf8, pl.Utf8, pl.Float32, pl.Int64]

py-polars/tests/test_lazy.py

@@ -1236,8 +1236,7 @@ def test_quantile_filtered_agg() -> None:
         .groupby("group")
         .agg(pl.col("value").filter(pl.col("value") < 2).quantile(0.5))["value"]
         .to_list()
-        == [1.0, 1.0]
-    )
+    ) == [1.0, 1.0]
 
 
 def test_lazy_schema() -> None:
@@ -1269,8 +1268,7 @@ def test_deadlocks_3409() -> None:
         )
         .with_columns([pl.col("col1").arr.eval(pl.element().apply(lambda x: x))])
         .to_dict(False)
-        == {"col1": [[1, 2, 3]]}
-    )
+    ) == {"col1": [[1, 2, 3]]}
 
     assert (
         pl.DataFrame(
@@ -1280,5 +1278,4 @@ def test_deadlocks_3409() -> None:
         )
         .with_columns([pl.col("col1").cumulative_eval(pl.element().map(lambda x: 0))])
         .to_dict(False)
-        == {"col1": [0, 0, 0]}
-    )
+    ) == {"col1": [0, 0, 0]}

py-polars/tests/test_queries.py

@@ -88,8 +88,7 @@ def test_overflow_uint16_agg_mean() -> None:
         .groupby(["col1"])
         .agg(pl.col("col3").mean())
         .to_dict(False)
-        == {"col1": ["A"], "col3": [64.0]}
-    )
+    ) == {"col1": ["A"], "col3": [64.0]}
 
 
 def test_binary_on_list_agg_3345() -> None:

py-polars/tests/test_series.py

@@ -91,7 +91,7 @@ def test_init_inputs(monkeypatch: Any) -> None:
     with pytest.raises(ValueError):
         pl.Series({"a": [1, 2, 3]})
     with pytest.raises(OverflowError):
-        pl.Series("bigint", [2 ** 64])
+        pl.Series("bigint", [2**64])
 
 
 def test_concat() -> None:
@@ -202,7 +202,7 @@ def test_arithmetic(s: pl.Series) -> None:
     with pytest.raises(ValueError):
         a % 2
     with pytest.raises(ValueError):
-        a ** 2
+        a**2
     with pytest.raises(ValueError):
         2 / a
     with pytest.raises(ValueError):
@@ -212,7 +212,7 @@ def test_arithmetic(s: pl.Series) -> None:
     with pytest.raises(ValueError):
         2 % a
     with pytest.raises(ValueError):
-        2 ** a
+        2**a
 
 
 def test_add_string() -> None:
@@ -473,7 +473,7 @@ def test_fill_null() -> None:
 
 def test_apply() -> None:
     a = pl.Series("a", [1, 2, None])
-    b = a.apply(lambda x: x ** 2)
+    b = a.apply(lambda x: x**2)
     assert b == [1, 4, None]
 
     a = pl.Series("a", ["foo", "bar", None])
@@ -849,9 +849,9 @@ def test_range() -> None:
 
 def test_strict_cast() -> None:
     with pytest.raises(pl.ComputeError):
-        pl.Series("a", [2 ** 16]).cast(dtype=pl.Int16, strict=True)
+        pl.Series("a", [2**16]).cast(dtype=pl.Int16, strict=True)
     with pytest.raises(pl.ComputeError):
-        pl.DataFrame({"a": [2 ** 16]}).select([pl.col("a").cast(pl.Int16, strict=True)])
+        pl.DataFrame({"a": [2**16]}).select([pl.col("a").cast(pl.Int16, strict=True)])
 
 
 def test_list_concat_dispatch() -> None:

py-polars/tests/test_strings.py

@@ -35,8 +35,7 @@ def test_extract_all_count() -> None:
                 pl.col("foo").str.count_match(r"a").alias("count"),
             ]
         ).to_dict(False)
-        == {"extract": [["a", "a"], None], "count": [2, 0]}
-    )
+    ) == {"extract": [["a", "a"], None], "count": [2, 0]}
 
     assert df["foo"].str.extract_all(r"a").dtype == pl.List
     assert df["foo"].str.count_match(r"a").dtype == pl.UInt32
@@ -71,14 +70,16 @@ def test_zfill() -> None:
 
 def test_ljust_and_rjust() -> None:
     df = pl.DataFrame({"a": ["foo", "longer_foo", "longest_fooooooo", "hi"]})
-    assert df.select(
-        [
-            pl.col("a").str.rjust(10).alias("rjust"),
-            pl.col("a").str.rjust(10).str.lengths().alias("rjust_len"),
-            pl.col("a").str.ljust(10).alias("ljust"),
-            pl.col("a").str.ljust(10).str.lengths().alias("ljust_len"),
-        ]
-    ).to_dict(False) == {
+    assert (
+        df.select(
+            [
+                pl.col("a").str.rjust(10).alias("rjust"),
+                pl.col("a").str.rjust(10).str.lengths().alias("rjust_len"),
+                pl.col("a").str.ljust(10).alias("ljust"),
+                pl.col("a").str.ljust(10).str.lengths().alias("ljust_len"),
+            ]
+        ).to_dict(False)
+    ) == {
         "rjust": ["       foo", "longer_foo", "longest_fooooooo", "        hi"],
         "rjust_len": [10, 10, 16, 10],
         "ljust": ["foo       ", "longer_foo", "longest_fooooooo", "hi        "],

py-polars/tests/test_struct.py

@@ -431,8 +431,7 @@ def test_struct_order() -> None:
                 {"a": 2, "b": 20},
             ],
         ).to_list()
-        == [{"a": 1, "b": None}, {"a": 2, "b": 20}]
-    )
+    ) == [{"a": 1, "b": None}, {"a": 2, "b": 20}]
 
     assert (
         pl.Series(
@@ -441,8 +440,7 @@ def test_struct_order() -> None:
                 {"a": 2, "b": None},
             ],
         ).to_list()
-        == [{"a": 1, "b": 10}, {"a": 2, "b": None}]
-    )
+    ) == [{"a": 1, "b": 10}, {"a": 2, "b": None}]
 
 
 def test_struct_schema_on_append_extend_3452() -> None:

py-polars/tests/test_window.py

@@ -148,12 +148,14 @@ def test_cumulative_eval_window_functions() -> None:
         }
     )
 
-    assert df.with_column(
-        pl.col("val")
-        .cumulative_eval(pl.element().max())
-        .over("group")
-        .alias("cumulative_eval_max")
-    ).to_dict(False) == {
+    assert (
+        df.with_column(
+            pl.col("val")
+            .cumulative_eval(pl.element().max())
+            .over("group")
+            .alias("cumulative_eval_max")
+        ).to_dict(False)
+    ) == {
         "group": [0, 0, 0, 1, 1, 1],
         "val": [20, 40, 30, 2, 4, 3],
         "cumulative_eval_max": [20, 40, 40, 2, 4, 4],
@@ -169,5 +171,4 @@ def test_count_window() -> None:
         )
         .with_column(pl.count().over("a"))["count"]
         .to_list()
-        == [2, 2, 1]
-    )
+    ) == [2, 2, 1]