black

Author: tv3141

pandas/tests/groupby/test_counting.py

@@ -3,8 +3,7 @@
 import numpy as np
 import pytest
 
-from pandas import (DataFrame, Index, MultiIndex, Period, Series, Timedelta,
-                    Timestamp)
+from pandas import DataFrame, Index, MultiIndex, Period, Series, Timedelta, Timestamp
 import pandas._testing as tm
 
 
@@ -226,6 +225,7 @@ def test_count_groupby_column_with_nan_in_groupby_column(self):
         # https://github.com/pandas-dev/pandas/issues/32841
         df = DataFrame({"A": [1, 1, 1, 1, 1], "B": [5, 4, np.NaN, 3, 0]})
         res = df.groupby(["B"]).count()
-        expected = DataFrame(index=Index([0.0, 3.0, 4.0, 5.0], name="B"),
-                             data={"A": [1, 1, 1, 1]})
+        expected = DataFrame(
+            index=Index([0.0, 3.0, 4.0, 5.0], name="B"), data={"A": [1, 1, 1, 1]}
+        )
         tm.assert_frame_equal(expected, res)