Skip to content

Commit 66c9d9e

Browse files
committed
TST: fix spurious tests for test_groupby (GH6436)
1 parent af63b99 commit 66c9d9e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pandas/tests/test_groupby.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -961,6 +961,7 @@ def test_frame_groupby(self):
961961
assert_frame_equal(stragged, aggregated, check_names=False)
962962

963963
# transform
964+
grouped = self.tsframe.head(30).groupby(lambda x: x.weekday())
964965
transformed = grouped.transform(lambda x: x - x.mean())
965966
self.assertEqual(len(transformed), 30)
966967
self.assertEqual(len(transformed.columns), 4)
@@ -2203,7 +2204,7 @@ def test_panel_groupby(self):
22032204
grouped = self.panel.groupby(lambda x: x.month, axis='major')
22042205
agged = grouped.mean()
22052206

2206-
self.assert_numpy_array_equal(agged.major_axis, [1, 2])
2207+
self.assert_numpy_array_equal(agged.major_axis, sorted(list(set(self.panel.major_axis.month))))
22072208

22082209
grouped = self.panel.groupby({'A': 0, 'B': 0, 'C': 1, 'D': 1},
22092210
axis='minor')

0 commit comments

Comments
 (0)