Example #1
0
def test_increase_indentation():
    red = RedBaron(test_indent_code)
    red.increase_indentation(4)
    indented_code = "\n" + "\n".join(
        map(lambda x: "    " + x,
            test_indent_code.split("\n")[1:-2])) + "\n\n"
    assert red.dumps() == indented_code
Example #2
0
def test_increase_indentation_brackets_2():
    code = """
a = (fun()
  .filter())
"""
    red = RedBaron(code)
    red.increase_indentation("  ")
    assert red.dumps() == indent_str(code, "  ")
Example #3
0
def test_increase_indentation_assignement_backslash():
    code = """
    a = \\
        1
"""
    red = RedBaron(code)
    red.increase_indentation("  ")
    assert red.dumps() == indent_str(code, "  ")
Example #4
0
def test_increase_indentation():
    red = RedBaron(test_indent_code)
    red.increase_indentation("    ")
    assert red.dumps() == indent_str(test_indent_code, "    ")
Example #5
0
def test_increase_indentation():
    red = RedBaron(test_indent_code)
    red.increase_indentation(4)
    indented_code = "\n" + "\n".join(map(lambda x: "    " + x, test_indent_code.split("\n")[1:-2])) + "\n\n"
    assert red.dumps() == indented_code