def test_extract_from_imports(self):
for stmt, modules in FROM_IMPORT_STATEMENTS:
imports = extract_imports(stmt)
self.assertEqual(imports, modules)
def test_extract_relative_imports_from_file(self):
from test.testfiles import relative_imports
with open('testfiles/relative_imports.py', 'rb') as f:
source = f.read()
imports = extract_imports(source)
self.assertEqual(imports, relative_imports.EXPECTED)
def test_extract_simple_imports(self):
for stmt, modules in SIMPLE_IMPORT_STATEMENTS:
imports = extract_imports(stmt)
self.assertEqual(imports, modules)
