def test_import_multi_line():
content = '''import alonglonglonglonglonglonglonglonglonglong, alonglonglonglonglonglonglonglonglonglonglong2'''
fixed_source = get_fixed_source(content)
fixed = '''import alonglonglonglonglonglonglonglonglonglong\nimport alonglonglonglonglonglonglonglonglonglonglong2'''
assert fixed_source == fixed
content = '''import alonglonglonglonglonglonglonglonglonglong, alonglonglonglonglonglonglonglonglonglonglong, \
alonglonglonglonglong'''
fixed_source = get_fixed_source(content)
fixed = '''import alonglonglonglonglonglonglonglonglonglong\nimport alonglonglonglonglonglonglonglonglonglonglong\nimport alonglonglonglonglong'''
assert fixed_source == fixed
def test_from_import_multi_line():
content = '''from a import osdfsffsfsfdffsfs, syssdfsdfsdfsdfsdfsfsf, sffdfsdfsdfsdfsddfsdfsdfsdfsddddsddssddssf, \
rhdfgkaa'''
fixed_source = get_fixed_source(content)
fixed = '''from a import (osdfsffsfsfdffsfs, syssdfsdfsdfsdfsdfsfsf,
sffdfsdfsdfsdfsddfsdfsdfsdfsddddsddssddssf, rhdfgkaa)'''
assert fixed_source == fixed
def test_from_import_with_backslash():
content = '''import alonglonglonglonglonglonglonglonglonglonglong, alonglonglonglonglonglonglonglonglonglonglong2
from sdfsfdd import osdfsffsfsfdffsfs, syssdfsdfsdfsdfsdfsfsf,\
sdsddsddsd, sdfsdfsdfdsfsdfgsfgsdjfdsgafd, sdfsdfffffffffffffffffffffffffffffffffffffffffffffffffffffsdfgshdgfjhsdgf, \
dffff
from sdsdddfsfdd import (osdfsffsfsfdffsfs,
sdfsdfsdfsdsfgksflsfdljlsdjflsljfhlsdhfklshflshflsfhlshflsffff)
from sdfgsggsdfdsaf import osdfsffsfsfdffsfs, sdfsdfsdfsdfsdfhlhlhlhlhlzfd, sdfsdf, \
sdfsdfsdfsfsdfsfd, sdfsdfsdfsdfsdf
import sdfsdff
from sadfsadr import sdfsdff, sdfsdfsdf, asdfsdfsdaadadsd, sdfsgwerwerr, opsoidpfp, sadfsdfdsff
def t():
pass'''
fixed_source = get_fixed_source(content)
fixed = '''import alonglonglonglonglonglonglonglonglonglonglong
import alonglonglonglonglonglonglonglonglonglonglong2
from sdfsfdd import (osdfsffsfsfdffsfs, syssdfsdfsdfsdfsdfsfsf, sdsddsddsd, # noqa
sdfsdfsdfdsfsdfgsfgsdjfdsgafd,
sdfsdfffffffffffffffffffffffffffffffffffffffffffffffffffffsdfgshdgfjhsdgf,
dffff)
from sdsdddfsfdd import (osdfsffsfsfdffsfs, # noqa
sdfsdfsdfsdsfgksflsfdljlsdjflsljfhlsdhfklshflshflsfhlshflsfff)
from sdfgsggsdfdsaf import (osdfsffsfsfdffsfs, sdfsdfsdfsdfsdfhlhlhlhlhlzfd,
sdfsdf, sdfsdfsdfsfsdfsfd, sdfsdfsdfsdfsdf)
import sdfsdff
from sadfsadr import (sdfsdff, sdfsdfsdf, asdfsdfsdaadadsd, sdfsgwerwerr,
opsoidpfp, sadfsdfdsff)
def t():
pass'''
assert fixed_source == fixed
def test_from_import():
content = '''import longlonglonglonglonglonglonglonglong, alonglonglonglonglonglonglonglonglonglonglong2
from sdfsfdd import (osdfsffsfsfdffsfs, syssdfsdfsdfsdfsdfsfsf,
sdsddsddsd, sdfsdfsdfdsfsdfgsfgsdjfdsgafd, sdfsdfffffffffffffffffffffffffffffffffffffffffffffffffffffsdfgshdgfjhsdgf,
dffff)
# from dfdff import asd
# sdfsf
import dff
import sdfsdfsdfjkljlkjlsjdlfjsf, sdfskdfhoweuriwrtitaidgiadgiadadsdjfbskdfgksfg
from ttsdsdddfsfdd import (osdfsffsfsfdffsfs,
sdfsdfsdfsdsfgksflsfdljlsdjflsljfhlsdhfklshflshflsfhlshflsffff)
from sdsdfhlsdfhlhlkhl import sdfsdfsadlk, adfiuwteiudtqiet,asdggiiigigiigsdfjskdgkj, sdfsdf
def t():
pass'''
fixed_source = get_fixed_source(content)
fixed = '''import longlonglonglonglonglonglonglonglong
import alonglonglonglonglonglonglonglonglonglonglong2
from sdfsfdd import (osdfsffsfsfdffsfs, syssdfsdfsdfsdfsdfsfsf, sdsddsddsd, # noqa
sdfsdfsdfdsfsdfgsfgsdjfdsgafd,
sdfsdfffffffffffffffffffffffffffffffffffffffffffffffffffffsdfgshdgfjhsdgf,
dffff)
# from dfdff import asd
# sdfsf
import dff
import sdfsdfsdfjkljlkjlsjdlfjsf
import sdfskdfhoweuriwrtitaidgiadgiadadsdjfbskdfgksfg
from ttsdsdddfsfdd import (osdfsffsfsfdffsfs, # noqa
sdfsdfsdfsdsfgksflsfdljlsdjflsljfhlsdhfklshflshflsfhlshflsfff)
from sdsdfhlsdfhlhlkhl import (sdfsdfsadlk, adfiuwteiudtqiet,
asdggiiigigiigsdfjskdgkj, sdfsdf)
def t():
pass'''
assert fixed_source == fixed
def test_import_noqa():
content = '''import alonglonglonglonglonglonglonglonglonglonglonglonglonglonglonglonglonglong'''
fixed_source = get_fixed_source(content)
fixed = '''import alonglonglonglonglonglonglonglonglonglonglonglonglonglonglonglonglonglong # noqa'''
assert fixed_source == fixed
