File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -61,7 +61,7 @@ def test_df_iterrows(native_df):
6161 snowpark_df = pd .DataFrame (native_df )
6262 # One query is used to get the number of rows. One query is used to retrieve each row - each query has 4 JOIN
6363 # operations performed due to iloc.
64- with SqlCounter (query_count = len (native_df ) + 1 ):
64+ with SqlCounter (query_count = len (native_df ) + 2 ):
6565 eval_snowpark_pandas_result (
6666 snowpark_df ,
6767 native_df ,
@@ -70,7 +70,7 @@ def test_df_iterrows(native_df):
7070 )
7171
7272
73- @sql_count_checker (query_count = 8 , union_count = 7 )
73+ @sql_count_checker (query_count = 9 , union_count = 7 )
7474def test_df_iterrows_mixed_types (default_index_native_df ):
7575 # Same test as above on bigger df with mixed types.
7676 # One query is used to get the number of rows. One query is used to retrieve each row - each query has 4 JOIN
@@ -85,7 +85,7 @@ def test_df_iterrows_mixed_types(default_index_native_df):
8585 )
8686
8787
88- @sql_count_checker (query_count = 7 , union_count = 6 )
88+ @sql_count_checker (query_count = 8 , union_count = 6 )
8989def test_df_iterrows_multindex_df ():
9090 # Create df with a MultiIndex index.
9191 # One query is used to get the number of rows. One query is used to retrieve each row - each query has 4 JOIN
You can’t perform that action at this time.
0 commit comments