Skip to content

Commit

Permalink
Add test
Browse files Browse the repository at this point in the history
  • Loading branch information
dcherian committed Mar 28, 2024
1 parent eb37aed commit e0a3e10
Showing 1 changed file with 18 additions and 1 deletion.
19 changes: 18 additions & 1 deletion xarray/tests/test_backends.py
Original file line number Diff line number Diff line change
Expand Up @@ -2261,7 +2261,6 @@ def test_write_uneven_dask_chunks(self) -> None:
original = create_test_data().chunk({"dim1": 3, "dim2": 4, "dim3": 3})
with self.roundtrip(original, open_kwargs={"chunks": {}}) as actual:
for k, v in actual.data_vars.items():
print(k)
assert v.chunks == actual[k].chunks

def test_chunk_encoding(self) -> None:
Expand Down Expand Up @@ -2468,6 +2467,24 @@ def test_group(self) -> None:
) as actual:
assert_identical(original, actual)

@requires_zarr
def test_zarr_mode_w_overwrites_encoding(self) -> None:
import zarr

data = Dataset({"foo": ("x", [1.0, 1.0, 1.0])})
with self.create_zarr_target() as store:
data.to_zarr(
store, **self.version_kwargs, encoding={"foo": {"add_offset": 1}}
)
np.testing.assert_equal(zarr.open_group(store)["foo"], data.foo.data - 1)
data.to_zarr(
store,
**self.version_kwargs,
encoding={"foo": {"add_offset": 0}},
mode="w",
)
np.testing.assert_equal(zarr.open_group(store)["foo"], data.foo.data)

def test_encoding_kwarg_fixed_width_string(self) -> None:
# not relevant for zarr, since we don't use EncodedStringCoder
pass
Expand Down

0 comments on commit e0a3e10

Please sign in to comment.