def test_compute_pagerank(self):
graph = WikidataGraph()
graph.load_from_matrix(
os.path.join(self.testdir, 'data/sample_wikidata_items.npz'))
graph.compute_pagerank()
self.assertTrue(
graph.get_pagerank('Q45') > 0.0003
and graph.get_pagerank('Q45') < 0.0004)
def compute_pagerank(filename, outfile):
"""
Computes the pagerank of a Wikidata adjacency matrix as represented by a Numpy sparse matrix in NPZ format.
"""
if outfile is None:
outfile = '.'.join(filename.split('.')[:-1] + ['pgrank.npy'])
g = WikidataGraph()
g.load_from_matrix(filename)
g.compute_pagerank()
g.save_pagerank(outfile)