def test_spam_probability(self): training_set = [("foo bar", 1), ("bar baz", 0)] counts = nb.count_words(training_set) word_probs = nb.word_probabilities(counts, 1, 1) message = "there is no foo bar like bar bar" actual = nb.spam_probability(word_probs, message) expected = 0.9 self.assertAlmostEqual(actual, expected)
def classify(self, message): return nb.spam_probability(self.word_probs, message)