Skip to content
Snippets Groups Projects
Commit f85f5908 authored by Emmanuel Raviart's avatar Emmanuel Raviart
Browse files

Accept unary subtract in decompositions

parent 50f8c7d3
Branches
Tags
No related merge requests found
Pipeline #18664 failed
......@@ -221,6 +221,27 @@ class FormulaExtractor extends AstVisitor<PythonAstNode> {
}
break switchNodeAstClass
}
case "UnaryOp": {
switch (node.op.ast_class) {
case "USub": {
const decomposition = this.extractDecomposition(node.operand)
if (decomposition !== null) {
return decomposition.map((decompositionReference) => {
decompositionReference = { ...decompositionReference }
if (decompositionReference.negate) {
delete decompositionReference.negate
} else {
decompositionReference.negate = true
}
return decompositionReference
})
}
break switchNodeAstClass
}
default:
break switchNodeAstClass
}
}
}
const openFiscaVariablesName = this.extractOpenFiscaVariablesName(node)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment