def test_pdf_search_known_response(self):
self._vocabulary, self._query = set_up_pdf_test()
self._query.ask(answer="artilheiro brasil 1994 gols")
response = self._query.search(model=Probabilistic())
some_expected_results = [Result("./tests/static/pdf-files/d1.pdf", 5.811),
Result("./tests/static/pdf-files/d3.pdf", 5.811),
Result("./tests/static/pdf-files/d15.pdf", 4.358),
Result("./tests/static/pdf-files/d11.pdf", 3.353)]
for expected in some_expected_results:
self.assertTrue(expected in response)
def test_pdf_search_known_response(self):
self._vocabulary, self._query = set_up_pdf_test()
self._query.ask(answer="artilheiro brasil 1994 gols")
response = self._query.search(model=Vector(), idf=InverseFrequency(), tf=TermFrequency())
some_expected_results = [Result("./tests/static/pdf-files/d1.pdf", 1.0),
Result("./tests/static/pdf-files/d3.pdf", 0.808),
Result("./tests/static/pdf-files/d15.pdf", 0.74),
Result("./tests/static/pdf-files/d11.pdf", 0.604)]
for expected in some_expected_results:
self.assertTrue(expected in response)
def test_txt_search_known_response(self):
self._vocabulary, self._query = set_up_txt_test()
self._query.ask(answer="artilheiro brasil 1994 gols")
response = self._query.search(model=Boolean())
some_expected_results = [
Result("./tests/static/files/d1.txt", 1.0),
Result("./tests/static/files/d3.txt", 1.0),
Result("./tests/static/files/d15.txt", 0.75),
Result("./tests/static/files/d11.txt", 0.5)
]
for expected in some_expected_results:
self.assertTrue(expected in response)
def test_txt_search_known_response(self):
self._vocabulary, self._query = set_up_txt_test()
self._query.ask(answer="artilheiro brasil 1994 gols")
response = self._query.search(model=BeliefNetwork(),
idf=InverseFrequency(),
tf=TermFrequency())
some_expected_results = [
Result("./tests/static/files/d1.txt", 0.062),
Result("./tests/static/files/d3.txt", 0.051),
Result("./tests/static/files/d15.txt", 0.046),
Result("./tests/static/files/d11.txt", 0.038)
]
for expected in some_expected_results:
self.assertTrue(expected in response)
def test_txt_search_known_response(self):
self._vocabulary, self._query = set_up_txt_test()
self._query.ask(answer="artilheiro brasil 1994 gols")
response = self._query.search(model=ExtendedBoolean(3.0),
idf=InverseFrequency(),
tf=TermFrequency())
some_expected_results = [
Result("./tests/static/files/d1.txt", 0.564),
Result("./tests/static/files/d3.txt", 0.180),
Result("./tests/static/files/d15.txt", 0.308),
Result("./tests/static/files/d11.txt", 0.179)
]
for expected in some_expected_results:
self.assertTrue(expected in response)
def test_cast_solution(self):
ipt = [("d1.txt", 1.0), ("d2.txt", 0.421)]
opt = [Result("d1.txt", 1.0), Result("d2.txt", 0.421)]
self.assertEqual(Model.cast_solution(ipt), opt)