Add __repr__ to data types of Record, Enum and IList

sopherapps · Feb 27, 2023 · 3bfc61e · 3bfc61e
1 parent 739d37e
commit 3bfc61e
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/funml/data/enum.py b/funml/data/enum.py
@@ -229,6 +229,9 @@ def __str__(self):
         """Generates a readable presentation of the enum."""
         return f"<{self.name}: {self.value}>"
 
+    def __repr__(self):
+        return f"<{self.name}: {self.value}>"
+
 
 def _get_enum_captured_value(instance: Enum):
     """Gets the captured value for a given enum instance"""

diff --git a/funml/data/lists.py b/funml/data/lists.py
@@ -345,6 +345,10 @@ def __str__(self):
         map_to_str = imap(str)
         return f"[{', '.join(map_to_str(self))}]"
 
+    def __repr__(self):
+        map_to_str = imap(str)
+        return f"IList({', '.join(map_to_str(self))})"
+
 
 Q = TypeVar("Q")
 

diff --git a/funml/data/records.py b/funml/data/records.py
@@ -278,6 +278,9 @@ def __str__(self):
         """A readable representation of this type."""
         return f"{self.__attrs}"
 
+    def __repr__(self):
+        return f"{self.__class__.__name__}{self.__attrs}"
+
     def __iter__(self):
         return ((k, v) for k, v in self.__attrs.items())