Fixed problem with span op comparison tests

Author: BryanCutler

text_extensions_for_pandas/array/span.py

@@ -446,6 +446,9 @@ def __eq__(self, other):
                              "'{}' and '{}'".format(type(self), type(other)))
 
     def __ne__(self, other):
+        if isinstance(other, (ABCDataFrame, ABCSeries, ABCIndexClass)):
+            # Rely on pandas to unbox and dispatch to us.
+            return NotImplemented
         return ~(self == other)
 
     def __hash__(self):

text_extensions_for_pandas/array/test_span.py

@@ -542,12 +542,15 @@ def _compare_other(self, s, data, op_name, other):
             # Compare with scalar
             other = data[0]
 
-        # TODO check result
-        op(data, other)
-
-    @pytest.mark.skip("assert result is NotImplemented")
-    def test_direct_arith_with_series_returns_not_implemented(self, data):
-        pass
+        result = op(data, other)
+
+        if op_name in ["__gt__", "__ne__"]:
+            assert not result[0]
+            assert result[1:].all()
+        elif op_name in ["__lt__", "__eq__"]:
+            assert not result.all()
+        else:
+            raise NotImplementedError("Unknown Operation Comparison")
 
 
 class TestPandasReshaping(base.BaseReshapingTests):

text_extensions_for_pandas/array/test_token_span.py

@@ -560,12 +560,15 @@ def _compare_other(self, s, data, op_name, other):
             # Compare with scalar
             other = data[0]
 
-        # TODO check result
-        op(data, other)
+        result = op(data, other)
 
-    @pytest.mark.skip("assert result is NotImplemented")
-    def test_direct_arith_with_series_returns_not_implemented(self, data):
-        pass
+        if op_name in ["__gt__", "__ne__"]:
+            assert not result[0]
+            assert result[1:].all()
+        elif op_name in ["__lt__", "__eq__"]:
+            assert not result.all()
+        else:
+            raise NotImplementedError("Unknown Operation Comparison")
 
 
 class TestPandasReshaping(base.BaseReshapingTests):