Test StmtExprMutator.
()
| 982 | |
| 983 | |
| 984 | def test_stmt_expr_mutator(): |
| 985 | """Test StmtExprMutator.""" |
| 986 | evaluate_stmt = create_test_statements()["evaluate"] |
| 987 | mutator = StmtExprMutatorPrinter() |
| 988 | result = mutator.visit_stmt(evaluate_stmt) |
| 989 | tvm.ir.assert_structural_equal(result, evaluate_stmt) |
| 990 | |
| 991 | expected = "\n".join(["Var", "IntImm", "Add", "Evaluate"]) |
| 992 | assert str(mutator.log) == expected |
| 993 | |
| 994 | |
| 995 | def test_stmt_expr_visitor(): |
nothing calls this directly
no test coverage detected
searching dependent graphs…