TST: #31922 assert no segmentation fault with numpy.array.__contains__ (

#36283)
pandas-dev · Sep 19, 2020 · c33c3c0 · c33c3c0
1 parent 2705dd6
commit c33c3c0
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/pandas/tests/scalar/test_na_scalar.py b/pandas/tests/scalar/test_na_scalar.py
@@ -305,3 +305,11 @@ def test_pickle_roundtrip_containers(as_frame, values, dtype):
         s = s.to_frame(name="A")
     result = tm.round_trip_pickle(s)
     tm.assert_equal(result, s)
+
+
+@pytest.mark.parametrize("array", [np.array(["a"], dtype=object), ["a"]])
+def test_array_contains_na(array):
+    # GH 31922
+    msg = "boolean value of NA is ambiguous"
+    with pytest.raises(TypeError, match=msg):
+        NA in array