def setUp(self):
self.msg = email.message_from_string(spam1, _class=SBHeaderMessage)
c = Classifier()
self.u_prob, clues = c.spamprob(tokenize(good1), True)
c.learn(tokenize(good1), False)
self.g_prob, clues = c.spamprob(tokenize(good1), True)
c.unlearn(tokenize(good1), False)
c.learn(tokenize(spam1), True)
self.s_prob, self.clues = c.spamprob(tokenize(spam1), True)
self.ham = options['Headers','header_ham_string']
self.spam = options['Headers','header_spam_string']
self.unsure = options['Headers','header_unsure_string']
self.to = "[email protected];[email protected]"
self.msg["to"] = self.to
def setUp(self):
self.msg = email.message_from_string(spam1, _class=SBHeaderMessage)
# Get a prob and some clues.
c = Classifier()
self.u_prob, clues = c.spamprob(tokenize(good1), True)
c.learn(tokenize(good1), False)
self.g_prob, clues = c.spamprob(tokenize(good1), True)
c.unlearn(tokenize(good1), False)
c.learn(tokenize(spam1), True)
self.s_prob, self.clues = c.spamprob(tokenize(spam1), True)
self.ham = options['Headers', 'header_ham_string']
self.spam = options['Headers', 'header_spam_string']
self.unsure = options['Headers', 'header_unsure_string']
self.to = "[email protected];[email protected]"
self.msg["to"] = self.to
