Check visitor and mutator behavior on the given statement.
(stmt, visitor_str, mutator_str)
| 600 | |
| 601 | |
| 602 | def basic_check(stmt, visitor_str, mutator_str): |
| 603 | """Check visitor and mutator behavior on the given statement.""" |
| 604 | # Check basic visitor |
| 605 | basic_visitor = BasicStmtVisitor() |
| 606 | basic_visitor.visit_stmt(stmt) |
| 607 | |
| 608 | # Check AST printer visitor |
| 609 | log_visitor = ASTPrinter() |
| 610 | log_visitor.visit_stmt(stmt) |
| 611 | assert str(log_visitor.log) == visitor_str |
| 612 | |
| 613 | # Check AST printer mutator |
| 614 | log_mutator = ASTPrinterMutator() |
| 615 | result = log_mutator.visit_stmt(stmt) |
| 616 | # Check we get back structurally equivalent statement |
| 617 | tvm.ir.assert_structural_equal(result, stmt) |
| 618 | assert str(log_mutator.log) == mutator_str |
| 619 | |
| 620 | |
| 621 | def create_test_statements(): |
no test coverage detected
searching dependent graphs…