Options: implement __eq__ assert for possible checks

This commit is contained in:
Fabian Dill 2021-08-30 19:08:10 +02:00
parent 3c74f561d5
commit 12957db90f
1 changed files with 2 additions and 0 deletions

View File

@ -149,8 +149,10 @@ class Choice(Option):
def __eq__(self, other):
if type(other) == str:
assert other in self.options
return other == self.current_key
elif type(other) == int:
assert other in self.name_lookup
return other == self.value
elif type(other) == bool:
return other == bool(self.value)