def test_aic_defroutinerootname():
'''Check that the _def_routine_root_name() internal method behaves as
expected.
'''
symbol_name = "dummy"
kernel_functor = KernelFunctor(DataTypeSymbol(symbol_name, REAL_TYPE))
routine = RoutineSymbol("hello")
index = 3
call = AlgorithmInvokeCall(routine, index)
call.children = [kernel_functor]
assert call._def_routine_root_name() == "invoke_{0}_{1}".format(
index, symbol_name)
call.children.append(kernel_functor.copy())
assert call._def_routine_root_name() == "invoke_{0}".format(index)
def test_kernelfunctor_str():
'''Check the str method of the KernelFunctor class.'''
symbol = DataTypeSymbol("hello", StructureType())
arg = Reference(Symbol("dummy"))
klr = KernelFunctor.create(symbol, [arg])
assert klr.__str__() == "KernelFunctor[name='hello']"
def test_kernelfunctor_node_str():
'''Check the node_str method of the KernelFunctor class.'''
symbol = DataTypeSymbol("hello", StructureType())
arg = Reference(Symbol("dummy"))
klr = KernelFunctor.create(symbol, [arg])
coloredtext = colored("KernelFunctor", KernelFunctor._colour)
assert klr.node_str() == coloredtext + "[name='hello']"
def test_kernelfunctor_invalid_symbol():
'''Check KernelFunctor raises the expected exception if the type of
the symbol argument is invalid.
'''
with pytest.raises(TypeError) as info:
_ = KernelFunctor(Symbol("hello"))
assert ("KernelFunctor symbol argument should be a DataTypeSymbol but "
"found 'Symbol'." in str(info.value))
def test_kernelfunctor_parent():
'''Check that the optional parent argument to a KernelFunctor class
constructor is stored correctly.
'''
parent = Node()
symbol = DataTypeSymbol("hello", StructureType())
klr = KernelFunctor(symbol, parent=parent)
assert klr.parent == parent
def test_kernelfunctor_create_invalid_args1():
'''Check that the create method of KernelFunctor raises the expected
exception if the provided 'arguments' argument is not a list.
'''
symbol = DataTypeSymbol("hello", StructureType())
with pytest.raises(GenerationError) as info:
_ = KernelFunctor.create(symbol, "Not a list")
assert ("KernelFunctor create() arguments argument should be a list "
"but found 'str'." in str(info.value))
def test_kernelfunctor_create_invalid_symbol():
'''Check that the create method of KernelFunctor raises the expected
exception if the provided symbol argument is not the correct type.
'''
symbol = Symbol("hello")
with pytest.raises(GenerationError) as info:
_ = KernelFunctor.create(symbol, [])
assert ("KernelFunctor create() symbol argument should be a DataTypeSymbol"
" but found 'Symbol'." in str(info.value))
def test_kernelfunctor_invalid_args2():
'''Check that the create method of KernelFunctor raises the expected
exception if its supplied list of children are not the expected
type (tests _validate_child method and _children_valid_format
variable)
'''
symbol = DataTypeSymbol("hello", StructureType())
with pytest.raises(GenerationError) as info:
_ = KernelFunctor.create(symbol, ["hello"])
assert ("Item 'str' can't be child 0 of 'KernelFunctor'. The valid "
"format is: '[DataNode]*'." in str(info.value))
def test_kernelfunctor():
'''Check that an instance of KernelFunctor class can be created. Also
check that the symbol method works as expected.
'''
symbol = DataTypeSymbol("hello", StructureType())
klr = KernelFunctor(symbol)
assert klr._symbol == symbol
assert klr.symbol == symbol
assert klr._colour == "yellow"
assert klr._text_name == "KernelFunctor"
assert klr.parent is None
def test_aic_validate_child():
'''Check that the _validate_child method behaves as expected.'''
kernel_functor = KernelFunctor(DataTypeSymbol("dummy", REAL_TYPE))
assert AlgorithmInvokeCall._validate_child(0, kernel_functor)
assert not AlgorithmInvokeCall._validate_child(0, "Invalid")
routine = RoutineSymbol("hello")
call = AlgorithmInvokeCall(routine, 0)
with pytest.raises(GenerationError) as info:
call.children = ["invalid"]
assert ("Item 'str' can't be child 0 of 'AlgorithmInvokeCall'. The valid "
"format is: '[KernelFunctor]*'." in str(info.value))
call.children = [kernel_functor]
def test_kernelfunctor_create(cls):
'''Check that the create method of KernelFunctor works as expected.
'''
symbol = DataTypeSymbol("hello", StructureType())
klr = cls.create(symbol, [])
# pylint: disable=unidiomatic-typecheck
assert type(klr) is cls
assert klr._symbol == symbol
assert len(klr.children) == 0
arg = Reference(Symbol("dummy"))
klr = KernelFunctor.create(symbol, [arg])
assert len(klr.children) == 1
assert klr.children[0] == arg
assert arg.parent == klr
def test_aic_create():
'''Check that the create method behaves as expected.'''
kernel_functor = KernelFunctor(DataTypeSymbol("dummy", REAL_TYPE))
routine = RoutineSymbol("hello")
index = 10
aic = AlgorithmInvokeCall.create(routine, [kernel_functor], index)
assert isinstance(aic, AlgorithmInvokeCall)
assert len(aic.children) == 1
assert aic.children[0] is kernel_functor
assert aic._routine is routine
assert aic._index == index
with pytest.raises(GenerationError) as info:
AlgorithmInvokeCall.create(routine, kernel_functor, index)
assert ("AlgorithmInvokeCall create arguments argument should be a list "
"but found 'KernelFunctor'." in str(info.value))
def apply(self, call, index, options=None):
''' Apply the transformation to the supplied node.
:param call: a PSyIR call node capturing an invoke call in \
generic PSyIR.
:type call: :py:class:`psyclone.psyir.nodes.Call`
:param int index: the position of this invoke call relative to \
other invokes in the algorithm layer.
:param options: a dictionary with options for transformations.
:type options: dictionary of string:values or None
'''
self.validate(call, options=options)
kernel_calls = []
for call_arg in call.children:
arg_info = []
if isinstance(call_arg, ArrayReference):
# Structure constructor mis-parsed as an array
# reference.
type_symbol = call_arg.symbol
args = call_arg.pop_all_children()
arg_info.append((type_symbol, args))
else:
# CodeBlock containing a structure constructor
for fp2_node in call_arg._fp2_nodes:
type_symbol = self._get_symbol(call, fp2_node)
args = self._parse_args(call_arg, fp2_node)
arg_info.append((type_symbol, args))
for (type_symbol, args) in arg_info:
self._specialise_symbol(type_symbol)
kernel_calls.append(KernelFunctor.create(type_symbol, args))
invoke_call = AlgorithmInvokeCall.create(call.routine, kernel_calls,
index)
call.replace_with(invoke_call)
