Skip to content

Commit

Permalink
Add testing
Browse files Browse the repository at this point in the history
  • Loading branch information
sfc-gh-rdurrani committed Apr 26, 2024
1 parent 42b6f0e commit 36d9460
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions tests/integ/modin/frame/test_where.py
Original file line number Diff line number Diff line change
Expand Up @@ -991,3 +991,14 @@ def perform_where(df):
native_df,
perform_where,
)


@sql_count_checker(query_count=1, join_count=1)
def test_where_series_cond_after_join():
snow_df1 = pd.DataFrame({"A": [1, 2]})
snow_df = snow_df1.join(snow_df1, lsuffix="_l", rsuffix="_r")
snow_df = snow_df.where(snow_df["A_l"] != snow_df["A_r"])
native_df1 = native_pd.DataFrame({"A": [1, 2]})
native_df = native_df1.join(native_df1, lsuffix="_l", rsuffix="_r")
native_df = native_df.where(native_df["A_l"] != native_df["A_r"])
assert_snowpark_pandas_equals_to_pandas_without_dtypecheck(snow_df, native_df)

0 comments on commit 36d9460

Please sign in to comment.