vtlengine 1.0.2__py3-none-any.whl → 1.0.3__py3-none-any.whl

vtlengine/AST/ASTConstructorModules/ExprComponents.py

@@ -1,23 +1,23 @@
 from antlr4.tree.Tree import TerminalNodeImpl
 
 from vtlengine.AST import (
+    ID,
     Aggregation,
-    If,
+    Analytic,
     BinOp,
-    UnaryOp,
-    ID,
-    ParamOp,
-    MulOp,
+    Case,
+    CaseObj,
     Constant,
+    EvalOp,
+    Identifier,
+    If,
+    MulOp,
     ParamConstant,
+    ParamOp,
     TimeAggregation,
-    Identifier,
-    EvalOp,
-    VarID,
-    Analytic,
     UDOCall,
-    Case,
-    CaseObj,
+    UnaryOp,
+    VarID,
 )
 from vtlengine.AST.ASTConstructorModules.Terminals import Terminals
 from vtlengine.AST.Grammar.parser import Parser
@@ -36,20 +36,20 @@ class ExprComp(VtlVisitor):
     def visitExprComponent(self, ctx: Parser.ExprComponentContext):
         """
         exprComponent:
-            LPAREN exprComponent RPAREN                                                                             # parenthesisExprComp # noqa E501
-            | functionsComponents                                                                                   # functionsExpressionComp # noqa E501
-            | op=(PLUS|MINUS|NOT) right=exprComponent                                                               # unaryExprComp # noqa E501
-            | left=exprComponent op=(MUL|DIV) right=exprComponent                                                   # arithmeticExprComp # noqa E501
-            | left=exprComponent op=(PLUS|MINUS|CONCAT) right=exprComponent                                         # arithmeticExprOrConcatComp # noqa E501
-            | left=exprComponent comparisonOperand right=exprComponent                                              # comparisonExprComp # noqa E501
-            | left=exprComponent op=(IN|NOT_IN)(lists|valueDomainID)                                                # inNotInExprComp # noqa E501
-            | left=exprComponent op=AND right=exprComponent                                                         # booleanExprComp # noqa E501
-            | left=exprComponent op=(OR|XOR) right=exprComponent                                                    # booleanExprComp # noqa E501
-            | IF  conditionalExpr=exprComponent  THEN thenExpr=exprComponent ELSE elseExpr=exprComponent            # ifExprComp # noqa E501
-            | constant                                                                                              # constantExprComp # noqa E501
-            | componentID                                                                                           # compId # noqa E501
+            LPAREN exprComponent RPAREN                                                                             # parenthesisExprComp
+            | functionsComponents                                                                                   # functionsExpressionComp
+            | op=(PLUS|MINUS|NOT) right=exprComponent                                                               # unaryExprComp
+            | left=exprComponent op=(MUL|DIV) right=exprComponent                                                   # arithmeticExprComp
+            | left=exprComponent op=(PLUS|MINUS|CONCAT) right=exprComponent                                         # arithmeticExprOrConcatComp
+            | left=exprComponent comparisonOperand right=exprComponent                                              # comparisonExprComp
+            | left=exprComponent op=(IN|NOT_IN)(lists|valueDomainID)                                                # inNotInExprComp
+            | left=exprComponent op=AND right=exprComponent                                                         # booleanExprComp
+            | left=exprComponent op=(OR|XOR) right=exprComponent                                                    # booleanExprComp
+            | IF  conditionalExpr=exprComponent  THEN thenExpr=exprComponent ELSE elseExpr=exprComponent            # ifExprComp
+            | constant                                                                                              # constantExprComp
+            | componentID                                                                                           # compId
         ;
-        """
+        """ # noqa E501
         ctx_list = list(ctx.getChildren())
         c = ctx_list[0]
 
@@ -279,8 +279,8 @@ class ExprComp(VtlVisitor):
 
     def visitCallComponent(self, ctx: Parser.CallComponentContext):
         """
-        callFunction: operatorID LPAREN (parameterComponent (COMMA parameterComponent)*)? RPAREN    # callComponent  # noqa E501
-        """
+        callFunction: operatorID LPAREN (parameterComponent (COMMA parameterComponent)*)? RPAREN    # callComponent
+        """ # noqa E501
         ctx_list = list(ctx.getChildren())
         c = ctx_list[0]
 
