def test_get(tmp_path: Path):
file = tmp_path / "matrix-daily.txt"
file.write_text("\n\n\n\nmock\tfile")
config = StudyConfig(study_path=file)
serializer = Mock()
serializer.extract_date.return_value = (
pd.Index(["01/02", "01/01"]),
pd.DataFrame(
data={
"01_solar": ["MWh", "EXP", "27000", "48000"],
"02_wind_on": ["MWh", "EXP", "600", "34400"],
}
),
)
matrix = pd.DataFrame(
data={
("01_solar", "MWh", "EXP"): [27000, 48000],
("02_wind_on", "MWh", "EXP"): [600, 34400],
},
index=["01/02", "01/01"],
)
node = OutputSeriesMatrix(config, serializer, AreaHeadWriter("", ""))
assert node.get() == matrix.to_dict(orient="split")
def test_save(tmp_path: Path):
file = tmp_path / "matrix-daily.txt"
config = StudyConfig(study_path=file)
serializer = Mock()
serializer.build_date.return_value = pd.DataFrame(
{
0: ["DE", "", "", "", ""],
1: ["hourly", "", "index", 1, 2],
2: ["", "", "day", "1", "1"],
3: ["", "", "month", "JAN", "JAN"],
4: ["", "", "hourly", "00:00", "01:00"],
}
)
node = OutputSeriesMatrix(
config, serializer, AreaHeadWriter(area="de", freq="hourly")
)
matrix = pd.DataFrame(
data={
("01_solar", "MWh", "EXP"): [27000, 48000],
("02_wind_on", "MWh", "EXP"): [600, 34400],
},
index=["01/01", "01/02"],
)
node.save(matrix.to_dict(orient="split"))
print(file.read_text())
assert (
file.read_text()
== """DE area de hourly
VARIABLES BEGIN END
2 1 2
DE hourly 01_solar 02_wind_on
MWh MWh
index day month hourly EXP EXP
1 1 JAN 00:00 27000 600
2 1 JAN 01:00 48000 34400
"""
)
def __init__(self, config: StudyConfig, freq: str, area: str):
super(OutputSimulationModeMcIndScnAreasItemDetails, self).__init__(
config=config,
date_serializer=FactoryDateSerializer.create(freq, area),
head_writer=AreaHeadWriter(area, freq),
)
def test_area():
writer = AreaHeadWriter(area="de", freq="hourly")
assert (writer.build(var=3, start=2, end=4) ==
"DE\tarea\tde\thourly\n\tVARIABLES\tBEGIN\tEND\n\t3\t2\t4\n\n")