def test_get_value_reference_and_bad_selection(self):
arg = Argument('test', reference='a', selection=['a', 0, '1', 'invalid'])
input_ = {'a': [[{'one': 1},
{'three': 3, 'b': 4}],
[{'one': 1}, {'two': 2}]],
'b': 15,
'c': 'something'}
with self.assertRaises(InvalidArgument):
arg.get_value({'a': input_, 'b': 2})
def test_get_value_reference_and_selection(self):
arg = Argument('test', reference='a', selection=['a', 0, '1', 'b'])
input_ = {'a': [[{'one': 1},
{'three': 3, 'b': 4}],
[{'one': 1}, {'two': 2}]],
'b': 15,
'c': 'something'}
self.assertEqual(arg.get_value({'a': input_, 'b': 2}), 4)
def test_get_value_reference_not_found(self):
arg = Argument('test', reference='c')
with self.assertRaises(InvalidArgument):
arg.get_value({'a': 1, 'b': 2})
def test_get_value_reference_only(self):
arg = Argument('test', reference='a')
self.assertEqual(arg.get_value({'a': 1, 'b': 2}), 1)
def test_get_value_value_with_reference(self):
arg = Argument('test', value=42, reference='a')
self.assertEqual(arg.get_value({'a': 1}), 42)
def test_get_value_value_only(self):
arg = Argument('test', value=42)
self.assertEqual(arg.get_value({}), 42)
