diff --git a/modules/src/__init__.py b/modules/src/__init__.py index e47d4310..526dca5d 100644 --- a/modules/src/__init__.py +++ b/modules/src/__init__.py @@ -2,6 +2,7 @@ 'anime', 'book', 'bye', + 'coin', 'currency', 'dictionary', 'fact', diff --git a/modules/src/coin.py b/modules/src/coin.py new file mode 100644 index 00000000..cdf83dc9 --- /dev/null +++ b/modules/src/coin.py @@ -0,0 +1,12 @@ +import random +from templates.text import TextTemplate + +def process(input, entities): + output = {} + output['input'] = input + if random.randrange(0,2,1): + output['output'] = TextTemplate("Sure, it's heads!").get_message() + else: + output['output'] = TextTemplate("Sure, it's tails!").get_message() + output['success'] = True + return output diff --git a/modules/tests/test_coin.py b/modules/tests/test_coin.py new file mode 100644 index 00000000..186502fc --- /dev/null +++ b/modules/tests/test_coin.py @@ -0,0 +1,7 @@ +import modules + +def test_coinflip(): + assert('coin' == modules.process_query('Flip a coin')[0]) + assert('coin' == modules.process_query('Jarvis flip a coin')[0]) + assert('coin' == modules.process_query('Can you flip a coin')[0]) + assert('coin' != modules.process_query('something random')[0])