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