38
38
# Each test involves loading an image of shape SHAPE, and then slicing it
39
39
# NITERS times
40
40
NITERS = 50
41
- SHAPE = (100 , 100 , 100 , 100 )
41
+ SHAPE = (100 , 100 , 100 , 100 )
42
42
43
43
# One test is run for each combination of SLICEOBJS, KEEP_OPENS, and HAVE_IGZIP
44
44
61
61
62
62
@contextlib .contextmanager
63
63
def patch_indexed_gzip (have_igzip ):
64
- with mock .patch ('nibabel.openers.HAVE_INDEXED_GZIP' , have_igzip ), \
65
- mock .patch ('nibabel.arrayproxy.HAVE_INDEXED_GZIP' , have_igzip ):
64
+
65
+ atts = ['nibabel.openers.HAVE_INDEXED_GZIP' ,
66
+ 'nibabel.arrayproxy.HAVE_INDEXED_GZIP' ]
67
+
68
+ with mock .patch (atts [0 ], have_igzip ), mock .patch (atts [1 ], have_igzip ):
66
69
yield
67
70
68
71
@@ -78,12 +81,12 @@ def bench_arrayproxy_slicing():
78
81
# because if keep_file_open is False, HAVE_INDEXED_GZIP has no effect
79
82
tests = [t for t in tests if not (t [0 ] and not t [1 ])]
80
83
81
- testfile = 'testfile.nii'
84
+ testfile = 'testfile.nii'
82
85
testfilegz = 'test.nii.gz'
83
86
84
87
def get_test_label (test ):
85
88
have_igzip = test [0 ]
86
- keep_open = test [1 ]
89
+ keep_open = test [1 ]
87
90
88
91
if not (have_igzip and keep_open ):
89
92
return 'gzip'
@@ -166,9 +169,9 @@ def testfunc():
166
169
gc .collect ()
167
170
168
171
if memory_usage is not None :
169
- membaseline = max (memory_usage (lambda : None ))
170
- testmem = max (memory_usage (testfunc )) - membaseline
171
- basemem = max (memory_usage (basefunc )) - membaseline
172
+ membaseline = max (memory_usage (lambda : None ))
173
+ testmem = max (memory_usage (testfunc )) - membaseline
174
+ basemem = max (memory_usage (basefunc )) - membaseline
172
175
else :
173
176
testmem = np .nan
174
177
basemem = np .nan
0 commit comments