Skip to content

Commit 9791e1e

Browse files
committed
revert changes to test_transform
1 parent c63a24c commit 9791e1e

File tree

1 file changed

+9
-8
lines changed

1 file changed

+9
-8
lines changed

pandas/tests/groupby/transform/test_transform.py

+9-8
Original file line numberDiff line numberDiff line change
@@ -181,7 +181,7 @@ def test_transform_axis(tsframe):
181181
tm.assert_frame_equal(result, expected)
182182

183183
ts = ts.T
184-
grouped = ts.groupby(lambda x: x.weekday(), axis=1, dropna=False)
184+
grouped = ts.groupby(lambda x: x.weekday(), axis=1)
185185
result = ts - grouped.transform("mean")
186186
expected = grouped.apply(lambda x: (x.T - x.mean(1)).T)
187187
tm.assert_frame_equal(result, expected)
@@ -194,7 +194,7 @@ def test_transform_axis(tsframe):
194194
tm.assert_frame_equal(result, expected)
195195

196196
ts = ts.T
197-
grouped = ts.groupby(lambda x: x.weekday(), axis=1, dropna=False)
197+
grouped = ts.groupby(lambda x: x.weekday(), axis=1)
198198
result = ts - grouped.transform("mean")
199199
expected = grouped.apply(lambda x: (x.T - x.mean(1)).T)
200200
tm.assert_frame_equal(result, expected)
@@ -313,7 +313,7 @@ def test_dispatch_transform(tsframe):
313313

314314
filled = grouped.fillna(method="pad")
315315
fillit = lambda x: x.fillna(method="pad")
316-
expected = df.groupby(lambda x: x.month, dropna=False).transform(fillit)
316+
expected = df.groupby(lambda x: x.month).transform(fillit)
317317
tm.assert_frame_equal(filled, expected)
318318

319319

@@ -412,10 +412,10 @@ def nsum(x):
412412
return np.nansum(x)
413413

414414
results = [
415-
df.groupby("col1", dropna=False).transform(sum)["col2"],
416-
df.groupby("col1", dropna=False)["col2"].transform(sum),
417-
df.groupby("col1", dropna=False).transform(nsum)["col2"],
418-
df.groupby("col1", dropna=False)["col2"].transform(nsum),
415+
df.groupby("col1").transform(sum)["col2"],
416+
df.groupby("col1")["col2"].transform(sum),
417+
df.groupby("col1").transform(nsum)["col2"],
418+
df.groupby("col1")["col2"].transform(nsum),
419419
]
420420
for result in results:
421421
tm.assert_series_equal(result, expected, check_names=False)
@@ -612,7 +612,8 @@ def test_cython_transform_series(op, args, targop):
612612

613613
# series
614614
for data in [s, s_missing]:
615-
expected = data.groupby(labels, dropna=False).transform(targop)
615+
# print(data.head())
616+
expected = data.groupby(labels).transform(targop)
616617

617618
tm.assert_series_equal(expected, data.groupby(labels).transform(op, *args))
618619
tm.assert_series_equal(expected, getattr(data.groupby(labels), op)(*args))

0 commit comments

Comments
 (0)