()
| 997 | |
| 998 | |
| 999 | def test_pad_constant_chunksizes(): |
| 1000 | array = dask.array.ones((10, 10), chunks=(1, 1)) |
| 1001 | result = dask.array.pad( |
| 1002 | array, ((0, 16 - 10), (0, 0)), mode="constant", constant_values=0 |
| 1003 | ) |
| 1004 | assert tuple(map(max, result.chunks)) == (1, 1) |
| 1005 | assert_eq( |
| 1006 | result, |
| 1007 | np.pad( |
| 1008 | array.compute(), |
| 1009 | mode="constant", |
| 1010 | constant_values=0, |
| 1011 | pad_width=((0, 16 - 10), (0, 0)), |
| 1012 | ), |
| 1013 | ) |
| 1014 | |
| 1015 | |
| 1016 | def test_auto_chunks(): |