()
| 1551 | |
| 1552 | |
| 1553 | def test_map_blocks(): |
| 1554 | x = np.arange(400).reshape((20, 20)) |
| 1555 | d = from_array(x, chunks=(7, 7)) |
| 1556 | |
| 1557 | e = d.map_blocks(inc, dtype=d.dtype) |
| 1558 | |
| 1559 | assert d.chunks == e.chunks |
| 1560 | assert_eq(e, x + 1) |
| 1561 | |
| 1562 | e = d.map_blocks(inc, token="increment") |
| 1563 | assert e.name.startswith("increment-") |
| 1564 | |
| 1565 | assert d.map_blocks(inc, token="foo").name != d.map_blocks(dec, token="foo").name |
| 1566 | |
| 1567 | d = from_array(x, chunks=(10, 10)) |
| 1568 | e = d.map_blocks(lambda x: x[::2, ::2], chunks=(5, 5), dtype=d.dtype) |
| 1569 | |
| 1570 | assert e.chunks == ((5, 5), (5, 5)) |
| 1571 | assert_eq(e, x[::2, ::2]) |
| 1572 | |
| 1573 | d = from_array(x, chunks=(8, 8)) |
| 1574 | e = d.map_blocks( |
| 1575 | lambda x: x[::2, ::2], chunks=((4, 4, 2), (4, 4, 2)), dtype=d.dtype |
| 1576 | ) |
| 1577 | |
| 1578 | assert_eq(e, x[::2, ::2]) |
| 1579 | |
| 1580 | |
| 1581 | def test_map_blocks2(): |
nothing calls this directly
no test coverage detected