class TestSimulator(TestCase):
def setUp(self):
start = Mock(datetime)
network = ChargingNetwork()
evse1 = EVSE("PS-001", max_rate=32)
network.register_evse(evse1, 240, 0)
evse2 = EVSE("PS-002", max_rate=32)
network.register_evse(evse2, 240, 0)
evse3 = EVSE("PS-003", max_rate=32)
network.register_evse(evse3, 240, 0)
self.scheduler = BaseAlgorithm()
self.scheduler.max_recompute = None
events = EventQueue(events=[Event(1), Event(2)])
self.simulator = Simulator(network, self.scheduler, events, start)
self.simulator_no_schedule = Simulator(network, None, events, start)
def test_no_schedule_init(self):
self.assertIsNone(self.simulator_no_schedule.scheduler)
def test_no_schedule_error(self):
with self.assertRaises(TypeError):
self.simulator_no_schedule.run()
def test_correct_on_init_pilot_signals(self):
np.testing.assert_allclose(
self.simulator.pilot_signals,
np.zeros(
(
len(self.simulator.network.station_ids),
self.simulator.event_queue.get_last_timestamp() + 1,
)
),
)
def test_correct_on_init_charging_rates(self):
np.testing.assert_allclose(
self.simulator.charging_rates,
np.zeros(
(
len(self.simulator.network.station_ids),
self.simulator.event_queue.get_last_timestamp() + 1,
)
),
)
def test_correct_on_init_scheduler_data(self):
self.assertIsInstance(self.simulator.scheduler, BaseAlgorithm)
self.assertIsInstance(self.scheduler.interface, Interface)
self.assertIsNone(self.simulator.max_recompute)
def test_update_schedules_not_in_network(self):
new_schedule = {"PS-001": [24, 16], "PS-004": [16, 24]}
with self.assertRaises(KeyError):
self.simulator._update_schedules(new_schedule)
def test_update_schedules_valid_schedule(self):
new_schedule = {"PS-001": [24, 16], "PS-002": [16, 24]}
self.simulator._update_schedules(new_schedule)
np.testing.assert_allclose(
self.simulator.pilot_signals[:, :2], np.array([[24, 16], [16, 24], [0, 0]])
)
def test_index_of_evse_error(self):
with self.assertRaises(KeyError):
_ = self.simulator.index_of_evse("PS-004")
def test_index_of_evse(self):
idx = self.simulator.index_of_evse("PS-002")
self.assertEqual(idx, 1)
def test_pilot_signals_as_df(self):
self.simulator.pilot_signals = np.array([[1, 2], [3, 4], [5, 6]])
outframe = self.simulator.pilot_signals_as_df()
pd.testing.assert_frame_equal(
outframe,
pd.DataFrame(
np.array([[1, 3, 5], [2, 4, 6]]), columns=["PS-001", "PS-002", "PS-003"]
),
)
def test_charging_rates_as_df(self):
self.simulator.charging_rates = np.array([[1.1, 2.1], [3.1, 4.1], [5.1, 6.1]])
outframe = self.simulator.charging_rates_as_df()
pd.testing.assert_frame_equal(
outframe,
pd.DataFrame(
np.array([[1.1, 3.1, 5.1], [2.1, 4.1, 6.1]]),
columns=["PS-001", "PS-002", "PS-003"],
),
)
def test_update_scheduler(self):
new_scheduler = UncontrolledCharging()
self.assertIsNone(new_scheduler._interface)
self.simulator.update_scheduler(new_scheduler)
self.assertIsInstance(self.simulator.scheduler, UncontrolledCharging)
self.assertIsInstance(new_scheduler.interface, Interface)
self.assertEqual(self.simulator.max_recompute, 1)
class TestSimulator(TestCase):
def setUp(self):
start = Mock(datetime)
network = ChargingNetwork()
evse1 = EVSE('PS-001', max_rate=32)
network.register_evse(evse1, 240, 0)
evse2 = EVSE('PS-002', max_rate=32)
network.register_evse(evse2, 240, 0)
evse3 = EVSE('PS-003', max_rate=32)
network.register_evse(evse3, 240, 0)
scheduler = create_autospec(BaseAlgorithm)
scheduler.max_recompute = None
events = EventQueue(events=[Event(1), Event(2)])
self.simulator = Simulator(network, scheduler, events, start)
def test_correct_on_init_pilot_signals(self):
np.testing.assert_allclose(
self.simulator.pilot_signals,
np.zeros((len(self.simulator.network.station_ids),
self.simulator.event_queue.get_last_timestamp() + 1)))
def test_correct_on_init_charging_rates(self):
np.testing.assert_allclose(
self.simulator.charging_rates,
np.zeros((len(self.simulator.network.station_ids),
self.simulator.event_queue.get_last_timestamp() + 1)))
def test_update_schedules_not_in_network(self):
new_schedule = {'PS-001': [24, 16], 'PS-004': [16, 24]}
with self.assertRaises(KeyError):
self.simulator._update_schedules(new_schedule)
def test_update_schedules_valid_schedule(self):
new_schedule = {'PS-001': [24, 16], 'PS-002': [16, 24]}
self.simulator._update_schedules(new_schedule)
np.testing.assert_allclose(self.simulator.pilot_signals[:, :2],
np.array([[24, 16], [16, 24], [0, 0]]))
def test_index_of_evse_error(self):
with self.assertRaises(KeyError):
_ = self.simulator.index_of_evse('PS-004')
def test_index_of_evse(self):
idx = self.simulator.index_of_evse('PS-002')
self.assertEqual(idx, 1)
def test_pilot_signals_as_df(self):
self.simulator.pilot_signals = np.array([[1, 2], [3, 4], [5, 6]])
outframe = self.simulator.pilot_signals_as_df()
pd.testing.assert_frame_equal(
outframe,
pd.DataFrame(np.array([[1, 3, 5], [2, 4, 6]]),
columns=['PS-001', 'PS-002', 'PS-003']))
def test_charging_rates_as_df(self):
self.simulator.charging_rates = np.array([[1.1, 2.1], [3.1, 4.1],
[5.1, 6.1]])
outframe = self.simulator.charging_rates_as_df()
pd.testing.assert_frame_equal(
outframe,
pd.DataFrame(np.array([[1.1, 3.1, 5.1], [2.1, 4.1, 6.1]]),
columns=['PS-001', 'PS-002', 'PS-003']))