Test basic statement mutator functionality
()
| 386 | |
| 387 | |
| 388 | def test_stmt_mutator(): |
| 389 | """Test basic statement mutator functionality""" |
| 390 | x = Var("x", dtype="int32") |
| 391 | stmt = Evaluate(Add(x, IntImm("int32", 1))) |
| 392 | |
| 393 | unroller = ForLoopUnroller() |
| 394 | result = unroller.visit_stmt(stmt) |
| 395 | |
| 396 | # Should return the same statement (no actual unrolling implemented) |
| 397 | assert isinstance(result, Evaluate) |
| 398 | |
| 399 | |
| 400 | def test_nested_expressions(): |
nothing calls this directly
no test coverage detected
searching dependent graphs…