tests/profanity_filter_test.py

   1 #!/usr/bin/env python3
   2
   3 import unittest
   4
   5 import profanity_filter as pf
   6 import unittest_utils
   7
   8
   9 class TestProfanityFilter(unittest.TestCase):
  10     def test_basic_functionality(self):
  11         p = pf.ProfanityFilter()
  12         self.assertTrue(p.is_bad_word('shit'))
  13         self.assertTrue(p.contains_bad_word('this is another fucking test'))
  14         self.assertTrue(p.contains_bad_word('this is another fuckin test'))
  15         self.assertFalse(
  16             p.contains_bad_word('Mary had a little lamb whose fleese was white as snow.')
  17         )
  18
  19
  20 if __name__ == '__main__':
  21     unittest.main()