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_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_finds_imports(self):
imports = ExtractsImports.extract('import name\n'
'from module import name')
expect(imports) == [make_import('name', None),
make_import('module', 'name')]
