Fix period comparison to timedelta

Author: sdispater

pendulum/period.py

@@ -3,7 +3,7 @@
 import operator
 import pendulum
 
-from datetime import datetime, date
+from datetime import datetime, date, timedelta
 
 from pendulum.utils._compat import _HAS_FOLD
 from pendulum.utils._compat import decode
@@ -344,6 +344,17 @@ def __repr__(self):
     def __str__(self):
         return self.__repr__()
 
+    def _cmp(self, other):
+        # Only needed for PyPy
+        assert isinstance(other, timedelta)
+
+        if isinstance(other, Period):
+            other = other.as_timedelta()
+
+        td = self.as_timedelta()
+
+        return 0 if td == other else 1 if td > other else -1
+
     def _getstate(self, protocol=3):
         start, end = self.start, self.end
 

tests/duration/test_behavior.py

@@ -1,10 +1,18 @@
 import pickle
 import pendulum
 
+from datetime import timedelta
+
 
 def test_pickle():
     it = pendulum.duration(days=3, seconds=2456, microseconds=123456)
     s = pickle.dumps(it)
     it2 = pickle.loads(s)
 
     assert it == it2
+
+
+def test_comparison_to_timedelta():
+    duration = pendulum.duration(days=3)
+
+    assert duration < timedelta(days=4)

tests/period/test_behavior.py

@@ -1,6 +1,8 @@
 import pickle
 import pendulum
 
+from datetime import timedelta
+
 
 def test_pickle():
     dt1 = pendulum.datetime(2016, 11, 18)
@@ -29,3 +31,12 @@ def test_pickle():
     assert p.start == p2.start
     assert p.end == p2.end
     assert p.invert == p2.invert
+
+
+def test_comparison_to_timedelta():
+    dt1 = pendulum.datetime(2016, 11, 18)
+    dt2 = pendulum.datetime(2016, 11, 20)
+
+    period = dt2 - dt1
+
+    assert period < timedelta(days=4)