()
| 1579 | |
| 1580 | |
| 1581 | def test_map_blocks2(): |
| 1582 | x = np.arange(10, dtype="i8") |
| 1583 | d = from_array(x, chunks=(2,)) |
| 1584 | |
| 1585 | def func(block, block_id=None, c=0): |
| 1586 | return np.ones_like(block) * sum(block_id) + c |
| 1587 | |
| 1588 | out = d.map_blocks(func, dtype="i8") |
| 1589 | expected = np.array([0, 0, 1, 1, 2, 2, 3, 3, 4, 4], dtype="i8") |
| 1590 | |
| 1591 | assert_eq(out, expected) |
| 1592 | assert same_keys(d.map_blocks(func, dtype="i8"), out) |
| 1593 | |
| 1594 | out = d.map_blocks(func, dtype="i8", c=1) |
| 1595 | expected = expected + 1 |
| 1596 | |
| 1597 | assert_eq(out, expected) |
| 1598 | assert same_keys(d.map_blocks(func, dtype="i8", c=1), out) |
| 1599 | |
| 1600 | |
| 1601 | def test_map_blocks_block_info(): |
nothing calls this directly
no test coverage detected