fix typo

src/dbally/iql/_exceptions.py

@@ -1,5 +1,5 @@
 import ast
-from typing import List, Optional, Union
+from typing import List, Optional
 
 from dbally.exceptions import DbAllyError
 
@@ -31,7 +31,7 @@ def __init__(self, source: str) -> None:
 class IQLMultipleExpressionsError(IQLError):
     """Raised when IQL contains multiple expressions."""
 
-    def __init__(self, nodes: List[Union[ast.stmt, ast.expr]], source: str) -> None:
+    def __init__(self, nodes: List[ast.stmt], source: str) -> None:
         message = "Multiple expressions or statements in IQL are not supported"
         super().__init__(message, source)
         self.nodes = nodes
@@ -41,7 +41,7 @@ class IQLExpressionError(IQLError):
     """Raised when IQL expression is invalid."""
 
     def __init__(self, message: str, node: ast.expr, source: str) -> None:
-        message = message + ": " + source[node.col_offset : node.end_col_offset]
+        message = f"{message}: {source[node.col_offset : node.end_col_offset]}"
         super().__init__(message, source)
         self.node = node
 
@@ -84,7 +84,7 @@ def __init__(self, node: ast.Name, source: str) -> None:
         super().__init__(message, node, source)
 
 
-class IQLIcorrectNumberArgumentsError(IQLExpressionError):
+class IQLIncorrectNumberArgumentsError(IQLExpressionError):
     """Raised when IQL contains too many arguments for a function."""
 
     def __init__(self, node: ast.Call, source: str) -> None:

src/dbally/iql/_processor.py

@@ -8,7 +8,7 @@
     IQLArgumentValidationError,
     IQLEmptyExpressionError,
     IQLFunctionNotExists,
-    IQLIcorrectNumberArgumentsError,
+    IQLIncorrectNumberArgumentsError,
     IQLMultipleExpressionsError,
     IQLNoExpressionError,
     IQLSyntaxError,
@@ -93,7 +93,7 @@ async def _parse_call(self, node: ast.Call) -> syntax.FunctionCall:
         args = []
 
         if len(func_def.parameters) != len(node.args):
-            raise IQLIcorrectNumberArgumentsError(node, self.source)
+            raise IQLIncorrectNumberArgumentsError(node, self.source)
 
         for arg, arg_def in zip(node.args, func_def.parameters):
             arg_value = self._parse_arg(arg)

tests/unit/iql/test_iql_parser.py

@@ -8,7 +8,7 @@
     IQLArgumentValidationError,
     IQLEmptyExpressionError,
     IQLFunctionNotExists,
-    IQLIcorrectNumberArgumentsError,
+    IQLIncorrectNumberArgumentsError,
     IQLMultipleExpressionsError,
     IQLNoExpressionError,
     IQLSyntaxError,
@@ -185,7 +185,7 @@ async def test_iql_parser_method_not_exists():
 
 
 async def test_iql_parser_incorrect_number_of_arguments_fail():
-    with pytest.raises(IQLIcorrectNumberArgumentsError) as exc_info:
+    with pytest.raises(IQLIncorrectNumberArgumentsError) as exc_info:
         await IQLQuery.parse(
             "filter_by_age('too old', 40)",
             allowed_functions=[