implement equality operator for daterange
This commit is contained in:
parent
1d8ae65fd8
commit
49de4316f3
1 changed files with 12 additions and 0 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue