()
| 129 | |
| 130 | |
| 131 | def test_top(): |
| 132 | t = Task("inc", inc, TaskRef("A")) |
| 133 | assert _make_blockwise_graph( |
| 134 | t, "z", "ij", "x", "ij", numblocks={"x": (2, 2)}, keys=("A",) |
| 135 | ) == { |
| 136 | ("z", 0, 0): Task(("z", 0, 0), inc, TaskRef(("x", 0, 0))), |
| 137 | ("z", 0, 1): Task(("z", 0, 1), inc, TaskRef(("x", 0, 1))), |
| 138 | ("z", 1, 0): Task(("z", 1, 0), inc, TaskRef(("x", 1, 0))), |
| 139 | ("z", 1, 1): Task(("z", 1, 1), inc, TaskRef(("x", 1, 1))), |
| 140 | } |
| 141 | |
| 142 | keys = list(map(str, range(2))) |
| 143 | t = Task("add", add, *(TaskRef(k) for k in keys)) |
| 144 | assert _make_blockwise_graph( |
| 145 | t, |
| 146 | "z", |
| 147 | "ij", |
| 148 | "x", |
| 149 | "ij", |
| 150 | "y", |
| 151 | "ij", |
| 152 | numblocks={"x": (2, 2), "y": (2, 2)}, |
| 153 | keys=keys, |
| 154 | ) == { |
| 155 | ("z", 0, 0): Task(("z", 0, 0), add, TaskRef(("x", 0, 0)), TaskRef(("y", 0, 0))), |
| 156 | ("z", 0, 1): Task(("z", 0, 1), add, TaskRef(("x", 0, 1)), TaskRef(("y", 0, 1))), |
| 157 | ("z", 1, 0): Task(("z", 1, 0), add, TaskRef(("x", 1, 0)), TaskRef(("y", 1, 0))), |
| 158 | ("z", 1, 1): Task(("z", 1, 1), add, TaskRef(("x", 1, 1)), TaskRef(("y", 1, 1))), |
| 159 | } |
| 160 | keys = list(map(str, range(2))) |
| 161 | t = Task( |
| 162 | "dotmany", |
| 163 | dotmany, |
| 164 | TaskRef(keys[0]), |
| 165 | TaskRef(keys[1]), |
| 166 | ) |
| 167 | out = _make_blockwise_graph( |
| 168 | t, |
| 169 | "z", |
| 170 | "ik", |
| 171 | "x", |
| 172 | "ij", |
| 173 | "y", |
| 174 | "jk", |
| 175 | numblocks={"x": (2, 2), "y": (2, 2)}, |
| 176 | keys=keys, |
| 177 | ) |
| 178 | expected = { |
| 179 | ("z", 0, 0): Task( |
| 180 | ("z", 0, 0), |
| 181 | dotmany, |
| 182 | List(TaskRef(("x", 0, 0)), TaskRef(("x", 0, 1))), |
| 183 | List(TaskRef(("y", 0, 0)), TaskRef(("y", 1, 0))), |
| 184 | ), |
| 185 | ("z", 0, 1): Task( |
| 186 | ("z", 0, 1), |
| 187 | dotmany, |
| 188 | List(TaskRef(("x", 0, 0)), TaskRef(("x", 0, 1))), |
nothing calls this directly
no test coverage detected