test

Author: victorbalssa

__init__.py

@@ -11,6 +11,6 @@ pymongo==2.8
 pytz==2015.2
 six==1.9.0
 wsgiref==0.1.2
-unittest2==1.1.0
 celery==4.0.2
 requests==2.13.0
+pytest==3.2.3

requirements.txt

@@ -0,0 +1,47 @@
+import numpy as np
+
+from chatbot import clean_up_sentence
+from chatbot import bow
+from chatbot import classify
+from chatbot import chat_response
+
+
+def test_clean_up_hello_sentence_():
+    assert clean_up_sentence("Hi, HoW are you?!") == ['hi', ',', 'how', 'ar', 'you', '?', '!']
+
+
+def test_bow_with_no_equality_():
+    assert bow("test", ["plop"]) == [0]
+
+
+def test_bow_with_one_equality_():
+    assert bow("test", ["test"]) == [1]
+
+
+def test_classify_greeting_sentence_():
+    assert classify("Hi")[0][0] == "greeting"
+
+
+def test_classify_shutter_sentence_():
+    assert classify("bye")[0][0] == "goodbye"
+
+
+def test_classify_camera_sentence_():
+    assert classify("camera")[0][0] == "take_photo"
+
+
+def test_classify_type_():
+    assert type(classify("camera")[0][1]) == np.float32
+
+
+def test_chat_greeting_response_():
+    assert chat_response("Hi, HoW are you?!") == "Good to see you again" or "Hello, thanks for visiting" \
+           or "Hi there, how can I help?"
+
+
+def test_chat_camera_response_():
+    assert chat_response("camera") == "exec camera_photo"
+
+
+def test_chat_thanks_response_():
+    assert chat_response("thanks") == "Happy to help!" or "Any time!" or "My pleasure"

test_chatbot.py

@@ -0,0 +1,6 @@
+import pickle
+
+from chatbot import clean_up_sentence
+from chatbot import bow
+from chatbot import classify
+from chatbot import chat_response