def show_scores(program, spec):
"""Usage: <spec>
Shows the scores of documents in the specified `spec` ScoreDB file.
"""
from freequery.index.scoredb import ScoreDB
from freequery.client.client import Spec
spec = Spec(spec)
scoredb = ScoreDB(spec.scoredb_path)
for uri, score in scoredb.items():
print "%.8f\t%s" % (score, uri)
class TestScoreDB(unittest.TestCase):
sample_scores = {
"http://example.com/": 0.05,
"http://example.com/a.html": 0.04,
"http://stanford.edu/": 0.02,
"http://zzz.com/": 0.02,
}
def __write_fixture(self):
dbw = ScoreDBWriter(TEST_SCOREDB_PATH)
dbw.set_scores(self.sample_scores)
dbw.save_and_close()
def setUp(self):
self.__write_fixture()
self.scoredb = ScoreDB(TEST_SCOREDB_PATH)
def test_reads_writes(self):
self.assertAlmostEqual(0.05, self.scoredb.get_one("http://example.com/"))
self.assertAlmostEqual(0.04, self.scoredb.get_one("http://example.com/a.html"))
self.assertAlmostEqual(0.02, self.scoredb.get_one("http://stanford.edu/"))
self.assertAlmostEqual(0.02, self.scoredb.get_one("http://zzz.com/"))
def test_iterates(self):
self.assertEquals(sorted(self.sample_scores.items()), sorted(self.scoredb.items()))
def test_rank(self):
self.assertEquals(
[
("http://example.com/", 0.05),
("http://example.com/a.html", 0.04),
("http://zzz.com/", 0.02),
("http://stanford.edu/", 0.02),
],
self.scoredb.rank(),
)
self.assertEquals(
[("http://example.com/", 0.05), ("http://stanford.edu/", 0.02)],
self.scoredb.rank(["http://stanford.edu/", "http://example.com/"]),
)
