Skip to content

Commit 8bad697

Browse files
fix per comments
1 parent 808bff4 commit 8bad697

File tree

1 file changed

+8
-5
lines changed

1 file changed

+8
-5
lines changed

pandas/tests/groupby/test_groupby.py

+8-5
Original file line numberDiff line numberDiff line change
@@ -1196,19 +1196,22 @@ def test_groupby_complex():
11961196
tm.assert_series_equal(result, expected)
11971197

11981198

1199-
def test_groupby_complex_2():
1199+
def test_groupby_complex_mean():
1200+
# GH 26475
12001201
df = DataFrame(
12011202
[
12021203
{"a": 2, "b": 1 + 2j},
12031204
{"a": 1, "b": 1 + 1j},
12041205
{"a": 1, "b": 1 + 2j},
12051206
]
12061207
)
1207-
assert df.groupby("b").groups == {(1 + 1j): [1], (1 + 2j): [0, 2]}
1208-
tm.assert_frame_equal(
1209-
df.groupby("b").mean(),
1210-
DataFrame({"a": {(1 + 1j): 1.0, (1 + 2j): 1.5}}),
1208+
result = df.groupby("b").mean()
1209+
expected = DataFrame(
1210+
[[1.0], [1.5]],
1211+
index=Index([(1 + 1j), (1 + 2j)], name="b"),
1212+
columns=Index(["a"]),
12111213
)
1214+
tm.assert_frame_equal(result, expected)
12121215

12131216

12141217
def test_groupby_complex_numbers(using_infer_string):

0 commit comments

Comments
 (0)