Skip to content

Commit 0bac3c7

Browse files
committed
fix: updated auto test generator to use relative imports
1 parent 60133ac commit 0bac3c7

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

tools/checkspecs.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -207,8 +207,9 @@ def test_specs(self, uri):
207207
if not os.path.exists(nonautotest):
208208
with open(testfile, 'wt') as fp:
209209
cmd = ['# AUTO-GENERATED by tools/checkspecs.py - DO NOT EDIT',
210-
'from nipype.testing import assert_equal',
211-
'from %s import %s' % (uri, c),
210+
'from %stesting import assert_equal' %
211+
('.' * len(uri.split('.'))),
212+
'from ..%s import %s' % (uri.split('.')[-1], c),
212213
'']
213214
cmd.append('def test_%s_inputs():' % c)
214215
input_fields = ''
@@ -229,8 +230,8 @@ def test_specs(self, uri):
229230
cmd += [' input_map = dict(%s)' % input_fields]
230231
cmd += [' inputs = %s.input_spec()' % c]
231232
cmd += ["""
232-
for key, metadata in input_map.items():
233-
for metakey, value in metadata.items():
233+
for key, metadata in list(input_map.items()):
234+
for metakey, value in list(metadata.items()):
234235
yield assert_equal, getattr(inputs.traits()[key], metakey), value"""]
235236
fp.writelines('\n'.join(cmd) + '\n\n')
236237
else:
@@ -273,8 +274,8 @@ def test_specs(self, uri):
273274
cmd += [' output_map = dict(%s)' % input_fields]
274275
cmd += [' outputs = %s.output_spec()' % c]
275276
cmd += ["""
276-
for key, metadata in output_map.items():
277-
for metakey, value in metadata.items():
277+
for key, metadata in list(output_map.items()):
278+
for metakey, value in list(metadata.items()):
278279
yield assert_equal, getattr(outputs.traits()[key], metakey), value"""]
279280
fp.writelines('\n'.join(cmd) + '\n\n')
280281

0 commit comments

Comments
 (0)