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