From ad287932ef5e42d4e7b96d0d433559027e824932 Mon Sep 17 00:00:00 2001 From: jialuo Date: Tue, 9 Sep 2025 00:06:21 +0000 Subject: [PATCH] fix: Use the remote and managed functions for bigframes results --- tests/system/large/functions/test_managed_function.py | 2 +- tests/system/large/functions/test_remote_function.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/system/large/functions/test_managed_function.py b/tests/system/large/functions/test_managed_function.py index 0a04480a78..dd08ed17d9 100644 --- a/tests/system/large/functions/test_managed_function.py +++ b/tests/system/large/functions/test_managed_function.py @@ -1166,7 +1166,7 @@ def is_sum_positive_series(s): pd_int64_df_filtered = pd_int64_df.dropna() # Test callable condition in dataframe.where method. - bf_result = bf_int64_df_filtered.where(is_sum_positive_series).to_pandas() + bf_result = bf_int64_df_filtered.where(is_sum_positive_series_mf).to_pandas() pd_result = pd_int64_df_filtered.where(is_sum_positive_series) # Ignore any dtype difference. diff --git a/tests/system/large/functions/test_remote_function.py b/tests/system/large/functions/test_remote_function.py index f60786437f..665607f30e 100644 --- a/tests/system/large/functions/test_remote_function.py +++ b/tests/system/large/functions/test_remote_function.py @@ -3066,7 +3066,7 @@ def func_for_other(x): # Test callable condition in dataframe.where method. bf_result = bf_int64_df_filtered.where( - is_sum_positive_series, func_for_other + is_sum_positive_series_mf, func_for_other ).to_pandas() pd_result = pd_int64_df_filtered.where(is_sum_positive_series, func_for_other)