@@ -295,8 +295,8 @@ class ExprComp(VtlVisitor):
 
     def visitEvalAtomComponent(self, ctx: Parser.EvalAtomComponentContext):
         """
-        | EVAL LPAREN routineName LPAREN (componentID|scalarItem)? (COMMA (componentID|scalarItem))* RPAREN (LANGUAGE STRING_CONSTANT)? (RETURNS outputParameterTypeComponent)? RPAREN      # evalAtomComponent  # noqa E501
-        """
+        | EVAL LPAREN routineName LPAREN (componentID|scalarItem)? (COMMA (componentID|scalarItem))* RPAREN (LANGUAGE STRING_CONSTANT)? (RETURNS outputParameterTypeComponent)? RPAREN      # evalAtomComponent
+        """ # noqa E501
         ctx_list = list(ctx.getChildren())
 
         routine_name = Terminals().visitRoutineName(ctx_list[2])
@@ -346,8 +346,8 @@ class ExprComp(VtlVisitor):
 
     def visitCastExprComponent(self, ctx: Parser.CastExprComponentContext):
         """
-        | CAST LPAREN exprComponent COMMA (basicScalarType|valueDomainName) (COMMA STRING_CONSTANT)? RPAREN         # castExprComponent  # noqa E501
-        """
+        | CAST LPAREN exprComponent COMMA (basicScalarType|valueDomainName) (COMMA STRING_CONSTANT)? RPAREN         # castExprComponent
+        """ # noqa E501
         ctx_list = list(ctx.getChildren())
         c = ctx_list[0]
 
@@ -579,24 +579,19 @@ class ExprComp(VtlVisitor):
         elif isinstance(ctx, Parser.CurrentDateAtomComponentContext):
             return self.visitCurrentDateAtomComponent(ctx)
         elif isinstance(ctx, Parser.DateDiffAtomComponentContext):
-            return self.visitDateAddAtomComponentContext(ctx)
+            return self.visitDateDiffAtomComponent(ctx)
         elif isinstance(ctx, Parser.DateAddAtomComponentContext):
             return self.visitDateAddAtomComponentContext(ctx)
-        elif isinstance(ctx, Parser.YearAtomComponentContext):
-            return self.visitTimeUnaryAtomComponent(ctx)
-        elif isinstance(ctx, Parser.MonthAtomComponentContext):
-            return self.visitTimeUnaryAtomComponent(ctx)
-        elif isinstance(ctx, Parser.DayOfMonthAtomComponentContext):
-            return self.visitTimeUnaryAtomComponent(ctx)
-        elif isinstance(ctx, Parser.DayOfYearAtomComponentContext):
-            return self.visitTimeUnaryAtomComponent(ctx)
-        elif isinstance(ctx, Parser.DayToYearAtomComponentContext):
-            return self.visitTimeUnaryAtomComponent(ctx)
-        elif isinstance(ctx, Parser.DayToMonthAtomComponentContext):
-            return self.visitTimeUnaryAtomComponent(ctx)
-        elif isinstance(ctx, Parser.YearToDayAtomComponentContext):
-            return self.visitTimeUnaryAtomComponent(ctx)
-        elif isinstance(ctx, Parser.MonthToDayAtomComponentContext):
+        elif (isinstance(ctx, (
+            Parser.YearAtomComponentContext,
+            Parser.MonthAtomComponentContext,
+            Parser.DayOfMonthAtomComponentContext,
+            Parser.DayOfYearAtomComponentContext,
+            Parser.DayToYearAtomComponentContext,
+            Parser.DayToMonthAtomComponentContext,
+            Parser.YearToDayAtomComponentContext,
+            Parser.MonthToDayAtomComponentContext
+        ))):
             return self.visitTimeUnaryAtomComponent(ctx)
         else:
             raise NotImplementedError
