[#35] Add str and repr functions to Trace and NXMap

NREL · Jun 23, 2022 · 68eb738 · 68eb738
1 parent 4229bd2
commit 68eb738
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/mappymatch/constructs/trace.py b/mappymatch/constructs/trace.py
@@ -46,6 +46,17 @@ 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

diff --git a/mappymatch/maps/nx/nx_map.py b/mappymatch/maps/nx/nx_map.py
@@ -85,6 +85,17 @@ def _build_rtree(self) -> List[Road]:
         self.rtree = idx
         return road_lookup
 
+    def __str__(self):
+        output_lines = [
+            "Mappymatch NxMap object",
+            f"roads: {len(self._roads)} Road objects",
+            f"graph: {self.g}",
+        ]
+        return "\n".join(output_lines)
+
+    def __repr__(self):
+        return self.__str__()
+
     @property
     def roads(self) -> List[Road]:
         return self._roads