def test_dependency_tree(fake_dist, installed, root, expected): installed = {distribution.key: distribution for distribution in (fake_dist(name, deps) for name, deps in installed)} actual = dependency_tree(installed, root) assert actual == set(expected)