@@ -653,9 +648,9 @@ class ExprComp(VtlVisitor):
 
     def visitTimeAggAtomComponent(self, ctx: Parser.TimeAggAtomComponentContext):
         """
-        TIME_AGG LPAREN periodIndTo=STRING_CONSTANT (COMMA periodIndFrom=(STRING_CONSTANT| OPTIONAL ))? # noqa E501
-        (COMMA op=optionalExprComponent)? (COMMA (FIRST|LAST))? RPAREN    # timeAggAtomComponent; # noqa E501
-        """
+        TIME_AGG LPAREN periodIndTo=STRING_CONSTANT (COMMA periodIndFrom=(STRING_CONSTANT| OPTIONAL ))?
+        (COMMA op=optionalExprComponent)? (COMMA (FIRST|LAST))? RPAREN    # timeAggAtomComponent;
+        """ # noqa E501
         ctx_list = list(ctx.getChildren())
         c = ctx_list[0]
 
@@ -674,10 +669,7 @@ class ExprComp(VtlVisitor):
             and str_.getSymbol().type in [Parser.FIRST, Parser.LAST]
         ]
 
-        if len(conf) == 0:
-            conf = None
-        else:
-            conf = conf[0]
+        conf = None if len(conf) == 0 else conf[0]
 
         if ctx.op is not None:
             operand_node = self.visitOptionalExprComponent(ctx.op)
@@ -706,7 +698,7 @@ class ExprComp(VtlVisitor):
 
         op = c.getSymbol().text
         left_node = self.visitExprComponent(ctx_list[2])
-        right_node = Constant("INTEGER_CONSTANT", int(ctx_list[4].getSymbol().text))
+        right_node = self.visitExprComponent(ctx_list[4])
 
         return BinOp(left=left_node, op=op, right=right_node)
 
@@ -889,9 +881,7 @@ class ExprComp(VtlVisitor):
             elif isinstance(c, Parser.OrderByClauseContext):
                 order_by = Terminals().visitOrderByClause(c)
                 continue
-            elif isinstance(c, Parser.SignedIntegerContext) or isinstance(
-                c, Parser.ScalarItemContext
-            ):
+            elif isinstance(c, (Parser.SignedIntegerContext, Parser.ScalarItemContext)):
                 if params is None:
                     params = []
                 if isinstance(c, Parser.SignedIntegerContext):

vtlengine/AST/ASTConstructorModules/Terminals.py

@@ -1,13 +1,3 @@
-from vtlengine.DataTypes import (
-    Boolean,
-    Date,
-    Duration,
-    Integer,
-    Number,
-    String,
-    TimeInterval,
-    TimePeriod,
-)
 from antlr4.tree.Tree import TerminalNodeImpl
 
 from vtlengine.AST import (
@@ -24,6 +14,16 @@ from vtlengine.AST import (
 )
 from vtlengine.AST.Grammar.parser import Parser
 from vtlengine.AST.VtlVisitor import VtlVisitor
+from vtlengine.DataTypes import (
+    Boolean,
+    Date,
+    Duration,
+    Integer,
+    Number,
+    String,
+    TimeInterval,
+    TimePeriod,
+)
 from vtlengine.Model import Component, Dataset, Role, Scalar
 
 
@@ -377,21 +377,21 @@ class Terminals(VtlVisitor):
 
     def visitDpRuleset(self, ctx: Parser.DpRulesetContext):
         """
-        DATAPOINT                                                                               # dataPoint  # noqa E501
-            | DATAPOINT_ON_VD  (GLPAREN  valueDomainName (MUL valueDomainName)*  GRPAREN )?         # dataPointVd  # noqa E501
-            | DATAPOINT_ON_VAR  (GLPAREN  varID (MUL varID)*  GRPAREN )?                            # dataPointVar  # noqa E501
+        DATAPOINT                                                                               # dataPoint
+            | DATAPOINT_ON_VD  (GLPAREN  valueDomainName (MUL valueDomainName)*  GRPAREN )?         # dataPointVd
+            | DATAPOINT_ON_VAR  (GLPAREN  varID (MUL varID)*  GRPAREN )?                            # dataPointVar
         ;
