def test_periods_come_before_underscores(self): imports = [make_import('a.b'), make_import('a_b')] expect(AlphabetizesImports.alphabetize(imports)) == [make_import('a.b'), make_import('a_b')]
def test_uppercase_comes_before_lowercase(self): imports = [make_import('b'), make_import('a'), make_import('C')] expect(AlphabetizesImports.alphabetize(imports)) == [make_import('C'), make_import('a'), make_import('b')]
def test_periods_come_before_after_letters(self): imports = [make_import('ab'), make_import('a.c'), make_import('a.b')] expect(AlphabetizesImports.alphabetize(imports)) == [make_import('a.b'), make_import('a.c'), make_import('ab')]
def test_writes_imports_with_as_names(self): imports = [make_import('module', asname='asmodule'), make_import('module', 'name', 'asname')] expected = ('import module as asmodule\n' 'from module import name as asname') expect(WritesImports.write(imports)) == expected
def test_writes_imports(self): imports = [make_import('b'), make_import('a', 'name')] expect(WritesImports.write(imports)) == 'import b\nfrom a import name'
def test_extracts_from_as_imports(self): imports = ExtractsImports.extract('from mod import name as other_name') expect(imports) == [make_import('mod', 'name', 'other_name')]
def test_extracts_module_as_imports(self): imports = ExtractsImports.extract('import name as other_name') expect(imports) == [make_import('name', None, 'other_name')]
def test_finds_imports(self): imports = ExtractsImports.extract('import name\n' 'from module import name') expect(imports) == [make_import('name', None), make_import('module', 'name')]