示例#1
0
def test_decrease_indentation():
    red = RedBaron(test_indent_code)
    red.decrease_indentation(4)
    indented_code = "\ndef a():\n" + "\n".join(
        map(lambda x: x[4:],
            test_indent_code.split("\n")[2:-2])) + "\n\n"
    assert red.dumps() == indented_code
示例#2
0
def test_decrease_indentation_brackets_2():
    code = """
  a = (fun()
    .filter())
"""
    red = RedBaron(code)
    red.decrease_indentation("  ")
    assert red.dumps() == deindent_str(code, "  ")
示例#3
0
def test_decrease_indentation_assignement_backslash():
    code = """
    a = \\
        1
"""
    red = RedBaron(code)
    red.decrease_indentation("  ")
    assert red.dumps() == deindent_str(code, "  ")
示例#4
0
def test_decrease_indentation_assignement_and_atom_node_2():
    code = """
    a = fun(). \\
            more()
"""
    red = RedBaron(code)
    red.decrease_indentation("  ")
    assert red.dumps() == deindent_str(code, "  ")
示例#5
0
def test_decrease_indentation_atom_node():
    code = """
    fun() \\
        .more()
"""
    red = RedBaron(code)
    red.decrease_indentation("  ")
    assert red.dumps() == deindent_str(code, "  ")
示例#6
0
def test_decrease_indentation_for_else():
    code = """
    for a in b:
        pass
    else:
        pass
"""
    red = RedBaron(code)
    red.decrease_indentation("  ")
    assert red.dumps() == deindent_str(code, "  ")
示例#7
0
def test_decrease_indentation_try_finally():
    code = """
    try:
        pass
    finally:
        pass
"""
    red = RedBaron(code)
    red.decrease_indentation("  ")
    assert red.dumps() == deindent_str(code, "  ")
示例#8
0
def test_decrease_indentation_while_else():
    code = """
    while cond:
        pass
    else:
        pass
"""
    red = RedBaron(code)
    red.decrease_indentation("  ")
    assert red.dumps() == deindent_str(code, "  ")
示例#9
0
def test_decrease_indentation_try_except_else():
    code = """
    try:
        pass
    except:
        pass
    else:
        pass
"""
    red = RedBaron(code)
    red.decrease_indentation("  ")
    assert red.dumps() == deindent_str(code, "  ")
示例#10
0
def test_decrease_indentation():
    red = RedBaron(test_indent_code)
    red.decrease_indentation("  ")
    assert red.dumps() == deindent_str(test_indent_code, "  ")
示例#11
0
def test_decrease_indentation():
    red = RedBaron(test_indent_code)
    red.decrease_indentation(4)
    indented_code = "\ndef a():\n" + "\n".join(map(lambda x: x[4:], test_indent_code.split("\n")[2:-2])) + "\n\n"
    assert red.dumps() == indented_code