Skip to content

Commit c0510e0

Browse files
committed
fix pyarrow issue
1 parent 55831a1 commit c0510e0

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

pandas/tests/extension/base/interface.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -36,14 +36,15 @@ def test_contains(self, data):
3636
assert scalar in data
3737

3838
na_value = data.dtype.na_value
39-
other_na_value_types = {None, np.nan, pd.NA, pd.NaT}.difference({na_value})
4039

4140
if data.isna().any():
4241
assert na_value in data
4342
else:
4443
assert na_value not in data
4544

46-
for na_value_type in other_na_value_types:
45+
for na_value_type in {None, np.nan, pd.NA, pd.NaT}:
46+
if na_value_type is na_value:
47+
continue
4748
assert na_value_type not in data
4849

4950
def test_memory_usage(self, data):

0 commit comments

Comments
 (0)