@@ -123,9 +123,9 @@ def test_read_filter_join_on_index(session):
123123 )
124124
125125
126- @sql_count_checker (query_count = 3 )
127- def test_read_filter_groupby_agg (session ):
128- # test a chain of operations that are not fully supported in faster pandas
126+ @sql_count_checker (query_count = 3 , join_count = 2 )
127+ def test_read_filter_iloc_index (session ):
128+ # test a chain of operations that are not yet fully supported in faster pandas
129129
130130 # create tables
131131 table_name = Utils .random_name_for_temp_object (TempObjectType .TABLE )
@@ -135,19 +135,19 @@ def test_read_filter_groupby_agg(session):
135135
136136 # create snow dataframes
137137 df = pd .read_snowflake (table_name )
138- snow_result = df [ df [ "B" ] > 11 ]. groupby ( "A" ). min ()
138+ snow_result = df . iloc [[ 0 ], :]
139139
140140 # verify that the input dataframe has a populated relaxed query compiler
141141 assert df ._query_compiler ._relaxed_query_compiler is not None
142142 assert df ._query_compiler ._relaxed_query_compiler ._dummy_row_pos_mode is True
143143 # verify that the output dataframe has an empty relaxed query compiler
144- # because groupby() and min() are not supported in faster pandas yet
144+ # because iloc for index is not supported in faster pandas yet
145145 assert snow_result ._query_compiler ._relaxed_query_compiler is None
146146 assert snow_result ._query_compiler ._dummy_row_pos_mode is False
147147
148148 # create pandas dataframes
149149 native_df = df .to_pandas ()
150- native_result = native_df [ native_df [ "B" ] > 11 ]. groupby ( "A" ). min ()
150+ native_result = native_df . iloc [[ 0 ], :]
151151
152152 # compare results
153153 assert_frame_equal (snow_result , native_result )
0 commit comments