Python Literal.node_str示例

示例#1

def test_literal_node_str():
    ''' Check the node_str method of the Literal class.'''
    from psyclone.psyir.nodes.node import colored, SCHEDULE_COLOUR_MAP

    # scalar literal
    literal = Literal("1", INTEGER_SINGLE_TYPE)
    coloredtext = colored("Literal", SCHEDULE_COLOUR_MAP["Literal"])
    assert (coloredtext + "[value:'1', Scalar<INTEGER, SINGLE>]"
            in literal.node_str())

    # array literal
    array_type = ArrayType(REAL_DOUBLE_TYPE, [10, 10])
    literal = Literal("1", array_type)
    coloredtext = colored("Literal", SCHEDULE_COLOUR_MAP["Literal"])
    assert (coloredtext + "[value:'1', Array<Scalar<REAL, DOUBLE>, "
            "shape=[10, 10]>]" in literal.node_str())

示例#2

文件： literal_test.py 项目： stfc/PSyclone

def test_literal_node_str():
    ''' Check the node_str method of the Literal class.'''
    # scalar literal
    literal = Literal("1", INTEGER_SINGLE_TYPE)
    coloredtext = colored("Literal", Literal._colour)
    assert (coloredtext + "[value:'1', Scalar<INTEGER, SINGLE>]"
            in literal.node_str())

    # array literal
    array_type = ArrayType(REAL_DOUBLE_TYPE, [10, 10])
    literal = Literal("1", array_type)
    coloredtext = colored("Literal", Literal._colour)
    assert (coloredtext + "[value:'1', Array<Scalar<REAL, DOUBLE>, "
            "shape=[Literal[value:'10', Scalar<INTEGER, UNDEFINED>], "
            "Literal[value:'10', Scalar<INTEGER, UNDEFINED>]]>]"
            in literal.node_str())