def test_visit_import(self, pyfile): node = ast.Import(names=[ast.alias(name='Foo', asname='Bar')]) pyfile.visit_import(node) assert pyfile.ast_imported == {'Bar': 'Foo'}