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)""" | |
) |