Skip to content

Commit fc9f21b

Browse files
committed
Add a test for transform
1 parent cfff30d commit fc9f21b

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

pandas/tests/groupby/transform/test_transform.py

+12
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,18 @@ def assert_fp_equal(a, b):
2727
assert (np.abs(a - b) < 1e-12).all()
2828

2929

30+
def test_transform_simple_no_index():
31+
# GH 13519
32+
df = DataFrame({'K': [1, 1, 2, 2], 'V': [1, 2, 3, 4]})
33+
grouped = df.groupby('K', as_index=False)
34+
35+
expected = DataFrame({'K': [1, 1, 2, 2], 'V': [0, 1, 2, 3]})
36+
breakpoint()
37+
actual = grouped.transform(lambda x: x - 1)
38+
39+
tm.assert_frame_equal(expected, actual)
40+
41+
3042
def test_transform():
3143
data = Series(np.arange(9) // 3, index=np.arange(9))
3244

0 commit comments

Comments
 (0)