Skip to content

Commit 7dde2fd

Browse files
committed
Comply with black
1 parent fa82e07 commit 7dde2fd

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

pandas/tests/groupby/test_grouping.py

+7
Original file line numberDiff line numberDiff line change
@@ -129,12 +129,16 @@ def test_indices_grouped_by_tuple_with_lambda(self):
129129
df = pd.DataFrame(
130130
{"Tuples": ((x, y) for x in [0, 1] for y in np.random.randint(3, 5, 5))}
131131
)
132+
132133
gb = df.groupby("Tuples")
133134
gb_lambda = df.groupby(lambda x: df.iloc[x, 0])
135+
134136
expected = gb.indices
135137
result = gb_lambda.indices
138+
136139
tm.assert_dict_equal(result, expected)
137140

141+
138142
# grouping
139143
# --------------------------------
140144

@@ -790,10 +794,13 @@ def test_get_group_grouped_by_tuple_with_lambda(self):
790794
df = pd.DataFrame(
791795
{"Tuples": ((x, y) for x in [0, 1] for y in np.random.randint(3, 5, 5))}
792796
)
797+
793798
gb = df.groupby("Tuples")
794799
gb_lambda = df.groupby(lambda x: df.iloc[x, 0])
800+
795801
expected = gb.get_group(list(gb.groups.keys())[0])
796802
result = gb_lambda.get_group(list(gb_lambda.groups.keys())[0])
803+
797804
tm.assert_frame_equal(result, expected)
798805

799806
def test_groupby_with_empty(self):

0 commit comments

Comments
 (0)