starting refactor tests

This commit is contained in:
2025-05-04 16:40:10 +03:00
parent adf3431388
commit 477f3a7dec
8 changed files with 57 additions and 49 deletions
+6
View File
@@ -36,6 +36,9 @@ class BaseFlag:
"""
return self._prefix
def __eq__(self, other) -> bool:
return self.get_string_entity() == other.get_string_entity()
class Flag(BaseFlag):
def __init__(self, name: str,
@@ -110,3 +113,6 @@ class InputFlag(BaseFlag):
"""
self._flag_value = value
def __eq__(self, other) -> bool:
return self.get_string_entity() == other.get_string_entity() and self.get_value() == other.get_value()
+9
View File
@@ -58,6 +58,15 @@ class BaseFlags(Generic[FlagType]):
def __getitem__(self, item):
return self._flags[item]
def __eq__(self, other):
if len(self.get_flags()) != len(other.get_flags()):
return False
else:
for flag, other_flag in zip(self.get_flags(), other.get_flags()):
if not flag == other_flag:
return False
return True
class Flags(BaseFlags[Flag]): pass