Skip to content

Commit 384c45f

Browse files
sfc-gh-rdurranisfc-gh-joshi
authored andcommitted
Add 3 more query counts
1 parent 02c4f38 commit 384c45f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/integ/modin/frame/test_iterrows.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff 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)
7474
def 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)
8989
def 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

0 commit comments

Comments
 (0)