TST: pandas-dev#31922 assert no segmentation fault with numpy.array._…

…_contains__ (pandas-dev#36283)
kesmit13 · Nov 2, 2020 · 3170f9c · 3170f9c
1 parent 98e62c5
commit 3170f9c
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