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)