test_actor.py 文件源码

def test_unshelve_similarity_graph(data_root):
    with shelve.open(str(data_root / 'test-similarity_graph.shelve')) as db:
        graph = db['similarity_graph']
    assert graph is not None
    print()
    actor_groups = graph.group_similar_actors()
    pprint(actor_groups)