NA in arr should return True if arr contains NA

Author: topper-123

pandas/core/arrays/base.py

@@ -361,7 +361,7 @@ def __contains__(self, item) -> bool:
         # comparisons of any item to pd.NA always return pd.NA, so e.g. "a" in [pd.NA]
         # would raise a TypeError. The implementation below works around that.
         if isna(item):
-            return False
+            return isna(self).any() if self._can_hold_na else False
         else:
             return (item == self).any()
 

pandas/tests/arrays/categorical/test_operators.py

@@ -406,4 +406,9 @@ def test_contains(self, ordered):
         cat = Categorical([np.nan, "a"], ordered=ordered)
         assert "a" in cat
         assert "x" not in cat
-        assert pd.NA not in cat
+        assert pd.NA in cat
+
+        cat = cat[::-1]
+        assert "a" in cat
+        assert "x" not in cat
+        assert pd.NA in cat