def __write_fixture(self): dbw = ScoreDBWriter(TEST_SCOREDB_PATH) dbw.set_scores(self.sample_scores) dbw.save_and_close()
def __write_scores(self, results): db = ScoreDBWriter(self.spec.scoredb_path) score_iter = ((doc.uri, doc.pagerank) for doc, _ in result_iterator(results) if isinstance(doc, Document)) db.set_scores(score_iter) db.save_and_close()