Skip to content

Commit 3d5ac85

Browse files
committed
another strictness fix
1 parent 783ace9 commit 3d5ac85

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

tests/integ/modin/series/test_apply_and_map.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -380,7 +380,7 @@ def test_null_nan_input(self, method):
380380
# It will become a number column in Snowflake, and in vectorized UDF, we use Int32/64Dtype
381381
# where NULL is encoded as pd.NA
382382
snow_series = pd.Series([None, None])
383-
with SqlCounter(query_count=4, udf_count=1):
383+
with SqlCounter(query_count=5, udf_count=2, strict=False):
384384
types = (
385385
getattr(snow_series, method)(lambda x: str(type(x)))
386386
.to_pandas()

tests/integ/utils/sql_counter.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -462,6 +462,7 @@ def sql_count_checker(
462462
udf_count=None,
463463
udtf_count=None,
464464
union_count=None,
465+
strict=False,
465466
*args,
466467
**kwargs,
467468
):
@@ -497,6 +498,7 @@ def wrapper(*args, **kwargs):
497498
log_stack_trace=False,
498499
high_count_expected=high_count_expected,
499500
high_count_reason=high_count_reason,
501+
strict=strict,
500502
)
501503

502504
result = func(*args, **kwargs)

0 commit comments

Comments
 (0)