[#35] Add str and repr functions to Trace

NREL · May 31, 2022 · f929a1a · f929a1a
1 parent 7d8dadb
commit f929a1a
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/mappymatch/constructs/trace.py b/mappymatch/constructs/trace.py
@@ -40,6 +40,13 @@ def __add__(self, other: Trace) -> Trace:
     def __len__(self):
         return len(self._frame)
 
+    def __str__(self):
+        output_lines = ['Mappymatch Trace object', f'coords_list: {self.coords_list if hasattr(self, "coords_list") else None}', f'frame: {self._frame if hasattr(self, "_frame") else None}']
+        return '\n'.join(output_lines)
+
+    def __repr__(self):
+        return self.__str__()
+
     @property
     def index(self) -> pd.Index:
         return self._frame.index