4
4
import unittest
5
5
6
6
7
- from nose . tools import eq_ as eq , assert_raises
7
+ import pytest
8
8
9
9
10
10
from zarr .attrs import Attributes
@@ -24,14 +24,14 @@ def test_storage(self):
24
24
a = Attributes (store = store , key = 'attrs' )
25
25
assert 'foo' not in a
26
26
assert 'bar' not in a
27
- eq ( dict (), a .asdict () )
27
+ assert dict () == a .asdict ()
28
28
29
29
a ['foo' ] = 'bar'
30
30
a ['baz' ] = 42
31
31
assert 'attrs' in store
32
32
assert isinstance (store ['attrs' ], binary_type )
33
33
d = json .loads (text_type (store ['attrs' ], 'ascii' ))
34
- eq ( dict (foo = 'bar' , baz = 42 ), d )
34
+ assert dict (foo = 'bar' , baz = 42 ) == d
35
35
36
36
def test_get_set_del_contains (self ):
37
37
@@ -41,11 +41,11 @@ def test_get_set_del_contains(self):
41
41
a ['baz' ] = 42
42
42
assert 'foo' in a
43
43
assert 'baz' in a
44
- eq ( 'bar' , a ['foo' ])
45
- eq ( 42 , a ['baz' ])
44
+ assert 'bar' == a ['foo' ]
45
+ assert 42 == a ['baz' ]
46
46
del a ['foo' ]
47
47
assert 'foo' not in a
48
- with assert_raises (KeyError ):
48
+ with pytest . raises (KeyError ):
49
49
# noinspection PyStatementEffect
50
50
a ['foo' ]
51
51
@@ -57,44 +57,44 @@ def test_update_put(self):
57
57
assert 'baz' not in a
58
58
59
59
a .update (foo = 'spam' , bar = 42 , baz = 4.2 )
60
- eq ( a ['foo' ], 'spam' )
61
- eq ( a ['bar' ], 42 )
62
- eq ( a ['baz' ], 4.2 )
60
+ assert a ['foo' ] == 'spam'
61
+ assert a ['bar' ] == 42
62
+ assert a ['baz' ] == 4.2
63
63
64
64
a .put (dict (foo = 'eggs' , bar = 84 ))
65
- eq ( a ['foo' ], 'eggs' )
66
- eq ( a ['bar' ], 84 )
65
+ assert a ['foo' ] == 'eggs'
66
+ assert a ['bar' ] == 84
67
67
assert 'baz' not in a
68
68
69
69
def test_iterators (self ):
70
70
71
71
a = self .init_attributes (dict ())
72
- eq ( 0 , len (a ) )
73
- eq ( set (), set (a ) )
74
- eq ( set (), set (a .keys () ))
75
- eq ( set (), set (a .values () ))
76
- eq ( set (), set (a .items () ))
72
+ assert 0 == len (a )
73
+ assert set () == set (a )
74
+ assert set () == set (a .keys ())
75
+ assert set () == set (a .values ())
76
+ assert set () == set (a .items ())
77
77
78
78
a ['foo' ] = 'bar'
79
79
a ['baz' ] = 42
80
80
81
- eq ( 2 , len (a ) )
82
- eq ( {'foo' , 'baz' }, set (a ) )
83
- eq ( {'foo' , 'baz' }, set (a .keys () ))
84
- eq ( {'bar' , 42 }, set (a .values () ))
85
- eq ( {('foo' , 'bar' ), ('baz' , 42 )}, set (a .items () ))
81
+ assert 2 == len (a )
82
+ assert {'foo' , 'baz' } == set (a )
83
+ assert {'foo' , 'baz' } == set (a .keys ())
84
+ assert {'bar' , 42 } == set (a .values ())
85
+ assert {('foo' , 'bar' ), ('baz' , 42 )} == set (a .items ())
86
86
87
87
def test_read_only (self ):
88
88
store = dict ()
89
89
a = self .init_attributes (store , read_only = True )
90
90
store ['attrs' ] = json .dumps (dict (foo = 'bar' , baz = 42 )).encode ('ascii' )
91
- eq ( a ['foo' ], 'bar' )
92
- eq ( a ['baz' ], 42 )
93
- with assert_raises (PermissionError ):
91
+ assert a ['foo' ] == 'bar'
92
+ assert a ['baz' ] == 42
93
+ with pytest . raises (PermissionError ):
94
94
a ['foo' ] = 'quux'
95
- with assert_raises (PermissionError ):
95
+ with pytest . raises (PermissionError ):
96
96
del a ['foo' ]
97
- with assert_raises (PermissionError ):
97
+ with pytest . raises (PermissionError ):
98
98
a .update (foo = 'quux' )
99
99
100
100
def test_key_completions (self ):
@@ -118,108 +118,108 @@ def test_caching_on(self):
118
118
119
119
# setup store
120
120
store = CountingDict ()
121
- eq ( 0 , store .counter ['__getitem__' , 'attrs' ])
122
- eq ( 0 , store .counter ['__setitem__' , 'attrs' ])
121
+ assert 0 == store .counter ['__getitem__' , 'attrs' ]
122
+ assert 0 == store .counter ['__setitem__' , 'attrs' ]
123
123
store ['attrs' ] = json .dumps (dict (foo = 'xxx' , bar = 42 )).encode ('ascii' )
124
- eq ( 0 , store .counter ['__getitem__' , 'attrs' ])
125
- eq ( 1 , store .counter ['__setitem__' , 'attrs' ])
124
+ assert 0 == store .counter ['__getitem__' , 'attrs' ]
125
+ assert 1 == store .counter ['__setitem__' , 'attrs' ]
126
126
127
127
# setup attributes
128
128
a = self .init_attributes (store )
129
129
130
130
# test __getitem__ causes all attributes to be cached
131
- eq ( a ['foo' ], 'xxx' )
132
- eq ( 1 , store .counter ['__getitem__' , 'attrs' ])
133
- eq ( a ['bar' ], 42 )
134
- eq ( 1 , store .counter ['__getitem__' , 'attrs' ])
135
- eq ( a ['foo' ], 'xxx' )
136
- eq ( 1 , store .counter ['__getitem__' , 'attrs' ])
131
+ assert a ['foo' ] == 'xxx'
132
+ assert 1 == store .counter ['__getitem__' , 'attrs' ]
133
+ assert a ['bar' ] == 42
134
+ assert 1 == store .counter ['__getitem__' , 'attrs' ]
135
+ assert a ['foo' ] == 'xxx'
136
+ assert 1 == store .counter ['__getitem__' , 'attrs' ]
137
137
138
138
# test __setitem__ updates the cache
139
139
a ['foo' ] = 'yyy'
140
- eq ( 2 , store .counter ['__getitem__' , 'attrs' ])
141
- eq ( 2 , store .counter ['__setitem__' , 'attrs' ])
142
- eq ( a ['foo' ], 'yyy' )
143
- eq ( 2 , store .counter ['__getitem__' , 'attrs' ])
144
- eq ( 2 , store .counter ['__setitem__' , 'attrs' ])
140
+ assert 2 == store .counter ['__getitem__' , 'attrs' ]
141
+ assert 2 == store .counter ['__setitem__' , 'attrs' ]
142
+ assert a ['foo' ] == 'yyy'
143
+ assert 2 == store .counter ['__getitem__' , 'attrs' ]
144
+ assert 2 == store .counter ['__setitem__' , 'attrs' ]
145
145
146
146
# test update() updates the cache
147
147
a .update (foo = 'zzz' , bar = 84 )
148
- eq ( 3 , store .counter ['__getitem__' , 'attrs' ])
149
- eq ( 3 , store .counter ['__setitem__' , 'attrs' ])
150
- eq ( a ['foo' ], 'zzz' )
151
- eq ( a ['bar' ], 84 )
152
- eq ( 3 , store .counter ['__getitem__' , 'attrs' ])
153
- eq ( 3 , store .counter ['__setitem__' , 'attrs' ])
148
+ assert 3 == store .counter ['__getitem__' , 'attrs' ]
149
+ assert 3 == store .counter ['__setitem__' , 'attrs' ]
150
+ assert a ['foo' ] == 'zzz'
151
+ assert a ['bar' ] == 84
152
+ assert 3 == store .counter ['__getitem__' , 'attrs' ]
153
+ assert 3 == store .counter ['__setitem__' , 'attrs' ]
154
154
155
155
# test __contains__ uses the cache
156
156
assert 'foo' in a
157
- eq ( 3 , store .counter ['__getitem__' , 'attrs' ])
158
- eq ( 3 , store .counter ['__setitem__' , 'attrs' ])
157
+ assert 3 == store .counter ['__getitem__' , 'attrs' ]
158
+ assert 3 == store .counter ['__setitem__' , 'attrs' ]
159
159
assert 'spam' not in a
160
- eq ( 3 , store .counter ['__getitem__' , 'attrs' ])
161
- eq ( 3 , store .counter ['__setitem__' , 'attrs' ])
160
+ assert 3 == store .counter ['__getitem__' , 'attrs' ]
161
+ assert 3 == store .counter ['__setitem__' , 'attrs' ]
162
162
163
163
# test __delitem__ updates the cache
164
164
del a ['bar' ]
165
- eq ( 4 , store .counter ['__getitem__' , 'attrs' ])
166
- eq ( 4 , store .counter ['__setitem__' , 'attrs' ])
165
+ assert 4 == store .counter ['__getitem__' , 'attrs' ]
166
+ assert 4 == store .counter ['__setitem__' , 'attrs' ]
167
167
assert 'bar' not in a
168
- eq ( 4 , store .counter ['__getitem__' , 'attrs' ])
169
- eq ( 4 , store .counter ['__setitem__' , 'attrs' ])
168
+ assert 4 == store .counter ['__getitem__' , 'attrs' ]
169
+ assert 4 == store .counter ['__setitem__' , 'attrs' ]
170
170
171
171
# test refresh()
172
172
store ['attrs' ] = json .dumps (dict (foo = 'xxx' , bar = 42 )).encode ('ascii' )
173
- eq ( 4 , store .counter ['__getitem__' , 'attrs' ])
173
+ assert 4 == store .counter ['__getitem__' , 'attrs' ]
174
174
a .refresh ()
175
- eq ( 5 , store .counter ['__getitem__' , 'attrs' ])
176
- eq ( a ['foo' ], 'xxx' )
177
- eq ( 5 , store .counter ['__getitem__' , 'attrs' ])
178
- eq ( a ['bar' ], 42 )
179
- eq ( 5 , store .counter ['__getitem__' , 'attrs' ])
175
+ assert 5 == store .counter ['__getitem__' , 'attrs' ]
176
+ assert a ['foo' ] == 'xxx'
177
+ assert 5 == store .counter ['__getitem__' , 'attrs' ]
178
+ assert a ['bar' ] == 42
179
+ assert 5 == store .counter ['__getitem__' , 'attrs' ]
180
180
181
181
def test_caching_off (self ):
182
182
183
183
# setup store
184
184
store = CountingDict ()
185
- eq ( 0 , store .counter ['__getitem__' , 'attrs' ])
186
- eq ( 0 , store .counter ['__setitem__' , 'attrs' ])
185
+ assert 0 == store .counter ['__getitem__' , 'attrs' ]
186
+ assert 0 == store .counter ['__setitem__' , 'attrs' ]
187
187
store ['attrs' ] = json .dumps (dict (foo = 'xxx' , bar = 42 )).encode ('ascii' )
188
- eq ( 0 , store .counter ['__getitem__' , 'attrs' ])
189
- eq ( 1 , store .counter ['__setitem__' , 'attrs' ])
188
+ assert 0 == store .counter ['__getitem__' , 'attrs' ]
189
+ assert 1 == store .counter ['__setitem__' , 'attrs' ]
190
190
191
191
# setup attributes
192
192
a = self .init_attributes (store , cache = False )
193
193
194
194
# test __getitem__
195
- eq ( a ['foo' ], 'xxx' )
196
- eq ( 1 , store .counter ['__getitem__' , 'attrs' ])
197
- eq ( a ['bar' ], 42 )
198
- eq ( 2 , store .counter ['__getitem__' , 'attrs' ])
199
- eq ( a ['foo' ], 'xxx' )
200
- eq ( 3 , store .counter ['__getitem__' , 'attrs' ])
195
+ assert a ['foo' ] == 'xxx'
196
+ assert 1 == store .counter ['__getitem__' , 'attrs' ]
197
+ assert a ['bar' ] == 42
198
+ assert 2 == store .counter ['__getitem__' , 'attrs' ]
199
+ assert a ['foo' ] == 'xxx'
200
+ assert 3 == store .counter ['__getitem__' , 'attrs' ]
201
201
202
202
# test __setitem__
203
203
a ['foo' ] = 'yyy'
204
- eq ( 4 , store .counter ['__getitem__' , 'attrs' ])
205
- eq ( 2 , store .counter ['__setitem__' , 'attrs' ])
206
- eq ( a ['foo' ], 'yyy' )
207
- eq ( 5 , store .counter ['__getitem__' , 'attrs' ])
208
- eq ( 2 , store .counter ['__setitem__' , 'attrs' ])
204
+ assert 4 == store .counter ['__getitem__' , 'attrs' ]
205
+ assert 2 == store .counter ['__setitem__' , 'attrs' ]
206
+ assert a ['foo' ] == 'yyy'
207
+ assert 5 == store .counter ['__getitem__' , 'attrs' ]
208
+ assert 2 == store .counter ['__setitem__' , 'attrs' ]
209
209
210
210
# test update()
211
211
a .update (foo = 'zzz' , bar = 84 )
212
- eq ( 6 , store .counter ['__getitem__' , 'attrs' ])
213
- eq ( 3 , store .counter ['__setitem__' , 'attrs' ])
214
- eq ( a ['foo' ], 'zzz' )
215
- eq ( a ['bar' ], 84 )
216
- eq ( 8 , store .counter ['__getitem__' , 'attrs' ])
217
- eq ( 3 , store .counter ['__setitem__' , 'attrs' ])
212
+ assert 6 == store .counter ['__getitem__' , 'attrs' ]
213
+ assert 3 == store .counter ['__setitem__' , 'attrs' ]
214
+ assert a ['foo' ] == 'zzz'
215
+ assert a ['bar' ] == 84
216
+ assert 8 == store .counter ['__getitem__' , 'attrs' ]
217
+ assert 3 == store .counter ['__setitem__' , 'attrs' ]
218
218
219
219
# test __contains__
220
220
assert 'foo' in a
221
- eq ( 9 , store .counter ['__getitem__' , 'attrs' ])
222
- eq ( 3 , store .counter ['__setitem__' , 'attrs' ])
221
+ assert 9 == store .counter ['__getitem__' , 'attrs' ]
222
+ assert 3 == store .counter ['__setitem__' , 'attrs' ]
223
223
assert 'spam' not in a
224
- eq ( 10 , store .counter ['__getitem__' , 'attrs' ])
225
- eq ( 3 , store .counter ['__setitem__' , 'attrs' ])
224
+ assert 10 == store .counter ['__getitem__' , 'attrs' ]
225
+ assert 3 == store .counter ['__setitem__' , 'attrs' ]
0 commit comments