def test_multi_arg(self):
result = utils.ArgNormalizer.normalize(self.multi_arg)
self.assertEqual(
result, {
'multi_arg0':
utils.RawArg(
**{
'name': 'multi_arg0',
'default': None,
'alias': None,
'required': True
}),
'multi_arg1':
utils.RawArg(
**{
'name': 'multi_arg1',
'default': 'value',
'alias': None,
'required': True
}),
'multi_arg2':
utils.RawArg(
**{
'name': 'multi_arg2',
'default': 'value',
'alias': 'alias',
'required': False
})
})
def test_single_dict_arg_normalize(self):
result = utils.ArgNormalizer.normalize(self.single_dict_arg)
self.assertEqual(
result, {
'single_arg1':
utils.RawArg(
**{
'name': 'single_arg1',
'default': 'value',
'alias': None,
'required': True
})
})
def test_single_dict_arg_with_atrr(self):
result = utils.ArgNormalizer.normalize(self.single_dict_arg_with_attr)
self.assertEqual(
result, {
'single_arg2':
utils.RawArg(
**{
'name': 'single_arg2',
'default': 'value',
'alias': 'alias',
'required': False
})
})
def raw_args():
return dict(
raw=utils.RawArg(
**{
'name': 'name',
'alias': 'alias',
'required': True,
'default': 'default'
}),
raw_with_alias=utils.RawArg(**{
'name': 'name',
'alias': 'alias',
'required': True,
'default': None
}),
raw_required_with_default=utils.RawArg(**{
'name': 'name',
'alias': None,
'required': True,
'default': 'default'
}),
raw_required_without_default=utils.RawArg(**{
'name': 'name',
'alias': None,
'required': True,
'default': None
}),
raw_not_required_with_default=utils.RawArg(**{
'name': 'name',
'alias': None,
'required': False,
'default': 'default'
}),
raw_not_required_without_default=utils.RawArg(**{
'name': 'name',
'alias': None,
'required': False,
'default': None
}))