feat: add query for extraction station groups for each power unit

src/cimsparql/data_models.py

@@ -3,9 +3,9 @@
 import datetime as dt
 from typing import Self
 
 import pandera as pa
 from pandera.api.pandas.model_config import BaseConfig
 from pandera.typing import DataFrame, Index, Series
 
 
 class CoercingSchema(pa.DataFrameModel):
@@ -423,3 +423,13 @@
 
 
 GenUnitAndSyncMachineMridDataFrame = DataFrame[GenUnitAndSyncMachineMridSchema]
+
+
+class StationGroupForPowerUnitSchema(CoercingSchema):
+    power_system_model_mrid: str = pa.Field(unique=True)
+    market_unit_mrid: str = pa.Field(unique=True)
+    resource_name: str
+    market_code: str
+
+
+StationGroupForPowerUnitDataFrame = DataFrame[StationGroupForPowerUnitSchema]

src/cimsparql/model.py

@@ -41,6 +41,7 @@
     RASEquipmentDataFrame,
     RegionsDataFrame,
     StationGroupCodeNameDataFrame,
+    StationGroupForPowerUnitDataFrame,
     SubstationVoltageDataFrame,
     SvInjectionDataFrame,
     SvPowerDeviationDataFrame,
@@ -697,6 +698,11 @@ def gen_unit_and_sync_machine_mrid(self) -> GenUnitAndSyncMachineMridDataFrame:
         query = self.template_to_query(templates.GEN_UNIT_MRID_AND_SYNC_MACHINE)
         return GenUnitAndSyncMachineMridDataFrame(self.get_table_and_convert(query))
 
+    @time_it
+    def station_group_for_power_unit(self) -> StationGroupForPowerUnitDataFrame:
+        query = self.template_to_query(templates.STATION_GROUP_FOR_POWER_UNIT_QUERY)
+        return StationGroupForPowerUnitDataFrame(self.get_table_and_convert(query))
+
 
 class SingleClientModel(Model):
     def __init__(

src/cimsparql/sparql/station_group_for_power_unit.sparql

@@ -0,0 +1,13 @@
+# Name: station group for power resource
+PREFIX SN: <${SN}>
+PREFIX cim: <${cim}>
+
+select ?power_system_model_mrid  ?market_unit_mrid ?resource_name ?market_code where {
+    ?resource SN:ScheduleResource.marketCode ?market_code .
+    ?unit ?p ?resource;
+        cim:IdentifiedObject.name ?resource_name;
+        cim:IdentifiedObject.mRID ?market_unit_mrid;
+    values ?physical_model_predicate {cim:SynchronousMachine.GeneratingUnit cim:NonConformLoad.LoadGroup}
+    ?physical_object ?physical_model_predicate ?unit;
+        cim:IdentifiedObject.mRID ?power_system_model_mrid .
+}

src/cimsparql/sparql/test_configuration_modifications/add_gen_unit_mrid.sparql

@@ -4,7 +4,9 @@ insert {
     graph <http://cimsparql/xml-adpator/EQ-modifications> {
         _:b0 a SN:GeneratingUnit;
             cim:IdentifiedObject.name "GeneratingUnit" .
-        ?machine cim:SynchronousMachine.GeneratingUnit _:b0
+        ?machine cim:SynchronousMachine.GeneratingUnit _:b0 .
+        _:b0 SN:GeneratingUnit.ScheduleResource _:b1 .
+        _:b1 SN:ScheduleResource.marketCode "market0001"
             }}
 where {
     ?machine a cim:SynchronousMachine

src/cimsparql/templates.py

@@ -37,6 +37,7 @@ def _read_template(filename: pathlib.Path) -> Template:
 REGIONS_QUERY = _read_template(sparql_folder / "regions.sparql")
 SERIES_COMPENSATORS_QUERY = _read_template(sparql_folder / "series_compensators.sparql")
 STATION_GROUP_CODE_NAME_QUERY = _read_template(sparql_folder / "station_group_code_and_names.sparql")
+STATION_GROUP_FOR_POWER_UNIT_QUERY = _read_template(sparql_folder / "station_group_for_power_unit.sparql")
 SUBSTATION_VOLTAGE_LEVEL_QUERY = _read_template(sparql_folder / "substation_voltage_level.sparql")
 SV_BRANCH_QUERY = _read_template(sparql_folder / "sv_branch.sparql")
 SV_INJECTION_QUERY = _read_template(sparql_folder / "sv_injection.sparql")

tests/test_graphdb.py

@@ -75,6 +75,7 @@ async def collect_data(model: Model) -> dict[str, pd.DataFrame]:
         model.series_compensators,
         model.station_group_codes_and_names,
         model.substation_voltage_level,
+        model.station_group_for_power_unit,
         model.sv_power_deviation,
         model.switches,
         model.synchronous_machines,

tests/test_micro_t1_nl.py

@@ -361,3 +361,12 @@ def test_gen_unit_and_sync_machine_mrid(test_model: t_common.ModelTest) -> None:
 
     gen = test_model.model.gen_unit_and_sync_machine_mrid()
     assert len(gen) == 4
+
+
+@pytest.mark.parametrize("test_model", t_entsoe.micro_models())
+def test_station_group_for_power_unit(test_model: t_common.ModelTest) -> None:
+    t_common.check_model(test_model)
+    assert test_model.model
+
+    group_for_units = test_model.model.station_group_for_power_unit()
+    assert len(group_for_units) == 4