def __init__(self, ct=None):
"""Constructor.
:param dict ct: change table dict to be sent to MockChangeTable.
"""
if ct is None:
ct = {}
self.change_table = MockChangeTable(ct)
def test_increment_branch_scaling_ref_only(self):
change_table = MockChangeTable(grid=mock_grid, ct=self.ct)
expected_ct = self.ct.copy()
expected_ct["branch"] = {"branch_id": self.orig_branch_scaling.copy()}
self.assertNotEqual(change_table.ct, expected_ct)
_increment_branch_scaling(change_table,
branch_ids=set(),
ref_scenario=self.ref_scenario)
self.assertEqual(change_table.ct, expected_ct)
def test_increment_branch_scaling_ref_and_increment(self):
change_table = MockChangeTable(grid=mock_grid, ct=self.ct)
expected_ct = self.ct.copy()
expected_ct["branch"] = {"branch_id": self.orig_branch_scaling.copy()}
expected_ct["branch"]["branch_id"][102] = 3.5
expected_ct["branch"]["branch_id"][103] = 3
expected_ct["branch"]["branch_id"][107] = 2
self.assertNotEqual(change_table.ct, expected_ct)
_increment_branch_scaling(change_table,
branch_ids={102, 103, 107},
ref_scenario=self.ref_scenario)
self.assertEqual(change_table.ct, expected_ct)