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