From 54d4f2c2979a54574b3a5a9d2e28da0be71dee56 Mon Sep 17 00:00:00 2001 From: Martijn Govers Date: Wed, 23 Apr 2025 13:03:40 +0200 Subject: [PATCH] add small unit test for current sensor Signed-off-by: Martijn Govers --- tests/unit/test_error_handling.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/tests/unit/test_error_handling.py b/tests/unit/test_error_handling.py index 7e5300902..bc51acca0 100644 --- a/tests/unit/test_error_handling.py +++ b/tests/unit/test_error_handling.py @@ -230,7 +230,8 @@ def test_handle_id_not_found_error(): PowerGridModel(input_data={"node": node_input, "source": source_input}) -def test_handle_invalid_measured_object_error(): +@pytest.mark.parametrize("sensor_type", ["sym_power_sensor", "sym_current_sensor"]) +def test_handle_invalid_measured_object_error(sensor_type): node_input = initialize_array("input", "node", 2) node_input["id"] = [0, 1] node_input["u_rated"] = [0.0, 0.0] @@ -240,13 +241,13 @@ def test_handle_invalid_measured_object_error(): link_input["from_node"] = [0] link_input["to_node"] = [1] - sym_power_sensor_input = initialize_array("input", "sym_power_sensor", 1) - sym_power_sensor_input["id"] = [3] - sym_power_sensor_input["measured_object"] = [2] - sym_power_sensor_input["measured_terminal_type"] = [MeasuredTerminalType.branch_from] + sensor_input = initialize_array("input", sensor_type, 1) + sensor_input["id"] = [3] + sensor_input["measured_object"] = [2] + sensor_input["measured_terminal_type"] = [MeasuredTerminalType.branch_from] with pytest.raises(InvalidMeasuredObject): - PowerGridModel(input_data={"node": node_input, "link": link_input, "sym_power_sensor": sym_power_sensor_input}) + PowerGridModel(input_data={"node": node_input, "link": link_input, sensor_type: sensor_input}) def test_handle_invalid_regulated_object_error():