def argumentTypes(self):
return [
Argument(keyword="input1",
optional=False,
argument_type=DataType.number,
tags=[Argument.Tag('from', Argument.TagPosition.Before)]),
Argument(keyword='input2',
optional=False,
argument_type=DataType.number,
tags=[Argument.Tag('from', Argument.TagPosition.After)])
]
def argumentTypes(self):
return [
Argument(keyword="input1",
optional=False,
argument_type=DataType.user_string,
tags=[Argument.Tag('echo', Argument.TagPosition.After)])
]
def test_extract_arg_from_user(self):
key_words = 'find numbers between 2 and 4'.split(' ')
arg = Argument(DataType.number,
tags=[Argument.Tag('between', 1)],
number=2)
res = AlfaDataParser.extractArgFromUser(key_words, arg)
self.assertEqual(len(res), 2)
self.assertEqual(res[0].data, 2)
self.assertEqual(res[1].data, 4)