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)