From e2f21962f45c396d730c15696ca23d456b41aa89 Mon Sep 17 00:00:00 2001 From: mslipski Date: Wed, 22 Nov 2023 11:21:37 -0800 Subject: [PATCH] add gt etc comparisons --- mars_time/retimers.py | 19 +++++++++++++++++++ mars_time/tests/test_retimers.py | 6 ++++++ 2 files changed, 25 insertions(+) diff --git a/mars_time/retimers.py b/mars_time/retimers.py index 200d581..d547f81 100644 --- a/mars_time/retimers.py +++ b/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: diff --git a/mars_time/tests/test_retimers.py b/mars_time/tests/test_retimers.py index 66f7448..18dfe5f 100644 --- a/mars_time/tests/test_retimers.py +++ b/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)