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)
