implement equality operator for daterange

This commit is contained in:
Pablo Martin 2024-06-11 13:18:20 +02:00
parent 1d8ae65fd8
commit 49de4316f3

View file

@ -40,3 +40,15 @@ class DateRange:
raise ValueError("end_date can't be before start_date.")
self._end_date = value
def __eq__(self, other: "DateRange") -> bool:
if not isinstance(other, DateRange):
raise TypeError("DataRange can only be compared to other DateRange.")
start_dates_are_different = self.start_date != other.start_date
end_dates_are_different = self.end_date != other.end_date
if start_dates_are_different or end_dates_are_different:
return False
return True