def scale_renewable_stubs(self, **kwargs):
"""Scales undersized stub branches connected to renewable generators.
Optional kwargs as documented in the
:mod:`powersimdata.design.transmission.upgrade` module.
"""
scale_renewable_stubs(self, **kwargs)
def test_existing_ct_unrelated_branch_id(self):
ct = {"branch": {"branch_id": {105: 2}}}
change_table = MockChangeTable(mock_grid, ct=ct)
expected_ct = {
"branch": {
"branch_id": {
103: (11 / 8),
105: 2,
107: (21 / 15)
}
}
}
scale_renewable_stubs(change_table, verbose=False)
self.assertEqual(change_table.ct, expected_ct)
def test_existing_ct_zone_id_wind(self):
ct = {"wind": {"zone_id": {"Oregon": 2}}}
change_table = MockChangeTable(mock_grid, ct=ct)
expected_ct = {
"wind": {
"zone_id": {
"Oregon": 2
}
},
"branch": {
"branch_id": {
103: (21 / 8),
104: (31 / 25),
107: (21 / 15)
}
},
}
scale_renewable_stubs(change_table, verbose=False)
self.assertEqual(change_table.ct, expected_ct)
def test_empty_ct_no_fuzz(self):
expected_ct = {
"branch": {
"branch_id": {
103: (10 / 8),
107: (20 / 15)
}
}
}
change_table = MockChangeTable(mock_grid)
returned = scale_renewable_stubs(change_table, fuzz=0, verbose=False)
self.assertIsNone(returned)
self.assertEqual(change_table.ct, expected_ct)
def test_empty_ct_inplace_false(self):
expected_ct = {
"branch": {
"branch_id": {
103: (11 / 8),
107: (21 / 15)
}
}
}
change_table = MockChangeTable(mock_grid)
returned = scale_renewable_stubs(change_table,
inplace=False,
verbose=False)
self.assertEqual(change_table.ct, {})
self.assertEqual(returned, expected_ct)
