Skip to content

Commit ae42e8c

Browse files
committed
fix lint
1 parent 7b80215 commit ae42e8c

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

tests/system/small/test_dataframe.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -520,11 +520,10 @@ def test_where_callable_cond_constant_other(scalars_df_index, scalars_pandas_df_
520520
dataframe_bf = scalars_df_index[columns]
521521
dataframe_pd = scalars_pandas_df_index[columns]
522522

523-
cond = lambda x: x > 0
524523
other = 10
525524

526-
bf_result = dataframe_bf.where(cond, other).to_pandas()
527-
pd_result = dataframe_pd.where(cond, other)
525+
bf_result = dataframe_bf.where(lambda x: x > 0, other).to_pandas()
526+
pd_result = dataframe_pd.where(lambda x: x > 0, other)
528527
pandas.testing.assert_frame_equal(bf_result, pd_result)
529528

530529

@@ -554,10 +553,8 @@ def test_where_callable_cond_callable_other(scalars_df_index, scalars_pandas_df_
554553
def func(x):
555554
return x["int64_col"] > 0
556555

557-
other = lambda x: x * 2
558-
559-
bf_result = dataframe_bf.where(func, other).to_pandas()
560-
pd_result = dataframe_pd.where(func, other)
556+
bf_result = dataframe_bf.where(func, lambda x: x * 2).to_pandas()
557+
pd_result = dataframe_pd.where(func, lambda x: x * 2)
561558
pandas.testing.assert_frame_equal(bf_result, pd_result)
562559

563560

0 commit comments

Comments
 (0)