Blacken

Author: jseabold

pandas/tests/groupby/test_grouping.py

@@ -692,25 +692,27 @@ def test_groupby_multiindex_level_empty(self):
         tm.assert_frame_equal(result, expected)
 
     def test_default_observed_deprecated(self):
-        df = pd.DataFrame([
-            ['A', 1, 1], ['A', 2, 1], ['B', 1, 1]
-        ], columns=['x', 'y', 'z'])
-        df.x = df.x.astype('category')
-        df.y = df.x.astype('category')
+        df = pd.DataFrame(
+            [["A", 1, 1], ["A", 2, 1], ["B", 1, 1]], columns=["x", "y", "z"]
+        )
+        df.x = df.x.astype("category")
+        df.y = df.x.astype("category")
 
-        with tm.assert_produces_warning(expected_warning=FutureWarning,
-                                        check_stacklevel=False):
-            df.groupby(['x', 'y'])
+        with tm.assert_produces_warning(
+            expected_warning=FutureWarning, check_stacklevel=False
+        ):
+            df.groupby(["x", "y"])
 
         with pytest.warns(None) as any_warnings:
-            df.groupby(['x', 'y'], observed=True)
-            df.groupby(['x', 'y'], observed=False)
+            df.groupby(["x", "y"], observed=True)
+            df.groupby(["x", "y"], observed=False)
         assert len(any_warnings) == 0
 
-        cat = pd.Categorical(['A', 'B', 'C'], categories=['A', 'B', 'C', 'D'])
+        cat = pd.Categorical(["A", "B", "C"], categories=["A", "B", "C", "D"])
         s = Series(cat)
-        with tm.assert_produces_warning(expected_warning=FutureWarning,
-                                        check_stacklevel=False):
+        with tm.assert_produces_warning(
+            expected_warning=FutureWarning, check_stacklevel=False
+        ):
             s.groupby(cat)
 
 

pandas/tests/reshape/test_pivot.py

@@ -84,8 +84,8 @@ def test_pivot_table(self, observed):
         index = ["A", "B"]
         columns = "C"
         table = pivot_table(
-            self.data, values="D", index=index, columns=columns,
-            observed=observed)
+            self.data, values="D", index=index, columns=columns, observed=observed
+        )
 
         table2 = self.data.pivot_table(
             values="D", index=index, columns=columns, observed=observed
@@ -2077,10 +2077,7 @@ def agg(l):
             foo.pivot_table("notpresent", "X", "Y", aggfunc=agg)
 
     def test_pivot_table_observed_deprecated_default(self):
-        with tm.assert_produces_warning(FutureWarning,
-                                        check_stacklevel=False):
+        with tm.assert_produces_warning(FutureWarning, check_stacklevel=False):
             # make sure we actually have a category to warn on
-            self.data.A = self.data.A.astype('category')
-            self.data.pivot_table(values="D", index=["A", "B"],
-                                  columns=["C"])
-
+            self.data.A = self.data.A.astype("category")
+            self.data.pivot_table(values="D", index=["A", "B"], columns=["C"])