Python IfBlock.node_str Examples

Example #1

def test_ifblock_node_str():
    ''' Check the node_str method of the IfBlock class.'''
    colouredif = colored("If", IfBlock._colour)

    ifblock = IfBlock()
    output = ifblock.node_str()
    assert colouredif+"[]" in output

    ifblock = IfBlock(annotations=['was_elseif'])
    output = ifblock.node_str()
    assert colouredif+"[annotations='was_elseif']" in output

Example #2

File: ifblock_test.py Project: xyuan/PSyclone

def test_ifblock_node_str():
    ''' Check the node_str method of the IfBlock class.'''
    from psyclone.psyir.nodes.node import colored, SCHEDULE_COLOUR_MAP
    colouredif = colored("If", SCHEDULE_COLOUR_MAP["If"])

    ifblock = IfBlock()
    output = ifblock.node_str()
    assert colouredif + "[]" in output

    ifblock = IfBlock(annotations=['was_elseif'])
    output = ifblock.node_str()
    assert colouredif + "[annotations='was_elseif']" in output