Example #1
0
 def test_untrain(self):
     sb = SimpleBayes()
     sb.train('foo', 'hello world hello')
     self.assertEqual(sb.tally('foo'), 3)
     self.assertEqual(sb.tally('bar'), 0)
     sb.untrain('bar', 'for bar baz')
     self.assertEqual(sb.tally('foo'), 3)
     self.assertEqual(sb.tally('bar'), 0)
     sb.untrain('foo', 'hello world')
     self.assertEqual(sb.tally('foo'), 1)
Example #2
0
 def test_untrain(self):
     sb = SimpleBayes()
     sb.train('foo', 'hello world hello')
     self.assertEqual(sb.tally('foo'), 3)
     self.assertEqual(sb.tally('bar'), 0)
     sb.untrain('bar', 'for bar baz')
     self.assertEqual(sb.tally('foo'), 3)
     self.assertEqual(sb.tally('bar'), 0)
     sb.untrain('foo', 'hello world')
     self.assertEqual(sb.tally('foo'), 1)