class TestSearchIntegration(unittest.TestCase):
def setUp(self):
# The current directory depends on if you're running inside an IDE or from command line.
curdir = os.getcwd()
if curdir.endswith('sparse'):
anserini_root = '../../../anserini'
pyserini_root = '../..'
else:
anserini_root = '../anserini'
pyserini_root = '.'
self.checker = SimpleSercherAnseriniMatchChecker(
anserini_root=anserini_root,
index=os.path.join(anserini_root, 'indexes/lucene-index.disk45'),
topics=os.path.join(pyserini_root, 'tools/topics-and-qrels/topics.robust04.txt'),
pyserini_topics='robust04',
qrels=os.path.join(pyserini_root, 'tools/topics-and-qrels/qrels.robust04.txt'),
eval_root=pyserini_root)
def test_bm25(self):
self.assertTrue(self.checker.run('robust04_bm25', '-bm25', '--bm25'))
def test_bm25_rm3(self):
self.assertTrue(self.checker.run('robust04_bm25_rm3', '-bm25 -rm3', '--bm25 --rm3'))
def test_qld(self):
self.assertTrue(self.checker.run('robust04_qld', '-qld', '--qld'))
def test_qld_rm3(self):
self.assertTrue(self.checker.run('robust04_qld_rm3', '-qld -rm3', '--qld --rm3'))
def tearDown(self):
pass
def setUp(self):
# The current directory depends on if you're running inside an IDE or from command line.
curdir = os.getcwd()
if curdir.endswith('sparse'):
anserini_root = '../../../anserini'
pyserini_root = '../..'
else:
anserini_root = '../anserini'
pyserini_root = '.'
self.checker = SimpleSercherAnseriniMatchChecker(
anserini_root=anserini_root,
index=os.path.join(anserini_root, 'indexes/lucene-index.disk45'),
topics=os.path.join(pyserini_root, 'tools/topics-and-qrels/topics.robust04.txt'),
pyserini_topics='robust04',
qrels=os.path.join(pyserini_root, 'tools/topics-and-qrels/qrels.robust04.txt'),
eval_root=pyserini_root)