fix(tests): serialize ternary expressions
This commit is contained in:
@@ -22,6 +22,7 @@ from midas.ast.python import (
|
|||||||
ReturnStmt,
|
ReturnStmt,
|
||||||
SetExpr,
|
SetExpr,
|
||||||
Stmt,
|
Stmt,
|
||||||
|
TernaryExpr,
|
||||||
TypeAssign,
|
TypeAssign,
|
||||||
UnaryExpr,
|
UnaryExpr,
|
||||||
VariableExpr,
|
VariableExpr,
|
||||||
@@ -245,3 +246,11 @@ class PythonAstJsonSerializer(
|
|||||||
"type": expr.type.accept(self),
|
"type": expr.type.accept(self),
|
||||||
"expr": expr.expr.accept(self),
|
"expr": expr.expr.accept(self),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
def visit_ternary_expr(self, expr: TernaryExpr) -> dict:
|
||||||
|
return {
|
||||||
|
"_type": "TernaryExpr",
|
||||||
|
"test": expr.test.accept(self),
|
||||||
|
"if_true": expr.if_true.accept(self),
|
||||||
|
"if_false": expr.if_false.accept(self),
|
||||||
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user