Skip to content

Commit 120ced9

Browse files
committed
Blacken
1 parent d8be323 commit 120ced9

File tree

2 files changed

+20
-20
lines changed

2 files changed

+20
-20
lines changed

pandas/tests/groupby/test_grouping.py

Lines changed: 15 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -704,25 +704,27 @@ def test_groupby_multiindex_level_empty(self):
704704
tm.assert_frame_equal(result, expected)
705705

706706
def test_default_observed_deprecated(self):
707-
df = pd.DataFrame([
708-
['A', 1, 1], ['A', 2, 1], ['B', 1, 1]
709-
], columns=['x', 'y', 'z'])
710-
df.x = df.x.astype('category')
711-
df.y = df.x.astype('category')
707+
df = pd.DataFrame(
708+
[["A", 1, 1], ["A", 2, 1], ["B", 1, 1]], columns=["x", "y", "z"]
709+
)
710+
df.x = df.x.astype("category")
711+
df.y = df.x.astype("category")
712712

713-
with tm.assert_produces_warning(expected_warning=FutureWarning,
714-
check_stacklevel=False):
715-
df.groupby(['x', 'y'])
713+
with tm.assert_produces_warning(
714+
expected_warning=FutureWarning, check_stacklevel=False
715+
):
716+
df.groupby(["x", "y"])
716717

717718
with pytest.warns(None) as any_warnings:
718-
df.groupby(['x', 'y'], observed=True)
719-
df.groupby(['x', 'y'], observed=False)
719+
df.groupby(["x", "y"], observed=True)
720+
df.groupby(["x", "y"], observed=False)
720721
assert len(any_warnings) == 0
721722

722-
cat = pd.Categorical(['A', 'B', 'C'], categories=['A', 'B', 'C', 'D'])
723+
cat = pd.Categorical(["A", "B", "C"], categories=["A", "B", "C", "D"])
723724
s = Series(cat)
724-
with tm.assert_produces_warning(expected_warning=FutureWarning,
725-
check_stacklevel=False):
725+
with tm.assert_produces_warning(
726+
expected_warning=FutureWarning, check_stacklevel=False
727+
):
726728
s.groupby(cat)
727729

728730

pandas/tests/reshape/test_pivot.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -84,8 +84,8 @@ def test_pivot_table(self, observed):
8484
index = ["A", "B"]
8585
columns = "C"
8686
table = pivot_table(
87-
self.data, values="D", index=index, columns=columns,
88-
observed=observed)
87+
self.data, values="D", index=index, columns=columns, observed=observed
88+
)
8989

9090
table2 = self.data.pivot_table(
9191
values="D", index=index, columns=columns, observed=observed
@@ -2188,9 +2188,7 @@ def test_pivot_index_list_values_none_immutable_args(self):
21882188
assert columns == ["lev3"]
21892189

21902190
def test_pivot_table_observed_deprecated_default(self):
2191-
with tm.assert_produces_warning(FutureWarning,
2192-
check_stacklevel=False):
2191+
with tm.assert_produces_warning(FutureWarning, check_stacklevel=False):
21932192
# make sure we actually have a category to warn on
2194-
self.data.A = self.data.A.astype('category')
2195-
self.data.pivot_table(values="D", index=["A", "B"],
2196-
columns=["C"])
2193+
self.data.A = self.data.A.astype("category")
2194+
self.data.pivot_table(values="D", index=["A", "B"], columns=["C"])

0 commit comments

Comments
 (0)