style

Author: TrevorBergeron

tests/system/small/test_index.py

@@ -404,8 +404,9 @@ def test_index_drop_duplicates(scalars_df_index, scalars_pandas_df_index, keep):
 )
 def test_index_contains(scalars_df_index, scalars_pandas_df_index, key):
     col_name = "int64_col"
-    bf_result = scalars_df_index.set_index(col_name).index.__contains__(key)
-    pd_result = scalars_pandas_df_index.set_index(col_name).index.__contains__(key)
+    bf_result = key in scalars_df_index.set_index(col_name).index
+    pd_result = key in scalars_pandas_df_index.set_index(col_name)
+
     assert bf_result == pd_result
 
 

tests/system/small/test_multiindex.py

@@ -1407,6 +1407,7 @@ def test_column_multi_index_w_na_stack(scalars_df_index, scalars_pandas_df_index
 )
 def test_multi_index_contains(scalars_df_index, scalars_pandas_df_index, key):
     col_name = ["int64_col", "bool_col"]
-    bf_result = scalars_df_index.set_index(col_name).index.__contains__(key)
-    pd_result = scalars_pandas_df_index.set_index(col_name).index.__contains__(key)
+    bf_result = key in scalars_df_index.set_index(col_name).index
+    pd_result = key in scalars_pandas_df_index.set_index(col_name).index
+
     assert bf_result == pd_result