1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61
|
spm_metadata = ['field', 'mandatory', 'copyfile', 'xor', 'requires', 'usedefault']
cmd_metadata = ['argstr', 'mandatory', 'copyfile', 'xor', 'requires', 'usedefault',
'sep']
py_metadata = ['mandatory', 'copyfile', 'xor', 'requires', 'usedefault']
def create_spmtest_func(pkg, interface, object):
cmd = ['def test_%s():'%interface.lower()]
cmd += ["yield assert_equal, %s.%s._jobtype, '%s'"%(pkg, interface, object._jobtype)]
cmd += ["yield assert_equal, %s.%s._jobname, '%s'"%(pkg, interface, object._jobname)]
input_fields = ''
for field, spec in object.inputs.items():
input_fields += '%s = dict('%field
for key, value in spec.__dict__.items():
if key in spm_metadata:
if key == 'field':
value = "'%s'"%value
input_fields += "%s=%s,"%(key,str(value))
input_fields += '),\n'
cmd += ['input_map = dict(%s)'%input_fields]
cmd += ['instance = %s.%s()'%(pkg, interface)]
cmd += ["""
for key, metadata in input_map.items():
for metakey, value in metadata.items():
yield assert_equal, getattr(instance.inputs.traits()[key], metakey), value"""]
return cmd
def create_cmdtest_func(pkg, interface, object):
cmd = ['def test_%s():'%interface.lower()]
input_fields = ''
for field, spec in object.inputs.items():
input_fields += '%s = dict('%field
for key, value in spec.__dict__.items():
if key in cmd_metadata:
if key == 'argstr':
value = "'%s'"%value
input_fields += "%s=%s,"%(key,str(value))
input_fields += '),\n'
cmd += ['input_map = dict(%s)'%input_fields]
cmd += ['\tinstance = %s.%s()'%(pkg, interface)]
cmd += ["""\tfor key, metadata in input_map.items():
for metakey, value in metadata.items():
yield assert_equal, getattr(instance.inputs.traits()[key], metakey), value"""]
print '\n'.join(cmd)+'\n'
def create_pytest_func(pkg, interface, object):
cmd = ['def test_%s():'%interface.lower()]
input_fields = ''
for field, spec in object.inputs.items():
input_fields += '%s = dict('%field
for key, value in spec.__dict__.items():
if key in cmd_metadata:
if key == 'argstr':
value = "'%s'"%value
input_fields += "%s=%s,"%(key,str(value))
input_fields += '),\n'
cmd += ['input_map = dict(%s)'%input_fields]
cmd += ['\tinstance = %s.%s()'%(pkg, interface)]
cmd += ["""\tfor key, metadata in input_map.items():
for metakey, value in metadata.items():
yield assert_equal, getattr(instance.inputs.traits()[key], metakey), value"""]
print '\n'.join(cmd)+'\n'
|