Merge pull request #15 from marek-slipski/add-gt-lt

Add greater than, greater than or equal to, less than, and less than or equal to comparisons.

mars_time/retimers.py

@@ -271,6 +271,25 @@ def __eq__(self, other):
             return False
         else:
             return self.year == other.year and self.sol == other.sol
+
+    def __gt__(self, other):
+        if not isinstance(other, MarsTime):
+            return False
+        else:
+            if self.year == other.year:
+                return self.sol > other.sol
+            else:
+                return self.year > other.year
+
+    def __ge__(self, other):
+        if not isinstance(other, MarsTime):
+            return False
+        else:
+            if self.year == other.year:
+                return self.sol >= other.sol
+            else:
+                return self.year > other.year
+
 
 
 class MarsTimeDelta:

mars_time/tests/test_retimers.py

@@ -22,6 +22,12 @@ def test_two_identical_objects_are_equal(self):
         mt1 = MarsTime(0, 0)
         assert mt0 == mt1
 
+    def test_my_greater_than_other(self):
+        assert MarsTime(33, 10) > MarsTime(32, 10)
+
+    def test_ls_less_than_other(self):
+        assert MarsTime(33, 10) < MarsTime(33, 260)
+
     def test_adding_1_year_gives_expected_result(self):
         mt = MarsTime(0, 0)
         mtd = MarsTimeDelta(year=1)