handle NA per suggestion

pandas-dev · jorisvandenbossche · Nov 3, 2020 · Oct 14, 2020 · Oct 14, 2020 · Oct 14, 2020
commit ff640ea8f8ca52059ed85aac1e71ad0955fd97c1
diff --git a/pandas/core/arrays/interval.py b/pandas/core/arrays/interval.py
@@ -13,6 +13,7 @@
     IntervalMixin,
     intervals_to_interval_bounds,
 )
+from pandas._libs.missing import NA
 from pandas._typing import ArrayLike, Dtype
 from pandas.compat.numpy import function as nv
 from pandas.util._decorators import Appender
@@ -577,14 +578,15 @@ def _cmp_method(self, other, op):
 
         # object dtype -> iteratively check for intervals
         result = np.zeros(len(self), dtype=bool)
-        try:
-            for i, obj in enumerate(other):
-                result[i] = op(self[i], obj)
-        except TypeError:
-            # pd.NA
-            result = np.zeros(len(self), dtype=object)
-            for i, obj in enumerate(other):
+        for i, obj in enumerate(other):
+            try:
                 result[i] = op(self[i], obj)
+            except TypeError:
+                if obj is NA:
+                    # github.com/pandas-dev/pandas/pull/37124#discussion_r509095092
+                    result[i] = op is operator.ne
+                else:
+                    raise
         return result
 
     @unpack_zerodim_and_defer("__eq__")

diff --git a/pandas/tests/arithmetic/test_interval.py b/pandas/tests/arithmetic/test_interval.py
@@ -216,6 +216,11 @@ def test_compare_list_like_nan(self, op, array, nulls_fixture, request):
         result = op(array, other)
         expected = self.elementwise_comparison(op, array, other)
 
+        if nulls_fixture is pd.NA and array.dtype.subtype != "i8":
+            reason = "broken for non-integer IntervalArray; see GH 31882"
+            mark = pytest.mark.xfail(reason=reason)
+            request.node.add_marker(mark)
+
         tm.assert_numpy_array_equal(result, expected)
 
     @pytest.mark.parametrize(