fix Wadsworth's Constant edge case and add test
parent
29269c7c51
commit
a0597ba7a6
|
@ -0,0 +1,21 @@
|
||||||
|
"""
|
||||||
|
test_nsfw.py - some things just aren't safe for work, the test cases
|
||||||
|
author: mutantmonkey <mutantmonkey@mutantmonkey.in>
|
||||||
|
"""
|
||||||
|
|
||||||
|
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")
|
|
@ -1,13 +1,16 @@
|
||||||
#!/usr/bin/python3
|
#!/usr/bin/python3
|
||||||
"""
|
"""
|
||||||
wadsworth.py - Use Wadsworth's Constant on a string.
|
wadsworth.py - Apply Wadsworth's Constant to some text.
|
||||||
https://gist.github.com/1257195
|
https://gist.github.com/1257195
|
||||||
author: mutantmonkey <mutantmonkey@mutantmonkey.in>
|
author: mutantmonkey <mutantmonkey@mutantmonkey.in>
|
||||||
"""
|
"""
|
||||||
|
|
||||||
def wadsworth(phenny, input):
|
def wadsworth(phenny, input):
|
||||||
""".wadsworth - Use Wadsworth's Constant on a string."""
|
""".wadsworth - Apply Wadsworth's Constant to some text."""
|
||||||
text = input.group(2)
|
text = input.group(2)
|
||||||
|
if not text:
|
||||||
|
return phenny.say(".wadsworth <text> - apply Wadsworth's Constant")
|
||||||
|
|
||||||
text = text[text.find(' ', int(round(0.3 * len(text)))) + 1:]
|
text = text[text.find(' ', int(round(0.3 * len(text)))) + 1:]
|
||||||
phenny.say(text)
|
phenny.say(text)
|
||||||
wadsworth.commands = ['wadsworth']
|
wadsworth.commands = ['wadsworth']
|
||||||
|
|
Loading…
Reference in New Issue