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