[#35] Add str and repr functions to NxMap

mappymatch/constructs/trace.py

@@ -29,12 +29,14 @@ class Trace:
     def __init__(self, frame: GeoDataFrame):
         self._frame = frame
 
+
     def __getitem__(self, i) -> Trace:
         if isinstance(i, int):
             i = [i]
         new_frame = self._frame.iloc[i]
         return Trace(new_frame)
 
+
     def __add__(self, other: Trace) -> Trace:
         if self.crs != other.crs:
             raise TypeError(

mappymatch/maps/nx/nx_map.py

@@ -18,7 +18,6 @@
 DEFAULT_GEOMETRY_KEY = "geometry"
 DEFAULT_ROAD_ID_KEY = "road_id"
 
-
 class NxMap(MapInterface):
     """
     A road map that uses a networkx graph to represent its roads.
@@ -62,6 +61,7 @@ def __init__(self, graph: nx.MultiDiGraph):
         self._nodes = [nid for nid in self.g.nodes()]
         self._roads = self._build_rtree()
 
+
     def _build_rtree(self) -> List[Road]:
         road_lookup = []
 
@@ -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