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'
