change input type

Signed-off-by: zhen0427 <[email protected]>

Author: zhen0427

src/power_grid_model_io/converters/pgm_json_converter.py

@@ -13,7 +13,7 @@
 from typing import Any, Dict, List, Optional, Union
 
 import numpy as np
-from power_grid_model import DatasetType, initialize_array
+from power_grid_model import initialize_array, DatasetType, ComponentType
 from power_grid_model.data_types import ComponentList, Dataset, SingleDataset, SinglePythonDataset
 from power_grid_model.utils import json_deserialize, json_serialize
 
@@ -93,7 +93,7 @@ def _parse_data(self, data: StructuredData, data_type: DatasetType, extra_info:
         return result
 
     def _parse_dataset(
-        self, data: SinglePythonDataset, data_type: Enum, extra_info: Optional[ExtraInfo]
+        self, data: SinglePythonDataset, data_type: DatasetType, extra_info: Optional[ExtraInfo]
     ) -> SingleDataset:
         """This function parses a single Python dataset and returns a power-grid-model input or update dictionary
 
@@ -120,7 +120,7 @@ def _parse_dataset(
 
     @staticmethod
     def _parse_component(
-        objects: ComponentList, component: Enum, data_type: Enum, extra_info: Optional[ExtraInfo]
+        objects: ComponentList, component: ComponentType, data_type: DatasetType, extra_info: Optional[ExtraInfo]
     ) -> np.ndarray:
         """This function generates a structured numpy array (power-grid-model native) from a structured dataset
 

tests/unit/converters/test_pandapower_converter_input.py

@@ -180,15 +180,15 @@ def test_fill_pgm_extra_info():
     assert extra_info[2] == {"id_reference": {"table": "bus", "index": 103}}
     assert extra_info[3] == {
         "id_reference": {"table": "load", "name": "const_current", "index": 201},
-        "pgm_input": {"node": 0},
+        "pgm_input": {ComponentType.node: 0},
     }
     assert extra_info[4] == {
         "id_reference": {"table": "load", "name": "const_current", "index": 202},
-        "pgm_input": {"node": 1},
+        "pgm_input": {ComponentType.node: 1},
     }
     assert extra_info[5] == {
         "id_reference": {"table": "load", "name": "const_current", "index": 203},
-        "pgm_input": {"node": 2},
+        "pgm_input": {ComponentType.node: 2},
     }
     assert extra_info[6] == {"pgm_input": {"from_node": 0, "to_node": 1, "i_n": 106.0}}
     assert extra_info[7] == {"pgm_input": {"from_node": 1, "to_node": 2, "i_n": 105.0}}