def check_visit(expr):
ef = ExprFunctor()
try:
ef.visit(expr)
assert False
except NotImplementedError:
pass
em = ExprMutator()
assert em.visit(expr)
def check_visit(expr):
try:
ef = ExprFunctor()
ef.visit(expr)
assert False
except NotImplementedError:
pass
ev = ExprVisitor()
ev.visit(expr)
em = ExprMutator()
assert em.visit(expr)
def test_match_completeness():
p = relay.prelude.Prelude()
for completeness in [True, False]:
match_expr = relay.adt.Match(p.nil, [], complete=completeness)
result_expr = ExprMutator().visit(match_expr)
# ensure the mutator doesn't mangle the completeness flag
assert result_expr.complete == completeness