def test_binaryoperation_initialization():
''' Check the initialization method of the BinaryOperation class works
as expected.'''
with pytest.raises(TypeError) as err:
_ = BinaryOperation("not an operator")
assert "BinaryOperation operator argument must be of type " \
"BinaryOperation.Operator but found" in str(err.value)
bop = BinaryOperation(BinaryOperation.Operator.ADD)
assert bop._operator is BinaryOperation.Operator.ADD
def test_binaryoperation_operator():
'''Test that the operator property returns the binaryoperator in the
binaryoperation.
'''
binary_operation = BinaryOperation(BinaryOperation.Operator.ADD)
assert binary_operation.operator == BinaryOperation.Operator.ADD
def test_binaryoperation_node_str():
''' Check the node_str method of the Binary Operation class.'''
binary_operation = BinaryOperation(BinaryOperation.Operator.ADD)
op1 = Literal("1", INTEGER_SINGLE_TYPE)
op2 = Literal("1", INTEGER_SINGLE_TYPE)
binary_operation.addchild(op1)
binary_operation.addchild(op2)
coloredtext = colored("BinaryOperation", BinaryOperation._colour)
assert coloredtext+"[operator:'ADD']" in binary_operation.node_str()
def test_binaryoperation_node_str():
''' Check the node_str method of the Binary Operation class.'''
from psyclone.psyir.nodes.node import colored, SCHEDULE_COLOUR_MAP
binary_operation = BinaryOperation(BinaryOperation.Operator.ADD)
op1 = Literal("1", INTEGER_SINGLE_TYPE, parent=binary_operation)
op2 = Literal("1", INTEGER_SINGLE_TYPE, parent=binary_operation)
binary_operation.addchild(op1)
binary_operation.addchild(op2)
coloredtext = colored("BinaryOperation", SCHEDULE_COLOUR_MAP["Operation"])
assert coloredtext + "[operator:'ADD']" in binary_operation.node_str()
def test_cw_binaryoperator():
'''Check the CWriter class binary_operation method correctly
prints out the C representation of any given BinaryOperation.
'''
cwriter = CWriter()
# Test UnaryOperation without children.
binary_operation = BinaryOperation(BinaryOperation.Operator.ADD)
with pytest.raises(VisitorError) as err:
_ = cwriter(binary_operation)
assert ("BinaryOperation malformed or incomplete. It should have "
"exactly 2 children, but found 0." in str(err.value))
# Test with children
ref1 = Reference(DataSymbol("a", REAL_TYPE))
ref2 = Reference(DataSymbol("b", REAL_TYPE))
binary_operation = BinaryOperation.create(BinaryOperation.Operator.ADD,
ref1, ref2)
assert cwriter(binary_operation) == '(a + b)'
# Test all supported Operators
test_list = ((BinaryOperation.Operator.ADD,
'(a + b)'), (BinaryOperation.Operator.SUB, '(a - b)'),
(BinaryOperation.Operator.MUL,
'(a * b)'), (BinaryOperation.Operator.DIV, '(a / b)'),
(BinaryOperation.Operator.REM,
'(a % b)'), (BinaryOperation.Operator.POW, 'pow(a, b)'),
(BinaryOperation.Operator.EQ,
'(a == b)'), (BinaryOperation.Operator.NE, '(a != b)'),
(BinaryOperation.Operator.GT,
'(a > b)'), (BinaryOperation.Operator.GE, '(a >= b)'),
(BinaryOperation.Operator.LT,
'(a < b)'), (BinaryOperation.Operator.LE, '(a <= b)'),
(BinaryOperation.Operator.AND,
'(a && b)'), (BinaryOperation.Operator.OR,
'(a || b)'), (BinaryOperation.Operator.SIGN,
'copysign(a, b)'))
for operator, expected in test_list:
binary_operation._operator = operator
assert cwriter(binary_operation) == expected
# Test that an unsupported operator raises a error
class Unsupported(object):
'''Dummy class'''
def __init__(self):
pass
binary_operation._operator = Unsupported
with pytest.raises(VisitorError) as err:
_ = cwriter(binary_operation)
assert "The C backend does not support the '" in str(err.value)
assert "' operator." in str(err.value)
def test_binaryoperation_can_be_printed():
'''Test that a Binary Operation instance can always be printed (i.e. is
initialised fully)'''
binary_operation = BinaryOperation(BinaryOperation.Operator.ADD)
assert "BinaryOperation[operator:'ADD']" in str(binary_operation)
op1 = Literal("1", INTEGER_SINGLE_TYPE)
op2 = Literal("2", INTEGER_SINGLE_TYPE)
binary_operation.addchild(op1)
binary_operation.addchild(op2)
# Check the node children are also printed
assert ("Literal[value:'1', Scalar<INTEGER, SINGLE>]\n"
in str(binary_operation))
assert ("Literal[value:'2', Scalar<INTEGER, SINGLE>]"
in str(binary_operation))
def gen_code(self, parent):
if self.field_space == "every":
from psyclone.f2pygen import DeclGen
from psyclone.psyir.nodes import BinaryOperation
dim_var = DeclGen(parent,
datatype="INTEGER",
entity_decls=[self.variable.name])
parent.add(dim_var)
# Update start loop bound
self.start_expr = Literal("1", INTEGER_TYPE, parent=self)
# Update stop loop bound
if self._loop_type == "inner":
index = "1"
elif self._loop_type == "outer":
index = "2"
self.stop_expr = BinaryOperation(BinaryOperation.Operator.SIZE,
parent=self)
self.stop_expr.addchild(
Reference(DataSymbol(self.field_name, INTEGER_TYPE),
parent=self.stop_expr))
self.stop_expr.addchild(
Literal(index, INTEGER_TYPE, parent=self.stop_expr))
else: # one of our spaces so use values provided by the infrastructure
# loop bounds
# TODO: Issue 440. Implement derive types in PSyIR
if self._loop_type == "inner":
self.start_expr = Reference(self.field_space + "%istart",
parent=self)
self.stop_expr = Reference(self.field_space + "%istop",
parent=self)
elif self._loop_type == "outer":
self.start_expr = Reference(self.field_space + "%jstart",
parent=self)
self.stop_expr = Reference(self.field_space + "%jstop",
parent=self)
Loop.gen_code(self, parent)
def test_binaryoperation_children_validation():
'''Test that children added to BinaryOperation are validated.
BinaryOperations accept 2 DataNodes as children.
'''
operation = BinaryOperation(BinaryOperation.Operator.ADD)
literal1 = Literal("1", INTEGER_SINGLE_TYPE)
literal2 = Literal("2", INTEGER_SINGLE_TYPE)
literal3 = Literal("3", INTEGER_SINGLE_TYPE)
statement = Return()
# Statements are not valid
with pytest.raises(GenerationError) as excinfo:
operation.addchild(statement)
assert ("Item 'Return' can't be child 0 of 'BinaryOperation'. The valid "
"format is: 'DataNode, DataNode'.") in str(excinfo.value)
# First DataNodes is valid, but not subsequent ones
operation.addchild(literal1)
operation.addchild(literal2)
with pytest.raises(GenerationError) as excinfo:
operation.addchild(literal3)
assert ("Item 'Literal' can't be child 2 of 'BinaryOperation'. The valid "
"format is: 'DataNode, DataNode'.") in str(excinfo.value)