summaryrefslogtreecommitdiff
path: root/tests/unit/test_wrap.py
blob: 2e35951fe4183a52a6e5ed055d9e956158434c3f (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
from isort import wrap
from isort.settings import Config


def test_import_statement():
    assert wrap.import_statement("", [], []) == ""
    assert (
        wrap.import_statement("from x import ", ["y"], [], config=Config(balanced_wrapping=True))
        == "from x import (y)"
    )
    assert (
        wrap.import_statement("from long_import ", ["verylong"] * 10, [])
        == """from long_import (verylong, verylong, verylong, verylong, verylong, verylong,
                  verylong, verylong, verylong, verylong)"""
    )
    assert wrap.import_statement("from x import ", ["y", "z"], [], explode=True) == (
        "from x import (\n    y,\n    z,\n)"
    )