@@ -2026,47 +2026,48 @@ def test_reset_index_groupby():
20262026 lambda df : df .reset_index ().groupby (["index_0" , "index_1" ]).count (),
20272027 )
20282028
2029+
20292030def test_by_in_index_and_columns ():
20302031 pandas_df = pandas .DataFrame (
2031- [[1 , 2 , 3 ]], index = pd .Index ([0 ], name = "a" ), columns = ['a' , 'b' , 'c' ]
2032+ [[1 , 2 , 3 ]], index = pd .Index ([0 ], name = "a" ), columns = ["a" , "b" , "c" ]
20322033 )
20332034 modin_df = from_pandas (pandas_df )
20342035 eval_general (
20352036 modin_df ,
20362037 pandas_df ,
2037- lambda df : df .groupby (by = 'a' ).count (),
2038+ lambda df : df .groupby (by = "a" ).count (),
20382039 raising_exceptions = True ,
20392040 check_exception_type = True ,
20402041 )
20412042 eval_general (
20422043 modin_df ,
20432044 pandas_df ,
2044- lambda df : df .groupby (by = ['a' , 'b' ]).count (),
2045+ lambda df : df .groupby (by = ["a" , "b" ]).count (),
20452046 raising_exceptions = True ,
20462047 check_exception_type = True ,
20472048 )
20482049 pandas_df = pandas .DataFrame (
2049- [[1 , 2 , 3 ]], index = pd .Index ([(0 , 1 )], names = ["a" , 'b' ]), columns = ['a' , 'b' , 'c' ]
2050+ [[1 , 2 , 3 ]], index = pd .Index ([(0 , 1 )], names = ["a" , "b" ]), columns = ["a" , "b" , "c" ]
20502051 )
20512052 modin_df = from_pandas (pandas_df )
20522053 eval_general (
20532054 modin_df ,
20542055 pandas_df ,
2055- lambda df : df .groupby (by = 'a' ).count (),
2056+ lambda df : df .groupby (by = "a" ).count (),
20562057 raising_exceptions = True ,
20572058 check_exception_type = True ,
20582059 )
20592060 eval_general (
20602061 modin_df ,
20612062 pandas_df ,
2062- lambda df : df .groupby (by = ['a' , 'c' ]).count (),
2063+ lambda df : df .groupby (by = ["a" , "c" ]).count (),
20632064 raising_exceptions = True ,
20642065 check_exception_type = True ,
20652066 )
20662067 eval_general (
20672068 modin_df ,
20682069 pandas_df ,
2069- lambda df : df .groupby (by = ['a' , 'b' ]).count (),
2070+ lambda df : df .groupby (by = ["a" , "b" ]).count (),
20702071 raising_exceptions = True ,
20712072 check_exception_type = True ,
20722073 )
0 commit comments