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])
]