| import unittest | |
| from .. import BM25TestCase | |
| class TestBM25PTQuick(BM25TestCase): | |
| def test_bm25_sparse_vs_rank_bm25_on_nfcorpus(self): | |
| self.compare_with_bm25_pt("nfcorpus") | |
| def test_bm25_sparse_vs_rank_bm25_on_scifact(self): | |
| self.compare_with_bm25_pt("scifact") | |
| def test_bm25_sparse_vs_rank_bm25_on_scidocs(self): | |
| self.compare_with_bm25_pt("scidocs") | |
| # fiqa | |
| def test_bm25_sparse_vs_rank_bm25_on_fiqa(self): | |
| self.compare_with_bm25_pt("fiqa") | |
| # arguana | |
| def test_bm25_sparse_vs_rank_bm25_on_arguana(self): | |
| self.compare_with_bm25_pt("arguana") | |
| if __name__ == '__main__': | |
| unittest.main() | |