Skip to content

Commit 1a22c49

Browse files
committed
initial test of automatically using cubed
1 parent 669970c commit 1a22c49

File tree

1 file changed

+22
-0
lines changed

1 file changed

+22
-0
lines changed

cubed_xarray/tests/test_wrapping.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
1+
import xarray as xr
2+
from xarray.core.parallelcompat import list_chunkmanagers
3+
import cubed
4+
5+
from cubed_xarray.cubedmanager import CubedManager
6+
7+
8+
class TestDiscoverCubedManager:
9+
def test_list_cubedmanager(self):
10+
chunkmanagers = list_chunkmanagers()
11+
assert 'cubed' in chunkmanagers
12+
assert isinstance(chunkmanagers['cubed'], CubedManager)
13+
14+
def test_chunk(self):
15+
da = xr.DataArray([1, 2], dims='x')
16+
chunked = da.chunk(x=1, chunked_array_type='cubed')
17+
assert isinstance(chunked.data, cubed.Array)
18+
assert chunked.chunksizes == {'x': (1, 1)}
19+
20+
# TODO test cubed is default when dask not installed
21+
22+
# TODO test dask is default over cubed when both installed

0 commit comments

Comments
 (0)