()
| 178 | |
| 179 | |
| 180 | def test_slice_array_1d(): |
| 181 | # x[24::2] |
| 182 | expected = { |
| 183 | ("y", 0): Task(("y", 0), getitem, TaskRef(("x", 0)), (slice(24, 25, 2),)), |
| 184 | ("y", 1): Task(("y", 1), getitem, TaskRef(("x", 1)), (slice(1, 25, 2),)), |
| 185 | ("y", 2): Task(("y", 2), getitem, TaskRef(("x", 2)), (slice(0, 25, 2),)), |
| 186 | ("y", 3): Task(("y", 3), getitem, TaskRef(("x", 3)), (slice(1, 25, 2),)), |
| 187 | } |
| 188 | result, chunks = slice_array("y", "x", [[25] * 4], [slice(24, None, 2)]) |
| 189 | |
| 190 | assert expected == result |
| 191 | |
| 192 | # x[26::2] |
| 193 | expected = { |
| 194 | ("y", 0): Task(("y", 0), getitem, TaskRef(("x", 1)), (slice(1, 25, 2),)), |
| 195 | ("y", 1): Task(("y", 1), getitem, TaskRef(("x", 2)), (slice(0, 25, 2),)), |
| 196 | ("y", 2): Task(("y", 2), getitem, TaskRef(("x", 3)), (slice(1, 25, 2),)), |
| 197 | } |
| 198 | |
| 199 | result, chunks = slice_array("y", "x", [[25] * 4], [slice(26, None, 2)]) |
| 200 | assert expected == result |
| 201 | |
| 202 | # x[24::2] |
| 203 | expected = { |
| 204 | ("y", 0): Task(("y", 0), getitem, TaskRef(("x", 0)), (slice(24, 25, 2),)), |
| 205 | ("y", 1): Task(("y", 1), getitem, TaskRef(("x", 1)), (slice(1, 25, 2),)), |
| 206 | ("y", 2): Task(("y", 2), getitem, TaskRef(("x", 2)), (slice(0, 25, 2),)), |
| 207 | ("y", 3): Task(("y", 3), getitem, TaskRef(("x", 3)), (slice(1, 25, 2),)), |
| 208 | } |
| 209 | result, chunks = slice_array("y", "x", [(25,) * 4], (slice(24, None, 2),)) |
| 210 | |
| 211 | assert expected == result |
| 212 | |
| 213 | # x[26::2] |
| 214 | expected = { |
| 215 | ("y", 0): Task(("y", 0), getitem, TaskRef(("x", 1)), (slice(1, 25, 2),)), |
| 216 | ("y", 1): Task(("y", 1), getitem, TaskRef(("x", 2)), (slice(0, 25, 2),)), |
| 217 | ("y", 2): Task(("y", 2), getitem, TaskRef(("x", 3)), (slice(1, 25, 2),)), |
| 218 | } |
| 219 | |
| 220 | result, chunks = slice_array("y", "x", [(25,) * 4], (slice(26, None, 2),)) |
| 221 | assert expected == result |
| 222 | |
| 223 | |
| 224 | def test_slice_array_2d(): |
nothing calls this directly
no test coverage detected