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)
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="dummy", optional=True, argument_type=DataType.string, fill_from_cache=False) ]
def test_fill_optional_arguments_already_found(self): # Set up a few dummy arguments argumentTypes = [ Argument(keyword="dummy1", optional=True, argument_type=DataType.string), Argument(keyword="dummy2", optional=False, argument_type=DataType.string) ] self.history.add(DataType.string, ["input", "my"], "dummy input") argumentsFound = {'dummy1': 'Found'} self.parser.fillOptionalArguments(argumentsFound, argumentTypes) self.assertTrue('dummy1' in argumentsFound) self.assertFalse('dummy2' in argumentsFound) self.assertEqual(argumentsFound['dummy1'], 'Found')
def argumentTypes(self): return [ Argument(keyword="dummy", optional=False, argument_type=DataType.string, number=-1) ]
def test_fill_optional_arguments(self): # Set up a few dummy arguments argumentTypes = [ Argument(keyword="dummy1", optional=True, argument_type=DataType.string), Argument(keyword="dummy2", optional=False, argument_type=DataType.string) ] self.history.add(DataType.string, ["input", "my"], "dummy input") data_object = self.history.search(DataType.string, ["my", "input"])[0] argumentsFound = {} self.parser.fillOptionalArguments(argumentsFound, argumentTypes) self.assertTrue('dummy1' in argumentsFound) self.assertFalse('dummy2' in argumentsFound) self.assertEqual(argumentsFound['dummy1'], data_object)
def test_unsuccessful_fill_closest_arguments(self): argument_types = [ Argument(keyword="dummy1", optional=False, argument_type=DataType.string) ] self.history.add(DataType.string, ["mean", "tumor", "length"], "5.0") self.history.add(DataType.string, ["mean", "tumor", "id"], "20") self.history.add(DataType.string, ["tumor", "length"], "100") argument_search_results = { "dummy1": self.history.search(DataType.string, ["mean", "tumor"]) } arguments_found = {} self.parser.fillClosestArguments(argument_search_results, arguments_found, argument_types) self.assertFalse("dummy1" in arguments_found)
def argumentTypes(self): return [ Argument(keyword="input1", optional=False, argument_type=[DataType.string, DataType.array]) ]