TST: Fix existing tests for groupby

Author: Licht-T

pandas/tests/groupby/test_groupby.py

@@ -1795,8 +1795,8 @@ def test_groupby_level(self):
         frame = self.mframe
         deleveled = frame.reset_index()
 
-        result0 = frame.groupby(level=0).sum()
-        result1 = frame.groupby(level=1).sum()
+        result0 = frame.groupby(level=0, sort=False).sum()
+        result1 = frame.groupby(level=1, sort=False).sum()
 
         expected0 = frame.groupby(deleveled['first'].values).sum()
         expected1 = frame.groupby(deleveled['second'].values).sum()
@@ -1813,15 +1813,15 @@ def test_groupby_level(self):
         assert result1.index.name == frame.index.names[1]
 
         # groupby level name
-        result0 = frame.groupby(level='first').sum()
-        result1 = frame.groupby(level='second').sum()
+        result0 = frame.groupby(level='first', sort=False).sum()
+        result1 = frame.groupby(level='second', sort=False).sum()
         assert_frame_equal(result0, expected0)
         assert_frame_equal(result1, expected1)
 
         # axis=1
 
-        result0 = frame.T.groupby(level=0, axis=1).sum()
-        result1 = frame.T.groupby(level=1, axis=1).sum()
+        result0 = frame.T.groupby(level=0, axis=1, sort=False).sum()
+        result1 = frame.T.groupby(level=1, axis=1, sort=False).sum()
         assert_frame_equal(result0, expected0.T)
         assert_frame_equal(result1, expected1.T)
 
@@ -1843,7 +1843,7 @@ def test_groupby_level_with_nas(self):
         # factorizing doesn't confuse things
         s = Series(np.arange(8.), index=index)
         result = s.groupby(level=0).sum()
-        expected = Series([22., 6.], index=[1, 0])
+        expected = Series([6., 22.], index=[0, 1])
         assert_series_equal(result, expected)
 
         index = MultiIndex(levels=[[1, 0], [0, 1, 2, 3]],
@@ -1853,7 +1853,7 @@ def test_groupby_level_with_nas(self):
         # factorizing doesn't confuse things
         s = Series(np.arange(8.), index=index)
         result = s.groupby(level=0).sum()
-        expected = Series([18., 6.], index=[1, 0])
+        expected = Series([6., 18.], index=[0.0, 1.0])
         assert_series_equal(result, expected)
 
     def test_groupby_level_apply(self):
@@ -1937,7 +1937,7 @@ def test_groupby_complex(self):
         assert_series_equal(result, expected)
 
     def test_level_preserve_order(self):
-        grouped = self.mframe.groupby(level=0)
+        grouped = self.mframe.groupby(level=0, sort=False)
         exp_labels = np.array([0, 0, 0, 1, 1, 2, 2, 3, 3, 3], np.intp)
         assert_almost_equal(grouped.grouper.labels[0], exp_labels)
 

pandas/tests/groupby/test_whitelist.py

@@ -188,13 +188,13 @@ def test_regression_whitelist_methods(raw_frame, op, level, axis, skipna):
         frame = raw_frame.T
 
     if op in AGG_FUNCTIONS_WITH_SKIPNA:
-        grouped = frame.groupby(level=level, axis=axis)
+        grouped = frame.groupby(level=level, axis=axis, sort=False)
         result = getattr(grouped, op)(skipna=skipna)
         expected = getattr(frame, op)(level=level, axis=axis,
                                       skipna=skipna)
         tm.assert_frame_equal(result, expected)
     else:
-        grouped = frame.groupby(level=level, axis=axis)
+        grouped = frame.groupby(level=level, axis=axis, sort=False)
         result = getattr(grouped, op)()
         expected = getattr(frame, op)(level=level, axis=axis)
         tm.assert_frame_equal(result, expected)

pandas/tests/test_multilevel.py

@@ -1395,7 +1395,7 @@ def test_count(self):
     def test_series_group_min_max(self):
         for op, level, skipna in cart_product(self.AGG_FUNCTIONS, lrange(2),
                                               [False, True]):
-            grouped = self.series.groupby(level=level)
+            grouped = self.series.groupby(level=level, sort=False)
             aggf = lambda x: getattr(x, op)(skipna=skipna)
             # skipna=True
             leftside = grouped.agg(aggf)
@@ -1415,7 +1415,7 @@ def test_frame_group_ops(self):
             else:
                 frame = self.frame.T
 
-            grouped = frame.groupby(level=level, axis=axis)
+            grouped = frame.groupby(level=level, axis=axis, sort=False)
 
             pieces = []