Python ReferenceFinderの例

コード例 #1

ファイル: test_finder.py プロジェクト: Bogdanp/modviz

def test_find_external_from_import():
    modules = [Module("/", "foo/bar.py")]
    source = "from baz import bazz"
    finder = ReferenceFinder(modules, modules[0])
    finder.visit(ast.parse(source))
    assert finder.ext_references == set([
        Module("/", "baz.py"),
        Module("/", "baz/bazz.py")
    ])

コード例 #2

ファイル: test_finder.py プロジェクト: Bogdanp/modviz

def test_find_simple_from():
    modules = [Module("/", "foo/bar.py"), Module("/", "baz/__init__.py")]
    source = "from baz import Baz"
    finder = ReferenceFinder(modules, modules[0])
    finder.visit(ast.parse(source))
    assert finder.references == set([Module("/", "baz.py")])

コード例 #3

ファイル: test_finder.py プロジェクト: Bogdanp/modviz

def test_package_relative_from():
    modules = [Module("/", "foo/bar.py"), Module("/", "foo/baz.py")]
    source = "from . import bar"
    finder = ReferenceFinder(modules, modules[1])
    finder.visit(ast.parse(source))
    assert finder.references == set([Module("/", "foo/bar.py")])