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