fix pytest for genericbranch

PPSE/17
sudo-ac · Sep 20, 2024 · 2347eb7 · 2347eb7
1 parent bbcec5e
commit 2347eb7
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 5 deletions.
diff --git a/tests/unit/validation/test_input_validation.py b/tests/unit/validation/test_input_validation.py
@@ -415,6 +415,7 @@ def test_validate_input_data_sym_calculation(input_data):
             [
                 "node",
                 "line",
+                "genericbranch",
                 "transformer",
                 "three_winding_transformer",
                 "source",
@@ -433,7 +434,7 @@ def test_validate_input_data_sym_calculation(input_data):
             "sym_power_sensor",
             "measured_object",
             [7, 10],
-            ["line", "transformer"],
+            ["line", "genericbranch", "transformer"],
             {"measured_terminal_type": MeasuredTerminalType.branch_to},
         )
         in validation_errors
@@ -450,7 +451,8 @@ def test_validate_input_data_sym_calculation(input_data):
             [7, 9, 10],
             [
                 "node",
-                "line",
+                "line",   
+                "genericbranch",         
                 "transformer",
                 "three_winding_transformer",
                 "source",
@@ -469,7 +471,7 @@ def test_validate_input_data_sym_calculation(input_data):
             "asym_power_sensor",
             "measured_object",
             [7, 10],
-            ["line", "transformer"],
+            ["line","genericbranch","transformer"],
             {"measured_terminal_type": MeasuredTerminalType.branch_to},
         )
         in validation_errors

diff --git a/tests/unit/validation/test_validation_functions.py b/tests/unit/validation/test_validation_functions.py
@@ -622,8 +622,8 @@ def test_validate_generic_power_sensor__all_terminal_types(
 @pytest.mark.parametrize(
     ("ref_component", "measured_terminal_type"),
     [
-        (["line", "transformer"], MeasuredTerminalType.branch_from),
-        (["line", "transformer"], MeasuredTerminalType.branch_to),
+        (["line","genericbranch", "transformer"], MeasuredTerminalType.branch_from),
+        (["line", "genericbranch", "transformer"], MeasuredTerminalType.branch_to),
         ("source", MeasuredTerminalType.source),
         ("shunt", MeasuredTerminalType.shunt),
         (["sym_load", "asym_load"], MeasuredTerminalType.load),