def test_get_precission_selection_indices_1(self):
ranked_to_category = [('a',1),('b',0.9),('c',0.8),('d',0.7),('e',0.6),('f',0.5)]
documents_in_category=['a','b','f']
precission_levels=[0.1, 0.5, 0.6, 1.0]
expected = {0:6,1:6,2:3,3:2}
res = evaluater.get_precission_selection_indices(ranked_to_category, documents_in_category, precission_levels)
self.assertEqual(res, expected)
def test_get_precission_selection_indices_3(self):
ranked_to_category = []
documents_in_category=['f']
precission_levels=[0.1,1.0]
expected = {0:0,1:0}
res = evaluater.get_precission_selection_indices(ranked_to_category, documents_in_category, precission_levels)
self.assertEqual(res, expected)
