()
| 421 | |
| 422 | |
| 423 | def test_keys(): |
| 424 | dsk = {("x", i, j): () for i in range(5) for j in range(6)} |
| 425 | dx = Array(dsk, "x", chunks=(10, 10), shape=(50, 60), dtype="f8") |
| 426 | assert dx.__dask_keys__() == [[(dx.name, i, j) for j in range(6)] for i in range(5)] |
| 427 | # Cache works |
| 428 | assert dx.__dask_keys__() is dx.__dask_keys__() |
| 429 | # Test mutating names clears key cache |
| 430 | dx.dask = {("y", i, j): () for i in range(5) for j in range(6)} |
| 431 | dx._name = "y" |
| 432 | new_keys = [[(dx.name, i, j) for j in range(6)] for i in range(5)] |
| 433 | assert dx.__dask_keys__() == new_keys |
| 434 | assert np.array_equal(dx._key_array, np.array(new_keys, dtype="object")) |
| 435 | d = Array({}, "x", (), shape=(), dtype="f8") |
| 436 | assert d.__dask_keys__() == [("x",)] |
| 437 | |
| 438 | |
| 439 | def test_Array_computation(): |
nothing calls this directly
no test coverage detected