""" test_nsfw.py - some things just aren't safe for work, the test cases author: mutantmonkey """ import re import unittest from mock import MagicMock, Mock from modules.wadsworth import wadsworth class TestWadsworth(unittest.TestCase): def setUp(self): self.phenny = MagicMock() def test_wadsworth(self): input = Mock(group=lambda x: "Apply Wadsworth's Constant to a string") wadsworth(self.phenny, input) self.phenny.say.assert_called_once_with( "Constant to a string")