| 111 | |
| 112 | |
| 113 | def test_metadata(): |
| 114 | y = da.random.random((10, 10), chunks=(5, 5)) |
| 115 | y[y < 0.8] = 0 |
| 116 | z = sparse.COO.from_numpy(y.compute()) |
| 117 | y = y.map_blocks(sparse.COO.from_numpy) |
| 118 | |
| 119 | assert isinstance(y._meta, sparse.COO) |
| 120 | assert isinstance((y + 1)._meta, sparse.COO) |
| 121 | assert isinstance(y.sum(axis=0)._meta, sparse.COO) |
| 122 | assert isinstance(y.var(axis=0)._meta, sparse.COO) |
| 123 | assert isinstance(y[:5, ::2]._meta, sparse.COO) |
| 124 | assert isinstance(y.rechunk((2, 2))._meta, sparse.COO) |
| 125 | assert isinstance((y - z)._meta, sparse.COO) |
| 126 | assert isinstance(y.persist()._meta, sparse.COO) |
| 127 | assert isinstance(np.concatenate([y, y])._meta, sparse.COO) |
| 128 | assert isinstance(np.concatenate([y, y[:0], y])._meta, sparse.COO) |
| 129 | assert isinstance(np.stack([y, y])._meta, sparse.COO) |
| 130 | assert isinstance(np.stack([y[:0], y[:0]])._meta, sparse.COO) |
| 131 | assert isinstance(np.concatenate([y[:0], y[:0]])._meta, sparse.COO) |
| 132 | |
| 133 | |
| 134 | def test_html_repr(): |