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'