Make state Enum comparable with ==

hydrocomputing · Apr 6, 2024 · f615ddd · f615ddd
1 parent 4a5acc9
commit f615ddd
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/pymf6/api.py b/src/pymf6/api.py
@@ -25,6 +25,9 @@ def show_states(cls):
  # pylint: disable=no-member
  print(*cls._member_names_, sep='\n')
 
+ def __eq__(self, other):
+ return self.value == other.value
+
 
 class Simulator:
  """