Check if unit /and/ output number are non-negative.

SPF-OST · Nov 3, 2023 · b570cdd · b570cdd
1 parent 6ee8b0a
commit b570cdd
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/trnsys_dck_parser/parse/expression/parse.py b/src/trnsys_dck_parser/parse/expression/parse.py
@@ -101,18 +101,18 @@ def _argument_list(self) -> _tp.Sequence[_exp.ExpressionOrNumber]:
     def _unit_and_output_number(self) -> _tp.Tuple[int, int]:
         self._expect(_tok.Tokens.INTEGER)
         unit_number = int(self._current_token.value)
-        self._check_non_negative(unit_number)
+        self._check_non_negative(unit_number, "Unit number")
 
         self._expect(_tok.Tokens.COMMA)
 
         self._expect(_tok.Tokens.INTEGER)
         output_number = int(self._current_token.value)
-        self._check_non_negative(unit_number)
+        self._check_non_negative(output_number, "Output number")
 
         return unit_number, output_number
 
-    def _check_non_negative(self, integer):
+    def _check_non_negative(self, integer: int, what: str) -> None:
         if integer >= 0:
             return
 
-        self._raise_parsing_error("Unit numbers must be non-negative.")
+        self._raise_parsing_error(f"{what} numbers must be non-negative.")