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)