diff --git a/zarr/tests/test_storage.py b/zarr/tests/test_storage.py index 154ffbe3ed..abfcac20b7 100644 --- a/zarr/tests/test_storage.py +++ b/zarr/tests/test_storage.py @@ -14,6 +14,7 @@ from numpy.testing import assert_array_equal, assert_array_almost_equal import pytest +from numcodecs.compat import ensure_bytes from zarr.storage import (init_array, array_meta_key, attrs_key, DictStore, DirectoryStore, ZipStore, init_group, group_meta_key, @@ -47,7 +48,7 @@ def test_get_set_del_contains(self): store['foo'] store['foo'] = b'bar' assert 'foo' in store - assert b'bar' == store['foo'] + assert b'bar' == ensure_bytes(store['foo']) # test __delitem__ (optional) try: @@ -85,10 +86,10 @@ def test_pop(self): store['baz'] = b'qux' assert len(store) == 2 v = store.pop('foo') - assert v == b'bar' + assert ensure_bytes(v) == b'bar' assert len(store) == 1 v = store.pop('baz') - assert v == b'qux' + assert ensure_bytes(v) == b'qux' assert len(store) == 0 with pytest.raises(KeyError): store.pop('xxx') @@ -104,7 +105,7 @@ def test_popitem(self): store['foo'] = b'bar' k, v = store.popitem() assert k == 'foo' - assert v == b'bar' + assert ensure_bytes(v) == b'bar' assert len(store) == 0 with pytest.raises(KeyError): store.popitem() @@ -123,8 +124,8 @@ def test_update(self): assert 'foo' not in store assert 'baz' not in store store.update(foo=b'bar', baz=b'quux') - assert b'bar' == store['foo'] - assert b'quux' == store['baz'] + assert b'bar' == ensure_bytes(store['foo']) + assert b'quux' == ensure_bytes(store['baz']) def test_iterators(self): store = self.create_store() @@ -146,9 +147,9 @@ def test_iterators(self): assert 4 == len(store) assert {'a', 'b', 'c/d', 'c/e/f'} == set(store) assert {'a', 'b', 'c/d', 'c/e/f'} == set(store.keys()) - assert {b'aaa', b'bbb', b'ddd', b'fff'} == set(store.values()) + assert {b'aaa', b'bbb', b'ddd', b'fff'} == set(map(ensure_bytes, store.values())) assert ({('a', b'aaa'), ('b', b'bbb'), ('c/d', b'ddd'), ('c/e/f', b'fff')} == - set(store.items())) + set(map(lambda kv: (kv[0], ensure_bytes(kv[1])), store.items()))) def test_pickle(self): @@ -172,8 +173,8 @@ def test_pickle(self): # verify assert n == len(store2) assert keys == sorted(store2.keys()) - assert b'bar' == store2['foo'] - assert b'quux' == store2['baz'] + assert b'bar' == ensure_bytes(store2['foo']) + assert b'quux' == ensure_bytes(store2['baz']) def test_getsize(self): store = self.create_store() @@ -635,10 +636,10 @@ def setdel_hierarchy_checks(store): # test __setitem__ overwrite level store['x/y/z'] = b'xxx' store['x/y'] = b'yyy' - assert b'yyy' == store['x/y'] + assert b'yyy' == ensure_bytes(store['x/y']) assert 'x/y/z' not in store store['x'] = b'zzz' - assert b'zzz' == store['x'] + assert b'zzz' == ensure_bytes(store['x']) assert 'x/y' not in store # test __delitem__ overwrite level @@ -700,7 +701,7 @@ def test_pickle_ext(self): # check point to same underlying directory assert 'xxx' not in store store2['xxx'] = b'yyy' - assert b'yyy' == store['xxx'] + assert b'yyy' == ensure_bytes(store['xxx']) def test_setdel(self): store = self.create_store()