Merge pull request #966 from PowerGridModel/feature/unit-test-current-sensor

add small unit test for current sensor

Author: mgovers

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():