7
7
import numpy as np
8
8
9
9
10
+ from zarr .compat import PY2
10
11
from zarr .meta import decode_array_metadata , encode_dtype , decode_dtype , \
11
12
ZARR_FORMAT
12
13
from zarr .errors import MetadataError
15
16
def test_decode_array ():
16
17
17
18
# typical
18
- b = b '''{
19
+ b = '''{
19
20
"zarr_format": %s,
20
21
"shape": [100],
21
22
"chunks": [10],
@@ -24,7 +25,9 @@ def test_decode_array():
24
25
"compression_opts": 1,
25
26
"fill_value": null,
26
27
"order": "C"
27
- }''' % str (ZARR_FORMAT ).encode ('ascii' )
28
+ }''' % ZARR_FORMAT
29
+ if not PY2 :
30
+ b = b .encode ('ascii' )
28
31
meta = decode_array_metadata (b )
29
32
eq (ZARR_FORMAT , meta ['zarr_format' ])
30
33
eq ((100 ,), meta ['shape' ])
@@ -36,7 +39,7 @@ def test_decode_array():
36
39
eq ('C' , meta ['order' ])
37
40
38
41
# variations
39
- b = b '''{
42
+ b = '''{
40
43
"zarr_format": %s,
41
44
"shape": [100, 100],
42
45
"chunks": [10, 10],
@@ -49,7 +52,9 @@ def test_decode_array():
49
52
},
50
53
"fill_value": 42,
51
54
"order": "F"
52
- }''' % str (ZARR_FORMAT ).encode ('ascii' )
55
+ }''' % ZARR_FORMAT
56
+ if not PY2 :
57
+ b = b .encode ('ascii' )
53
58
meta = decode_array_metadata (b )
54
59
eq (ZARR_FORMAT , meta ['zarr_format' ])
55
60
eq ((100 , 100 ), meta ['shape' ])
@@ -63,7 +68,7 @@ def test_decode_array():
63
68
eq ('F' , meta ['order' ])
64
69
65
70
# unsupported format
66
- b = b '''{
71
+ b = '''{
67
72
"zarr_format": %s,
68
73
"shape": [100],
69
74
"chunks": [10],
@@ -72,14 +77,18 @@ def test_decode_array():
72
77
"compression_opts": 1,
73
78
"fill_value": null,
74
79
"order": "C"
75
- }''' % str (ZARR_FORMAT - 1 ).encode ('ascii' )
80
+ }''' % (ZARR_FORMAT - 1 )
81
+ if not PY2 :
82
+ b = b .encode ('ascii' )
76
83
with assert_raises (MetadataError ):
77
84
decode_array_metadata (b )
78
85
79
86
# missing fields
80
- b = b '''{
87
+ b = '''{
81
88
"zarr_format": %s
82
- }''' % str (ZARR_FORMAT ).encode ('ascii' )
89
+ }''' % ZARR_FORMAT
90
+ if not PY2 :
91
+ b = b .encode ('ascii' )
83
92
with assert_raises (MetadataError ):
84
93
decode_array_metadata (b )
85
94
0 commit comments