-        """
+        """ # noqa E501
         # AST_ASTCONSTRUCTOR.54
         raise NotImplementedError
 
     def visitHrRuleset(self, ctx: Parser.HrRulesetContext):
         """
-        hrRuleset: HIERARCHICAL                                                                                                            # hrRulesetType  # noqa E501
-            | HIERARCHICAL_ON_VD ( GLPAREN  vdName=IDENTIFIER (LPAREN valueDomainName (MUL valueDomainName)* RPAREN)?  GRPAREN )?   # hrRulesetVdType  # noqa E501
-            | HIERARCHICAL_ON_VAR ( GLPAREN  varName=varID (LPAREN  varID (MUL varID)* RPAREN)?  GRPAREN )?                         # hrRulesetVarType  # noqa E501
+        hrRuleset: HIERARCHICAL                                                                                                            # hrRulesetType
+            | HIERARCHICAL_ON_VD ( GLPAREN  vdName=IDENTIFIER (LPAREN valueDomainName (MUL valueDomainName)* RPAREN)?  GRPAREN )?   # hrRulesetVdType
+            | HIERARCHICAL_ON_VAR ( GLPAREN  varName=varID (LPAREN  varID (MUL varID)* RPAREN)?  GRPAREN )?                         # hrRulesetVarType
         ;
-        """
+        """ # noqa E501
         # AST_ASTCONSTRUCTOR.55
         raise NotImplementedError
 
@@ -407,15 +407,9 @@ class Terminals(VtlVisitor):
             for constraint in ctx_list
             if isinstance(constraint, Parser.ScalarTypeContext)
         ]
-        if len(data_type) > 0:
-            data_type = data_type[0]
-        else:
-            data_type = String()
+        data_type = data_type[0] if len(data_type) > 0 else String()
 
-        if role_node == Role.IDENTIFIER:
-            nullable = False
-        else:
-            nullable = True
+        nullable = role_node != Role.IDENTIFIER
 
         return Component(name="Component", data_type=data_type, role=role_node, nullable=nullable)
 
@@ -503,7 +497,7 @@ class Terminals(VtlVisitor):
     def visitScalarWithCast(self, ctx: Parser.ScalarWithCastContext):
         """
         |  CAST LPAREN constant COMMA (basicScalarType) (COMMA STRING_CONSTANT)? RPAREN    #scalarWithCast  # noqa E501
-        """
+        """ # noqa E501
         ctx_list = list(ctx.getChildren())
         c = ctx_list[0]
 
@@ -513,10 +507,7 @@ class Terminals(VtlVisitor):
         const_node = self.visitConstant(ctx_list[2])
         basic_scalar_type = [self.visitBasicScalarType(ctx_list[4])]
 
-        if len(ctx_list) > 6:
-            param_node = [ParamConstant("PARAM_CAST", ctx_list[6])]
-        else:
-            param_node = []
+        param_node = [ParamConstant("PARAM_CAST", ctx_list[6])] if len(ctx_list) > 6 else []
 
         if len(basic_scalar_type) == 1:
             children_nodes = [const_node, basic_scalar_type[0]]
@@ -681,14 +672,12 @@ class Terminals(VtlVisitor):
         first = num_rows_1  # unbounded (default value)
         second = num_rows_2  # current data point (default value)
 
