4
4
import base64
5
5
6
6
7
- from nose .tools import eq_ as eq , assert_is_none , assert_raises
8
7
import numpy as np
8
+ import pytest
9
9
10
10
11
11
from zarr .compat import binary_type , text_type , PY2
15
15
from zarr .codecs import Delta , Zlib , Blosc
16
16
17
17
18
- def assert_json_eq (expect , actual ):
18
+ def assert_json_equal (expect , actual ):
19
19
if isinstance (expect , binary_type ): # pragma: py3 no cover
20
20
expect = text_type (expect , 'ascii' )
21
21
if isinstance (actual , binary_type ):
22
22
actual = text_type (actual , 'ascii' )
23
23
ej = json .loads (expect )
24
24
aj = json .loads (actual )
25
- eq ( ej , aj )
25
+ assert ej == aj
26
26
27
27
28
28
def test_encode_decode_array_1 ():
@@ -50,18 +50,18 @@ def test_encode_decode_array_1():
50
50
51
51
# test encoding
52
52
meta_enc = encode_array_metadata (meta )
53
- assert_json_eq (meta_json , meta_enc )
53
+ assert_json_equal (meta_json , meta_enc )
54
54
55
55
# test decoding
56
56
meta_dec = decode_array_metadata (meta_enc )
57
- eq ( ZARR_FORMAT , meta_dec ['zarr_format' ])
58
- eq ( meta ['shape' ], meta_dec ['shape' ])
59
- eq ( meta ['chunks' ], meta_dec ['chunks' ])
60
- eq ( meta ['dtype' ], meta_dec ['dtype' ])
61
- eq ( meta ['compressor' ], meta_dec ['compressor' ])
62
- eq ( meta ['order' ], meta_dec ['order' ])
63
- assert_is_none ( meta_dec ['fill_value' ])
64
- assert_is_none ( meta_dec ['filters' ])
57
+ assert ZARR_FORMAT == meta_dec ['zarr_format' ]
58
+ assert meta ['shape' ] == meta_dec ['shape' ]
59
+ assert meta ['chunks' ] == meta_dec ['chunks' ]
60
+ assert meta ['dtype' ] == meta_dec ['dtype' ]
61
+ assert meta ['compressor' ] == meta_dec ['compressor' ]
62
+ assert meta ['order' ] == meta_dec ['order' ]
63
+ assert meta_dec ['fill_value' ] is None
64
+ assert meta_dec ['filters' ] is None
65
65
66
66
67
67
def test_encode_decode_array_2 ():
@@ -100,19 +100,19 @@ def test_encode_decode_array_2():
100
100
101
101
# test encoding
102
102
meta_enc = encode_array_metadata (meta )
103
- assert_json_eq (meta_json , meta_enc )
103
+ assert_json_equal (meta_json , meta_enc )
104
104
105
105
# test decoding
106
106
meta_dec = decode_array_metadata (meta_enc )
107
- eq ( ZARR_FORMAT , meta_dec ['zarr_format' ])
108
- eq ( meta ['shape' ], meta_dec ['shape' ])
109
- eq ( meta ['chunks' ], meta_dec ['chunks' ])
110
- eq ( meta ['dtype' ], meta_dec ['dtype' ])
111
- eq ( meta ['compressor' ], meta_dec ['compressor' ])
112
- eq ( meta ['order' ], meta_dec ['order' ])
107
+ assert ZARR_FORMAT == meta_dec ['zarr_format' ]
108
+ assert meta ['shape' ] == meta_dec ['shape' ]
109
+ assert meta ['chunks' ] == meta_dec ['chunks' ]
110
+ assert meta ['dtype' ] == meta_dec ['dtype' ]
111
+ assert meta ['compressor' ] == meta_dec ['compressor' ]
112
+ assert meta ['order' ] == meta_dec ['order' ]
113
113
np_fill_value = np .array (meta ['fill_value' ], dtype = meta ['dtype' ])[()]
114
- eq ( np_fill_value , meta_dec ['fill_value' ])
115
- eq ( [df .get_config ()], meta_dec ['filters' ])
114
+ assert np_fill_value == meta_dec ['fill_value' ]
115
+ assert [df .get_config ()] == meta_dec ['filters' ]
116
116
117
117
118
118
def test_encode_decode_fill_values_nan ():
@@ -148,7 +148,7 @@ def test_encode_decode_fill_values_nan():
148
148
149
149
# test encoding
150
150
meta_enc = encode_array_metadata (meta )
151
- assert_json_eq (meta_json , meta_enc )
151
+ assert_json_equal (meta_json , meta_enc )
152
152
153
153
# test decoding
154
154
meta_dec = decode_array_metadata (meta_enc )
@@ -191,13 +191,13 @@ def test_encode_decode_fill_values_bytes():
191
191
}''' % (s , ZARR_FORMAT )
192
192
193
193
# test encoding
194
- assert_json_eq (meta_json , meta_enc )
194
+ assert_json_equal (meta_json , meta_enc )
195
195
196
196
# test decoding
197
197
meta_dec = decode_array_metadata (meta_enc )
198
198
actual = meta_dec ['fill_value' ]
199
199
expect = np .array (v , dtype = dtype )[()]
200
- eq ( expect , actual )
200
+ assert expect == actual
201
201
202
202
203
203
def test_decode_array_unsupported_format ():
@@ -212,7 +212,7 @@ def test_decode_array_unsupported_format():
212
212
"fill_value": null,
213
213
"order": "C"
214
214
}''' % (ZARR_FORMAT - 1 )
215
- with assert_raises (MetadataError ):
215
+ with pytest . raises (MetadataError ):
216
216
decode_array_metadata (meta_json )
217
217
218
218
@@ -222,7 +222,7 @@ def test_decode_array_missing_fields():
222
222
meta_json = '''{
223
223
"zarr_format": %s
224
224
}''' % ZARR_FORMAT
225
- with assert_raises (MetadataError ):
225
+ with pytest . raises (MetadataError ):
226
226
decode_array_metadata (meta_json )
227
227
228
228
@@ -233,7 +233,7 @@ def test_encode_decode_dtype():
233
233
s = json .dumps (e ) # check JSON serializable
234
234
o = json .loads (s )
235
235
d = decode_dtype (o )
236
- eq ( np .dtype (dt ), d )
236
+ assert np .dtype (dt ) == d
237
237
238
238
239
239
def test_decode_group ():
@@ -243,11 +243,11 @@ def test_decode_group():
243
243
"zarr_format": %s
244
244
}''' % ZARR_FORMAT
245
245
meta = decode_group_metadata (b )
246
- eq ( ZARR_FORMAT , meta ['zarr_format' ])
246
+ assert ZARR_FORMAT == meta ['zarr_format' ]
247
247
248
248
# unsupported format
249
249
b = '''{
250
250
"zarr_format": %s
251
251
}''' % (ZARR_FORMAT - 1 )
252
- with assert_raises (MetadataError ):
252
+ with pytest . raises (MetadataError ):
253
253
decode_group_metadata (b )
0 commit comments