#!/usr/bin/env python3

import unittest

import profanity_filter as pf
import unittest_utils


class TestProfanityFilter(unittest.TestCase):

    def test_basic_functionality(self):
        p = pf.ProfanityFilter()
        self.assertTrue(p.is_bad_word('shit'))
        self.assertTrue(p.contains_bad_word('this is another fucking test'))
        self.assertTrue(p.contains_bad_word('this is another fuckin test'))
        self.assertFalse(p.contains_bad_word('Mary had a little lamb whose fleese was white as snow.'))


if __name__ == '__main__':
    unittest.main()