Example #1
0
    # Tests for title + corroborating detail
    'true grit john wayne',
    'true grit 2010',
    'true grit jeff bridges',
    'true grit coens brothers',
    'superman reeves',
    'superman brandon routh',
    'superman kirk alyn',
    'superman ii',
    'superman iii',
    'superman kevin spacey',
]

def summarize_proxy(proxy):
    return proxy.name, proxy.release_date

def summarize_cluster(cluster):
    return [summarize_proxy(result.resolverObject) for result in cluster.results]

if __name__ == '__main__':
    searcher = EntitySearch()
    samples = []
    for query in FILM_QUERIES:
        entitiesAndClusters = searcher.searchEntitiesAndClusters('film', query)
        samples.append([summarize_cluster(cluster) for _, cluster in entitiesAndClusters])

    print 
    for s in samples:
        print s