()
| 4075 | |
| 4076 | |
| 4077 | def test_uneven_chunks_blockwise(): |
| 4078 | rng = da.random.default_rng() |
| 4079 | x = rng.random((10, 10), chunks=((2, 3, 2, 3), (5, 5))) |
| 4080 | y = rng.random((10, 10), chunks=((4, 4, 2), (4, 2, 4))) |
| 4081 | z = da.blockwise(np.dot, "ik", x, "ij", y, "jk", dtype=x.dtype, concatenate=True) |
| 4082 | assert z.chunks == (x.chunks[0], y.chunks[1]) |
| 4083 | |
| 4084 | assert_eq(z, x.compute().dot(y)) |
| 4085 | |
| 4086 | |
| 4087 | def test_warn_bad_rechunking(): |