()
| 75 | |
| 76 | |
| 77 | def test_var() -> None: |
| 78 | v0 = rx.Var("v0") |
| 79 | v0_str = dump_ast(v0) |
| 80 | assert v0_str == 'Var(name_hint="v0")' |
| 81 | |
| 82 | v1 = rx.Var("v1", R.Tensor([54, 96], "float32")) |
| 83 | v1_no_annos = dump_ast(v1, include_struct_info_annotations=False) |
| 84 | assert v1_no_annos == 'Var(name_hint="v1")' |
| 85 | v1_annos = dump_ast(v1) |
| 86 | assert v1_annos != v1_no_annos |
| 87 | assert "PrimExpr" in v1_annos |
| 88 | assert "struct_info" in v1_annos |
| 89 | |
| 90 | |
| 91 | def test_dataflow_var() -> None: |