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