| import unittest | |
| from .. import BM25TestCase | |
| class TestRankBM25Quick(BM25TestCase): | |
| def test_bm25_sparse_vs_rank_bm25_on_nfcorpus(self): | |
| self.compare_with_rank_bm25("nfcorpus", corpus_subsample=2000, queries_subsample=200) | |
| def test_bm25_sparse_vs_rank_bm25_on_scifact(self): | |
| self.compare_with_rank_bm25("scifact", corpus_subsample=2000, queries_subsample=200) | |
| def test_bm25_sparse_vs_rank_bm25_on_scidocs(self): | |
| self.compare_with_rank_bm25("scidocs", corpus_subsample=2000, queries_subsample=200) | |
| # fiqa | |
| def test_bm25_sparse_vs_rank_bm25_on_fiqa(self): | |
| self.compare_with_rank_bm25("fiqa", corpus_subsample=2000, queries_subsample=200) | |
| # arguana | |
| def test_bm25_sparse_vs_rank_bm25_on_arguana(self): | |
| self.compare_with_rank_bm25("arguana", queries_subsample=100, corpus_subsample=1000) | |
| if __name__ == '__main__': | |
| unittest.main() | |