counterweight/data-xexe
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

implement equality operator for daterange

Browse source
This commit is contained in:
Pablo Martin 2024-06-11 13:18:20 +02:00
parent 1d8ae65fd8
commit 49de4316f3
1 changed files with 12 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

12
xexe/utils.py
View file

@ -40,3 +40,15 @@ class DateRange:
raise ValueError("end_date can't be before start_date.") raise ValueError("end_date can't be before start_date.")
self._end_date = value 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