()
| 4082 | |
| 4083 | |
| 4084 | def test_uneven_chunks_blockwise(): |
| 4085 | rng = da.random.default_rng() |
| 4086 | x = rng.random((10, 10), chunks=((2, 3, 2, 3), (5, 5))) |
| 4087 | y = rng.random((10, 10), chunks=((4, 4, 2), (4, 2, 4))) |
| 4088 | z = da.blockwise(np.dot, "ik", x, "ij", y, "jk", dtype=x.dtype, concatenate=True) |
| 4089 | assert z.chunks == (x.chunks[0], y.chunks[1]) |
| 4090 | |
| 4091 | assert_eq(z, x.compute().dot(y)) |
| 4092 | |
| 4093 | |
| 4094 | def test_warn_bad_rechunking(): |