def test_set_date(self):
now = time.time()
self.s.ResetTotal(permanently=True)
self.assertEqual(now, self.s.from_date)
for stat in [
"num_ham",
"num_spam",
"num_unsure",
"num_trained_spam",
"num_trained_spam_fn",
"num_trained_ham",
"num_trained_ham_fp",
]:
self.assertEqual(self.s.totals[stat], 0)
# Check that it was stored, too.
self.messageinfo_db.close()
self.messageinfo_db = MessageInfoPickle(self.messageinfo_db_name)
self.s = Stats(options, self.messageinfo_db)
self.assertEqual(now, self.s.from_date)
def setUp(self):
self.messageinfo_db_name = "__unittest.pik"
self.messageinfo_db = MessageInfoPickle(self.messageinfo_db_name)
self.s = Stats(options, self.messageinfo_db)
Message.message_info_db = self.messageinfo_db
