| 69 | |
| 70 | |
| 71 | def test_label(): |
| 72 | assert label("x") == "x" |
| 73 | assert label("elemwise-ffcd9aa2231d466b5aa91e8bfa9e9487") == "elemwise-#" |
| 74 | |
| 75 | cache = {} |
| 76 | result = label("elemwise-ffcd9aa2231d466b5aa91e8bfa9e9487", cache=cache) |
| 77 | assert result == "elemwise-#0" |
| 78 | # cached |
| 79 | result = label("elemwise-ffcd9aa2231d466b5aa91e8bfa9e9487", cache=cache) |
| 80 | assert result == "elemwise-#0" |
| 81 | assert len(cache) == 1 |
| 82 | |
| 83 | result = label("elemwise-e890b510984f344edea9a5e5fe05c0db", cache=cache) |
| 84 | assert result == "elemwise-#1" |
| 85 | assert len(cache) == 2 |
| 86 | |
| 87 | result = label("elemwise-ffcd9aa2231d466b5aa91e8bfa9e9487", cache=cache) |
| 88 | assert result == "elemwise-#0" |
| 89 | assert len(cache) == 2 |
| 90 | |
| 91 | assert label("x", cache=cache) == "x" |
| 92 | assert len(cache) == 2 |
| 93 | |
| 94 | |
| 95 | def test_to_graphviz(): |