@@ -164,13 +164,16 @@ def test_Normalize():
164
164
_mask_tester (norm ,vals )
165
165
166
166
167
- class _base_NormMixin (object ):
167
+ class BaseNormMixin (object ):
168
168
def test_call (self ):
169
169
normed_vals = self .norm (self .vals )
170
170
assert_array_almost_equal (normed_vals ,self .expected )
171
171
172
172
def test_inverse (self ):
173
- _inverse_tester (self .norm ,self .vals )
173
+ if self .test_inverse :
174
+ _inverse_tester (self .norm ,self .vals )
175
+ else :
176
+ pass
174
177
175
178
def test_scalar (self ):
176
179
_scalar_tester (self .norm ,self .vals )
@@ -224,39 +227,42 @@ def test_process_value_array(self):
224
227
assert_array_equal (res ,np .array ([5. ,10. ]))
225
228
226
229
227
- class test_OffsetNorm_Even (_base_NormMixin ):
230
+ class BaseOffsetNorm (BaseNormMixin ):
231
+ normclass = mcolors .OffsetNorm
232
+ test_inverse = False
233
+
234
+ class test_OffsetNorm_Even (BaseOffsetNorm ):
228
235
def setup (self ):
229
- self .normclass = mcolors .OffsetNorm
230
236
self .norm = self .normclass (vmin = - 1 ,vcenter = 0 ,vmax = 4 )
231
237
self .vals = np .array ([- 1.0 ,- 0.5 ,0.0 ,1.0 ,2.0 ,3.0 ,4.0 ])
232
238
self .expected = np .array ([0.0 ,0.25 ,0.5 ,0.625 ,0.75 ,0.875 ,1.0 ])
233
239
234
240
235
- class test_OffsetNorm_Odd (_base_NormMixin ):
241
+ class test_OffsetNorm_Odd (BaseOffsetNorm ):
236
242
def setup (self ):
237
243
self .normclass = mcolors .OffsetNorm
238
244
self .norm = self .normclass (vmin = - 2 ,vcenter = 0 ,vmax = 5 )
239
245
self .vals = np .array ([- 2.0 ,- 1.0 ,0.0 ,1.0 ,2.0 ,3.0 ,4.0 ,5.0 ])
240
246
self .expected = np .array ([0.0 ,0.25 ,0.5 ,0.6 ,0.7 ,0.8 ,0.9 ,1.0 ])
241
247
242
248
243
- class test_OffsetNorm_AllNegative (_base_NormMixin ):
249
+ class test_OffsetNorm_AllNegative (BaseOffsetNorm ):
244
250
def setup (self ):
245
251
self .normclass = mcolors .OffsetNorm
246
252
self .norm = self .normclass (vmin = - 10 ,vcenter = - 8 ,vmax = - 2 )
247
253
self .vals = np .array ([- 10. ,- 9. ,- 8. ,- 6. ,- 4. ,- 2. ])
248
254
self .expected = np .array ([0.0 ,0.25 ,0.5 ,0.666667 ,0.833333 ,1.0 ])
249
255
250
256
251
- class test_OffsetNorm_AllPositive (_base_NormMixin ):
257
+ class test_OffsetNorm_AllPositive (BaseOffsetNorm ):
252
258
def setup (self ):
253
259
self .normclass = mcolors .OffsetNorm
254
260
self .norm = self .normclass (vmin = 0 ,vcenter = 3 ,vmax = 9 )
255
261
self .vals = np .array ([0. ,1.5 ,3. ,4.5 ,6.0 ,7.5 ,9. ])
256
262
self .expected = np .array ([0.0 ,0.25 ,0.5 ,0.625 ,0.75 ,0.875 ,1.0 ])
257
263
258
264
259
- class test_OffsetNorm_NoVs (_base_NormMixin ):
265
+ class test_OffsetNorm_NoVs (BaseOffsetNorm ):
260
266
def setup (self ):
261
267
self .normclass = mcolors .OffsetNorm
262
268
self .norm = self .normclass (vmin = None ,vcenter = None ,vmax = None )
@@ -283,23 +289,23 @@ def test_vmax(self):
283
289
assert_equal (self .norm .vmax ,self .expected_vmax )
284
290
285
291
286
- class test_OffsetNorm_VminEqualsVcenter (_base_NormMixin ):
292
+ class test_OffsetNorm_VminEqualsVcenter (BaseOffsetNorm ):
287
293
def setup (self ):
288
294
self .normclass = mcolors .OffsetNorm
289
295
self .norm = self .normclass (vmin = - 2 ,vcenter = - 2 ,vmax = 2 )
290
296
self .vals = np .array ([- 2.0 ,- 1.0 ,0.0 ,1.0 ,2.0 ])
291
297
self .expected = np .array ([0.5 ,0.625 ,0.75 ,0.875 ,1.0 ])
292
298
293
299
294
- class test_OffsetNorm_VmaxEqualsVcenter (_base_NormMixin ):
300
+ class test_OffsetNorm_VmaxEqualsVcenter (BaseOffsetNorm ):
295
301
def setup (self ):
296
302
self .normclass = mcolors .OffsetNorm
297
303
self .norm = self .normclass (vmin = - 2 ,vcenter = 2 ,vmax = 2 )
298
304
self .vals = np .array ([- 2.0 ,- 1.0 ,0.0 ,1.0 ,2.0 ])
299
305
self .expected = np .array ([0.0 ,0.125 ,0.25 ,0.375 ,0.5 ])
300
306
301
307
302
- class test_OffsetNorm_VsAllEqual (_base_NormMixin ):
308
+ class test_OffsetNorm_VsAllEqual (BaseOffsetNorm ):
303
309
def setup (self ):
304
310
self .v = 10
305
311
self .normclass = mcolors .OffsetNorm