class ArticleAucPrEvaluationTest(CalculationAssertions): @patch('biocreative.evaluation.calculation.hits.Hits', spec=True) def setUp(self, unused): self.evaluator = ArticleAucPrEvaluation() self.evaluator.store_p_at_current_r = Mock() self.std_item = None for attr in C.HITS_ATTRIBUTES: setattr(self.evaluator.hits, attr, 2) def test_evaluate_with_true_item(self): self.evaluate_with_std_item(True) self.assert_hits(self.evaluator.hits, tp=3, tn=2, fp=2, fn=1) def test_evaluate_with_false_item(self): self.evaluate_with_std_item(False) self.assert_hits(self.evaluator.hits, tp=2, tn=2, fp=3, fn=2) def test_evaluate_with_illegal_item(self): self.assertRaises(TypeError, self.evaluator.evaluate, None, 1, None) def evaluate_with_std_item(self, flag): self.evaluator.evaluate(None, flag, None) self.assert_(self.evaluator.store_p_at_current_r.called)
def setUp(self, unused): self.evaluator = ArticleAucPrEvaluation() self.evaluator.store_p_at_current_r = Mock() self.std_item = None for attr in C.HITS_ATTRIBUTES: setattr(self.evaluator.hits, attr, 2)