From d38502edc8fb2385353915ded2e5aa098cd197f0 Mon Sep 17 00:00:00 2001 From: Ben RdO Date: Sat, 10 Sep 2022 17:44:45 -0700 Subject: [PATCH] test add test for storage --- .../input/converter/tests/test_pypsa_to_grid.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/powersimdata/input/converter/tests/test_pypsa_to_grid.py b/powersimdata/input/converter/tests/test_pypsa_to_grid.py index 3e46a6304..c696c5a7c 100644 --- a/powersimdata/input/converter/tests/test_pypsa_to_grid.py +++ b/powersimdata/input/converter/tests/test_pypsa_to_grid.py @@ -3,9 +3,11 @@ import pytest from pandas.testing import assert_series_equal +from powersimdata.input.change_table import ChangeTable from powersimdata.input.converter.pypsa_to_grid import FromPyPSA from powersimdata.input.exporter.export_to_pypsa import export_to_pypsa from powersimdata.input.grid import Grid +from powersimdata.input.transform_grid import TransformGrid @pytest.mark.skipif(find_spec("pypsa") is None, reason="Package PyPSA not available.") @@ -36,7 +38,16 @@ def test_import_network_including_storages_from_pypsa_to_grid(): @pytest.mark.skipif(find_spec("pypsa") is None, reason="Package PyPSA not available.") def test_import_exported_network(): - ref = Grid("Western") + grid = Grid("Western") + ct = ChangeTable(grid) + storage = [ + {"bus_id": 2021005, "capacity": 116.0}, + {"bus_id": 2028827, "capacity": 82.5}, + {"bus_id": 2028060, "capacity": 82.5}, + ] + ct.add_storage_capacity(storage) + ref = TransformGrid(grid, ct.ct).get_grid() + kwargs = dict(add_substations=True, add_load_shedding=False, add_all_columns=True) n = export_to_pypsa(ref, **kwargs) test = FromPyPSA(n, add_pypsa_cols=False)