@@ -818,11 +818,17 @@ def test_mean(self):
818818 df = self .get_cars_df ().sort_values (SORT_KEYS )
819819 tbl = self .table .sort_values (SORT_KEYS )
820820
821- dfgrp = df .groupby ('Origin' ).mean ()
821+ if pd_version >= (1 , 1 , 0 ):
822+ dfgrp = df .groupby ('Origin' ).mean (numeric_only = True )
823+ else :
824+ dfgrp = df .groupby ('Origin' ).mean ()
822825 tblgrp = tbl .groupby ('Origin' ).mean ()
823826 self .assertTablesEqual (dfgrp , tblgrp , sortby = None , include_index = True , decimals = 5 )
824827
825- dfgrp = df .groupby ('Origin' , as_index = False ).mean ()
828+ if pd_version >= (1 , 1 , 0 ):
829+ dfgrp = df .groupby ('Origin' , as_index = False ).mean (numeric_only = True )
830+ else :
831+ dfgrp = df .groupby ('Origin' , as_index = False ).mean ()
826832 tblgrp = tbl .groupby ('Origin' , as_index = False ).mean ()
827833 self .assertTablesEqual (dfgrp , tblgrp , sortby = None , decimals = 5 )
828834
@@ -831,7 +837,10 @@ def test_mean(self):
831837 #
832838 swat .options .cas .dataset .bygroup_casout_threshold = 2
833839
834- dfgrp = df .groupby ('Origin' , as_index = False ).mean ()
840+ if pd_version >= (1 , 1 , 0 ):
841+ dfgrp = df .groupby ('Origin' , as_index = False ).mean (numeric_only = True )
842+ else :
843+ dfgrp = df .groupby ('Origin' , as_index = False ).mean ()
835844 tblgrp = tbl .groupby ('Origin' , as_index = False ).mean ()
836845 self .assertEqual (tblgrp .__class__ .__name__ , 'CASTable' )
837846 self .assertTablesEqual (dfgrp , tblgrp ,
@@ -866,11 +875,17 @@ def test_median(self):
866875 df = self .get_cars_df ().sort_values (SORT_KEYS )
867876 tbl = self .table .sort_values (SORT_KEYS )
868877
869- dfgrp = df .groupby ('Origin' ).median ()
878+ if pd_version >= (1 , 1 , 0 ):
879+ dfgrp = df .groupby ('Origin' ).median (numeric_only = True )
880+ else :
881+ dfgrp = df .groupby ('Origin' ).median ()
870882 tblgrp = tbl .groupby ('Origin' ).median ()
871883 self .assertTablesEqual (dfgrp , tblgrp , sortby = None , include_index = True )
872884
873- dfgrp = df .groupby ('Origin' , as_index = False ).median ()
885+ if pd_version >= (1 , 1 , 0 ):
886+ dfgrp = df .groupby ('Origin' , as_index = False ).median (numeric_only = True )
887+ else :
888+ dfgrp = df .groupby ('Origin' , as_index = False ).median ()
874889 tblgrp = tbl .groupby ('Origin' , as_index = False ).median ()
875890 self .assertTablesEqual (dfgrp , tblgrp , sortby = None )
876891
@@ -879,7 +894,10 @@ def test_median(self):
879894 #
880895 swat .options .cas .dataset .bygroup_casout_threshold = 2
881896
882- dfgrp = df .groupby ('Origin' , as_index = False ).median ()
897+ if pd_version >= (1 , 1 , 0 ):
898+ dfgrp = df .groupby ('Origin' , as_index = False ).median (numeric_only = True )
899+ else :
900+ dfgrp = df .groupby ('Origin' , as_index = False ).median ()
883901 tblgrp = tbl .groupby ('Origin' , as_index = False ).median ()
884902 self .assertEqual (tblgrp .__class__ .__name__ , 'CASTable' )
885903 self .assertTablesEqual (dfgrp , tblgrp , sortby = ['Origin' ])
@@ -1028,11 +1046,17 @@ def test_sum(self):
10281046 df = self .get_cars_df ().sort_values (SORT_KEYS )
10291047 tbl = self .table .sort_values (SORT_KEYS )
10301048
1031- dfgrp = df .groupby ('Origin' ).sum ()
1049+ if pd_version >= (1 , 1 , 0 ):
1050+ dfgrp = df .groupby ('Origin' ).sum (numeric_only = True )
1051+ else :
1052+ dfgrp = df .groupby ('Origin' ).sum ()
10321053 tblgrp = tbl .groupby ('Origin' ).sum ()
10331054 self .assertTablesEqual (dfgrp , tblgrp , sortby = None , decimals = 5 )
10341055
1035- dfgrp = df .groupby ('Origin' , as_index = False ).sum ()
1056+ if pd_version >= (1 , 1 , 0 ):
1057+ dfgrp = df .groupby ('Origin' , as_index = False ).sum (numeric_only = True )
1058+ else :
1059+ dfgrp = df .groupby ('Origin' , as_index = False ).sum ()
10361060 tblgrp = tbl .groupby ('Origin' , as_index = False ).sum ()
10371061 self .assertTablesEqual (dfgrp , tblgrp , decimals = 5 , sortby = None )
10381062
@@ -1041,7 +1065,10 @@ def test_sum(self):
10411065 #
10421066 swat .options .cas .dataset .bygroup_casout_threshold = 2
10431067
1044- dfgrp = df .groupby ('Origin' , as_index = False ).sum ()
1068+ if pd_version >= (1 , 1 , 0 ):
1069+ dfgrp = df .groupby ('Origin' , as_index = False ).sum (numeric_only = True )
1070+ else :
1071+ dfgrp = df .groupby ('Origin' , as_index = False ).sum ()
10451072 tblgrp = tbl .groupby ('Origin' , as_index = False ).sum ()
10461073 self .assertEqual (tblgrp .__class__ .__name__ , 'CASTable' )
10471074 self .assertTablesEqual (dfgrp , tblgrp , decimals = 5 ,
@@ -1076,7 +1103,10 @@ def test_std(self):
10761103 df = self .get_cars_df ().sort_values (SORT_KEYS )
10771104 tbl = self .table .sort_values (SORT_KEYS )
10781105
1079- dfgrp = df .groupby ('Origin' ).std ()
1106+ if pd_version >= (1 , 5 , 0 ):
1107+ dfgrp = df .groupby ('Origin' ).std (numeric_only = True )
1108+ else :
1109+ dfgrp = df .groupby ('Origin' ).std ()
10801110 tblgrp = tbl .groupby ('Origin' ).std ()
10811111 self .assertTablesEqual (dfgrp , tblgrp , decimals = 5 , sortby = None )
10821112
@@ -1127,11 +1157,17 @@ def test_var(self):
11271157 df = self .get_cars_df ().sort_values (SORT_KEYS )
11281158 tbl = self .table .sort_values (SORT_KEYS )
11291159
1130- dfgrp = df .groupby ('Origin' ).var ()
1160+ if pd_version >= (1 , 5 , 0 ):
1161+ dfgrp = df .groupby ('Origin' ).var (numeric_only = True )
1162+ else :
1163+ dfgrp = df .groupby ('Origin' ).var ()
11311164 tblgrp = tbl .groupby ('Origin' ).var ()
11321165 self .assertTablesEqual (dfgrp , tblgrp , decimals = 3 , sortby = None )
11331166
1134- dfgrp = df .groupby ('Origin' , as_index = False ).var ()
1167+ if pd_version >= (1 , 5 , 0 ):
1168+ dfgrp = df .groupby ('Origin' , as_index = False ).var (numeric_only = True )
1169+ else :
1170+ dfgrp = df .groupby ('Origin' , as_index = False ).var ()
11351171 tblgrp = tbl .groupby ('Origin' , as_index = False ).var ()
11361172 self .assertTablesEqual (dfgrp , tblgrp , decimals = 3 , sortby = None )
11371173
@@ -1140,7 +1176,10 @@ def test_var(self):
11401176 #
11411177 swat .options .cas .dataset .bygroup_casout_threshold = 2
11421178
1143- dfgrp = df .groupby ('Origin' , as_index = False ).var ()
1179+ if pd_version >= (1 , 5 , 0 ):
1180+ dfgrp = df .groupby ('Origin' , as_index = False ).var (numeric_only = True )
1181+ else :
1182+ dfgrp = df .groupby ('Origin' , as_index = False ).var ()
11441183 tblgrp = tbl .groupby ('Origin' , as_index = False ).var ()
11451184 self .assertEqual (tblgrp .__class__ .__name__ , 'CASTable' )
11461185 self .assertTablesEqual (dfgrp , tblgrp , decimals = 3 ,
0 commit comments