-        if mode_2 == "preceding":
-            if (
-                mode_1 == "preceding" and num_rows_1 == -1 and num_rows_2 == -1
-            ):  # preceding and preceding (error)
-                raise Exception(
-                    f"Cannot have 2 preceding clauses with unbounded in analytic clause, "
-                    f"line {ctx_list[3].start.line}"
-                )
+        if (mode_2 == "preceding" and mode_1 == "preceding" and num_rows_1 == -1
+                and num_rows_2 == -1):  # preceding and preceding (error)
+            raise Exception(
+                f"Cannot have 2 preceding clauses with unbounded in analytic clause, "
+                f"line {ctx_list[3].start.line}"
+            )
 
         if (
             mode_1 == "following" and num_rows_1 == -1 and num_rows_2 == -1

vtlengine/AST/ASTTemplate.py

@@ -10,7 +10,6 @@ Template to start a new visitor for the AST.
 from typing import Any
 
 import vtlengine.AST as AST
-
 from vtlengine.AST.ASTVisitor import NodeVisitor
 
 

vtlengine/AST/DAG/__init__.py

@@ -15,24 +15,24 @@ import networkx as nx
 
 from vtlengine.AST import (
     AST,
-    BinOp,
-    VarID,
     Aggregation,
     Analytic,
-    JoinOp,
-    ParamOp,
-    Operator,
-    Identifier,
+    Assignment,
+    BinOp,
     DefIdentifier,
-    Start,
+    DPRuleset,
     HRuleset,
-    RegularAggregation,
+    Identifier,
+    JoinOp,
+    Operator,
+    ParamOp,
     PersistentAssignment,
-    Assignment,
-    DPRuleset,
+    RegularAggregation,
+    Start,
+    VarID,
 )
 from vtlengine.AST.ASTTemplate import ASTTemplate
-from vtlengine.AST.DAG._words import INSERT, DELETE, OUTPUTS, PERSISTENT, INPUTS, GLOBAL
+from vtlengine.AST.DAG._words import DELETE, GLOBAL, INPUTS, INSERT, OUTPUTS, PERSISTENT
 from vtlengine.AST.Grammar.tokens import AS, MEMBERSHIP, TO
 from vtlengine.Exceptions import SemanticError
 
@@ -313,10 +313,7 @@ class DAGAnalyzer(ASTTemplate):
             self.visit(node.left)
             self.isDataset = False
             self.visit(node.right)
-        elif node.op == AS:
-            self.visit(node.left)
-            self.alias.append(node.right.value)
-        elif node.op == TO:
+        elif node.op == AS or node.op == TO:
             self.visit(node.left)
             self.alias.append(node.right.value)
         else:

vtlengine/AST/Grammar/tokens.py

@@ -91,8 +91,8 @@ STOCK_TO_FLOW = "stock_to_flow"
 TIMESHIFT = "timeshift"
 TIME_AGG = "time_agg"
 CURRENT_DATE = "current_date"
-DATE_DIFF = "date_diff"
-DATE_ADD = "date_add"
+DATEDIFF = "datediff"
+DATE_ADD = "dateadd"
 YEAR = "year"
 MONTH = "month"
 DAYOFMONTH = "dayofmonth"

vtlengine/AST/VtlVisitor.py

@@ -2,7 +2,6 @@ from antlr4 import ParseTreeVisitor
 
 from vtlengine.AST.Grammar.parser import Parser
 
-
 # This class defines a complete generic visitor for a parse tree produced by Parser.
 
 

vtlengine/AST/__init__.py

@@ -11,7 +11,6 @@ from dataclasses import dataclass
 from typing import Any, Dict, List, Optional, Type, Union
 
 from vtlengine.DataTypes import ScalarType
-
 from vtlengine.Model import Role
 
 
@@ -33,7 +32,7 @@ class AST:
         """Returns a human-friendly description."""
         out = []
         name = self.__class__.__name__
-        for k in self.__all_annotations().keys():
+        for k in self.__all_annotations():
             v = self.__getattribute__(k)
             if v:
                 out.append(f"{k}={str(v)}")
@@ -41,7 +40,7 @@ class AST:
 
     def toJSON(self):
         base = {"class_name": self.__class__.__name__}
-        for k in self.__all_annotations().keys():
+        for k in self.__all_annotations():
             v = self.__getattribute__(k)
             base[k] = v
         return base

vtlengine/DataTypes/TimeHandling.py

@@ -1,8 +1,9 @@
 import calendar
 import copy
 import operator
-from datetime import date, datetime as dt
-from typing import Union, Optional, Any, Dict
+from datetime import date
+from datetime import datetime as dt
+from typing import Any, Dict, Optional, Union
 
 import pandas as pd
 
@@ -31,9 +32,11 @@ def date_to_period(date_value: date, period_indicator: str) -> Any:
         return TimePeriodHandler(f"{date_value.year}D{date_value.timetuple().tm_yday}")
 
 
-def period_to_date(
-    year: int, period_indicator: str, period_number: int, start: bool = False
-) -> date:
+def period_to_date(year: int,
+                   period_indicator: str,
+                   period_number: int,
+                   start: bool = False
+                   ) -> date:
     if period_indicator == "A":
         return date(year, 1, 1) if start else date(year, 12, 31)
     periods = {
@@ -209,7 +212,7 @@ class TimePeriodHandler:
             raise SemanticError(
                 "2-1-19-7",
                 periods=PeriodDuration.periods[self.period_indicator],
-                period_inidcator=self.period_indicator,
+                period_indicator=self.period_indicator,
             )
             # raise ValueError(f'Period Number must be between 1 and '
             #                  f'{PeriodDuration.periods[self.period_indicator]} '
@@ -483,7 +486,7 @@ def check_max_date(str_: Optional[str]) -> Optional[str]:
 
     # Format 2010-01-01. Prevent passthrough of other ISO 8601 formats.
     if len(str_) != 10 or str_[7] != "-":
-        raise SemanticError("2-1-19-8", date=str)
+        raise SemanticError("2-1-19-8", date=str_)
         # raise ValueError(f"Invalid date format, must be YYYY-MM-DD: {str_}")
 
     result = date.fromisoformat(str_)

vtlengine/DataTypes/__init__.py

@@ -1,7 +1,8 @@
-from typing import Any, Optional, Type, Dict, Set, Union
+from typing import Any, Dict, Optional, Set, Type, Union
+
 import pandas as pd
 
-from vtlengine.DataTypes.TimeHandling import str_period_to_date, check_max_date, date_to_period_str
+from vtlengine.DataTypes.TimeHandling import check_max_date, date_to_period_str, str_period_to_date
 from vtlengine.Exceptions import SemanticError
 
 DTYPE_MAPPING: Dict[str, str] = {
@@ -310,7 +311,8 @@ class TimeInterval(ScalarType):
 
     @classmethod
     def explicit_cast(cls, value: Any, from_type: Any) -> Any:
-
+        if from_type == String:
+            return value  # check_time(value). TODO: resolve this to avoid a circular import.
         raise SemanticError(
             "2-1-5-1",
             value=value,
@@ -328,7 +330,7 @@ class Date(TimeInterval):
     def implicit_cast(cls, value: Any, from_type: Any) -> Any:
         # TODO: Remove String, only for compatibility with previous engine
         if from_type in {Date, String}:
-            return value
+            return check_max_date(value)
 
         raise SemanticError(
             "2-1-5-1",
@@ -341,7 +343,7 @@ class Date(TimeInterval):
     def explicit_cast(cls, value: Any, from_type: Any) -> Any:
         # TODO: Remove String, only for compatibility with previous engine
         if from_type == String:
-            return value
+            return check_max_date(value)
 
         raise SemanticError(
             "2-1-5-1",
@@ -384,7 +386,7 @@ class TimePeriod(TimeInterval):
             return period_str
         # TODO: Remove String, only for compatibility with previous engine
         elif from_type == String:
-            return value
+            return value  # check_time_period(value) TODO: resolve this to avoid a circular import.
 
         raise SemanticError(
             "2-1-5-1",
@@ -442,15 +444,9 @@ class Boolean(ScalarType):
             else:
                 return None
         if isinstance(value, int):
-            if value != 0:
-                return True
-            else:
-                return False
+            return value != 0
         if isinstance(value, float):
-            if value != 0.0:
-                return True
-            else:
-                return False
+            return value != 0.0
         if isinstance(value, bool):
             return value
         return value
@@ -470,9 +466,7 @@ class Boolean(ScalarType):
     @classmethod
     def explicit_cast(cls, value: Any, from_type: Any) -> bool:
         if from_type in {Number, Integer}:
-            if value in {0, 0.0}:
-                return False
-            return True
+            return value not in {0}
 
         raise SemanticError(
             "2-1-5-1",
@@ -672,13 +666,12 @@ def unary_implicit_promotion(
     return: The resulting type of the operation, after the implicit type promotion
     """
     operand_implicities = IMPLICIT_TYPE_PROMOTION_MAPPING[operand_type]
-    if type_to_check:
-        if not type_to_check.is_included(operand_implicities):
-            raise SemanticError(
-                code="1-1-1-1",
-                type_1=SCALAR_TYPES_CLASS_REVERSE[operand_type],
-                type_2=SCALAR_TYPES_CLASS_REVERSE[type_to_check],
-            )
+    if type_to_check and not type_to_check.is_included(operand_implicities):
+        raise SemanticError(
+            code="1-1-1-1",
+            type_1=SCALAR_TYPES_CLASS_REVERSE[operand_type],
+            type_2=SCALAR_TYPES_CLASS_REVERSE[type_to_check],
+        )
     if return_type:
         return return_type
     if (

vtlengine/Exceptions/__init__.py

@@ -7,7 +7,8 @@ Description
 All exceptions exposed by the Vtl engine.
 """
 
-from typing import Optional, Any, List
+from typing import Any, List, Optional
+
 from vtlengine.Exceptions.messages import centralised_messages
 
 dataset_output = None
@@ -165,10 +166,7 @@ def key_distance(key: str, objetive: str) -> int:
 
     for i in range(1, len(key) + 1):
         for j in range(1, len(objetive) + 1):
-            if key[i - 1] == objetive[j - 1]:
-                cost = 0
-            else:
-                cost = 1
+            cost = 0 if key[i - 1] == objetive[j - 1] else 1
             dp[i][j] = min(dp[i - 1][j] + 1,
                            dp[i][j - 1] + 1,
                            dp[i - 1][j - 1] + cost)