File size: 680 Bytes
74b1bac |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 |
import unittest
from .. import BM25TestCase
class TestRankBM25Full(BM25TestCase):
def test_bm25_sparse_vs_rank_bm25_on_nfcorpus(self):
self.compare_with_rank_bm25("nfcorpus")
def test_bm25_sparse_vs_rank_bm25_on_scifact(self):
self.compare_with_rank_bm25("scifact")
def test_bm25_sparse_vs_rank_bm25_on_scidocs(self):
self.compare_with_rank_bm25("scidocs")
# fiqa
def test_bm25_sparse_vs_rank_bm25_on_fiqa(self):
self.compare_with_rank_bm25("fiqa")
# arguana
def test_bm25_sparse_vs_rank_bm25_on_arguana(self):
self.compare_with_rank_bm25("arguana")
if __name__ == '__main__':
unittest.main()
|