Add from_pgm_data

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

Author: Thijss

src/power_grid_model_ds/_core/model/grids/base.py

@@ -440,6 +440,25 @@ def set_feeder_ids(self):
         set_is_feeder(grid=self)
         set_feeder_ids(grid=self)
 
+    @classmethod
+    def from_pgm_data(cls, pgm_data: dict[str, npt.NDArray], load_graphs: bool = True) -> "Grid":
+        """Initialize the grid from PowerGridModel input data.
+
+        Args:
+            pgm_data (dict[str, npt.NDArray]): The PowerGridModel input data.
+        """
+        new_grid = cls.empty()
+        for array_name, array_data in pgm_data.items():
+            pgm_array_class = getattr(new_grid, array_name).__class__
+            pgm_array = pgm_array_class(array_data)
+            setattr(new_grid, array_name, pgm_array)
+
+        new_grid._id_counter = new_grid.max_id
+
+        if load_graphs:
+            new_grid.graphs = GraphContainer.from_arrays(new_grid)
+        return new_grid
+
 
 def _add_branch_array(branch: BranchArray | Branch3Array, grid: Grid):
     """Add a branch array to the grid"""

tests/unit/model/grids/test_grid_base.py

@@ -348,3 +348,10 @@ def test_from_txt_file(self, tmp_path: Path):
         assert 1 == grid.branches.filter(to_status=0).size
         assert 1 == grid.transformer.size
         np.testing.assert_array_equal([14, 10, 11, 12, 13, 15, 16, 17], grid.branches.id)
+
+
+def test_from_pgm_input_data(input_data_pgm):
+    grid = Grid.from_pgm_data(input_data_pgm)
+    assert grid.node.size == 3
+    assert grid.line.size == 2
+    grid.check_ids()