Add parser tests

Signed-off-by: Thijs Baaijen <[email protected]>

Author: Thijss

src/power_grid_model_ds/_core/visualizer/parsers.py

@@ -21,7 +21,7 @@ def parse_node_array(nodes: NodeArray) -> list[dict[str, Any]]:
         cyto_elements["data"]["id"] = str(node.id.item())
         cyto_elements["data"]["group"] = "node"
         if with_coords:
-            cyto_elements["position"] = {"x": node.x.item(), "y": -node.y.item()}
+            cyto_elements["position"] = {"x": node.x.item(), "y": -node.y.item()}  # invert y-axis for visualization
         parsed_nodes.append(cyto_elements)
     return parsed_nodes
 

tests/integration/visualizer_tests.py

@@ -13,23 +13,23 @@
 from power_grid_model_ds.generators import RadialGridGenerator
 
 
-class TopoNodeArray(NodeArray):
+class CoordinatedNodeArray(NodeArray):
     x: NDArray[np.float64]
     y: NDArray[np.float64]
 
 
 @dataclass
-class TopoGrid(Grid):
-    node: TopoNodeArray
+class CoordinatedGrid(Grid):
+    node: CoordinatedNodeArray
 
 
 def get_radial_grid() -> Grid:
     return RadialGridGenerator(Grid).run()
 
 
-def get_topo_grid() -> TopoGrid:
+def get_coordinated_grid() -> CoordinatedGrid:
     scale = 500
-    grid = TopoGrid.from_txt("S1 2 open", "2 3", "3 4", "S1 500000000", "500000000 6", "6 7 transformer")
+    grid = CoordinatedGrid.from_txt("S1 2 open", "2 3", "3 4", "S1 500000000", "500000000 6", "6 7 transformer")
     grid.node.x = [3, 2.5, 2, 1.5, 3.5, 4, 4.5]
     grid.node.x *= scale
     grid.node.y = [3, 4, 3, 4, 3, 4, 3]
@@ -41,13 +41,13 @@ def visualize_grid():
     visualize(grid=get_radial_grid(), debug=True)
 
 
-def visualize_topo_grid():
+def visualize_coordinated_grid():
     visualize(
-        grid=get_topo_grid(),
+        grid=get_coordinated_grid(),
         debug=True,
     )
 
 
 if __name__ == "__main__":
     visualize_grid()
-    # visualize_topo_grid()
+    # visualize_coordinated_grid()

tests/unit/visualizer/__init__.py

@@ -0,0 +1,59 @@
+from power_grid_model_ds._core.model.arrays import NodeArray, LineArray
+from power_grid_model_ds._core.visualizer.parsers import parse_node_array, parse_branch_array
+from tests.integration.visualizer_tests import CoordinatedNodeArray
+
+
+class TestParseNodeArray:
+
+    def test_parse_node_array(self):
+        nodes = NodeArray.zeros(3)
+        nodes['id'] = [1, 2, 3]
+        nodes["u_rated"] = [10, 20.4, 30.99]
+
+        parsed = parse_node_array(nodes)
+        assert len(parsed) == 3
+
+        node_1_data = parsed[0]['data']
+        node_2_data = parsed[1]['data']
+        node_3_data = parsed[2]['data']
+
+        assert node_1_data['group'] == "node"
+        assert parsed[0].get('position') is None  # no coordinates
+
+        assert node_1_data['id'] == '1'  # ids are converted to strings
+        assert node_2_data['id'] == '2'
+        assert node_3_data['id'] == '3'
+
+        assert node_1_data['u_rated'] == 10
+        assert node_2_data['u_rated'] == 20.4
+        assert node_3_data['u_rated'] == 30.99
+
+
+    def test_parse_coordinated_node_array(self):
+        nodes = CoordinatedNodeArray.zeros(3)
+        nodes['id'] = [1, 2, 3]
+        nodes["x"] = [10, 20, 30]
+        nodes["y"] = [99, 88, 77]
+
+        parsed = parse_node_array(nodes)
+        position = parsed[0].get('position')
+        assert position is not None
+        assert position['x'] == 10
+        assert position['y'] == -99  # coordinates are inverted on y-axis
+
+
+class TestParseBranches:
+
+    def test_parse_line_array(self):
+        lines = LineArray.zeros(3)
+        lines['id'] = [100, 101, 102]
+        lines['from_node'] = [1, 2, 3]
+        lines['to_node'] = [4, 5, 6]
+        parsed = parse_branch_array(lines, "line")
+
+        assert len(parsed) == 3
+        assert parsed[0]['data']['id'] == '100'
+        assert parsed[0]['data']['source'] == '1'
+        assert parsed[0]['data']['target'] == '4'
+        assert parsed[0]['data']['group'] == "line"
+