def test_loads_json_error(self):
wrong_json_data = "something weird"
with self.assertRaisesRegex(
DriverEventDeserializationError,
f"Data object {wrong_json_data} is not compatible "
f"with the json.loads method and raises",
):
BaseDriverEvent.loads_json(wrong_json_data)
def test_loads_json(self):
start_data = {
"model_data": {
"parameter_names": ["p1", "p2"],
"kpi_names": ["k1", "k2", "k3"],
},
"id": "force_bdss.events.mco_events.MCOStartEvent",
}
start_event = BaseDriverEvent.loads_json(dumps(start_data))
self.assertIsInstance(start_event, MCOStartEvent)
self.assertDictEqual(start_event.__getstate__(), start_data)
self.assertEqual(start_event.dumps_json(), dumps(start_data))