def test_body_spam_ham_prob_Returns_expected_probability( self, mock_load_vocab): """ Il faut mocker la fonction "load_dict" Il faut vérifier que probabilité est calculée correctement donné le "body" à l'entrée (ces probabilites devron etre calcule selon l'enonce dans le TP1 ) """ email_analyzer = EmailAnalyzer() mock_load_vocab.return_value = self.vocab self.assertEqual(email_analyzer.body_spam_ham_prob(self.body_true), self.body_spam_ham_prob_expected)
def test_is_spam_Returns_False_if_spam_prob_is_lower( self, mock_subject_spam_ham_prob, mock_body_spam_ham_prob, mock_clean_text): """ Il faut mocker les fonctions "spam_ham_body_prob" et "subject_spam_ham_prob". La sortie de la fonction doit être False si probabilité spam probabilité ham (ces probabilites devron etre calcule selon l'enonce dans le TP1 ) """ mock_clean_text.return_value = self.clean_subject_false mock_subject_spam_ham_prob.return_value = self.subject_spam_ham_prob_false mock_clean_text.return_value = self.clean_body_false mock_body_spam_ham_prob.return_value = self.body_spam_ham_prob_false email_analyzer = EmailAnalyzer() self.assertEqual(email_analyzer.body_spam_ham_prob(self.body_false), self.body_spam_ham_prob_false) self.assertEqual( email_analyzer.subject_spam_ham_prob(self.subject_false), self.subject_spam_ham_prob_false) self.assertEqual( email_analyzer.is_spam(self.subject_false, self.body_false), False)