Updated logic adapter tests.

chatterbot/adapters/logic/closest_match.py

@@ -1,4 +1,5 @@
 from .logic import LogicAdapter
+from fuzzywuzzy import process
 
 
 class ClosestMatchAdapter(LogicAdapter):
@@ -8,8 +9,6 @@ def get(self, text, list_of_statements):
         Takes a statement string and a list of statement strings.
         Returns the closest matching statement from the list.
         """
-        from fuzzywuzzy import process
-
         # If the list is empty, return the statement
         if not list_of_statements:
             return text

chatterbot/adapters/logic/logic.py

@@ -3,5 +3,5 @@
 
 class LogicAdapter(object):
 
-    def get_response(self):
+    def get(self):
         raise AdapterNotImplementedError()

tests/io_adapter_tests/test_terminal_adapter.py

@@ -10,14 +10,6 @@ class TerminalAdapterTests(TestCase):
     a command line interface.
     """
 
-    def test_response_is_printed(self):
-        """
-        This test ensures that the bot's response is
-        correctly printed in the terminal.
-        """
-        # TODO: How can you test this?
-        self.assertTrue(True)
-
     def test_response_is_returned(self):
         """
         For consistency across io adapters, the

tests/test_logic.py → tests/logic_adapter_tests/test_closest_match.py

@@ -1,13 +1,13 @@
-from .base_case import ChatBotTestCase
+from unittest import TestCase
 from chatterbot.adapters.logic import ClosestMatchAdapter
 
 
-class ClosestMatchAdapterTests(ChatBotTestCase):
+class ClosestMatchAdapterTests(TestCase):
 
-    def test_get_closest_statement(self):
-
-        adapter = ClosestMatchAdapter()
+    def setUp(self):
+        self.adapter = ClosestMatchAdapter()
 
+    def test_get_closest_statement(self):
         possible_choices = [
             "Who do you love?",
             "What is the meaning of life?",
@@ -17,6 +17,13 @@ def test_get_closest_statement(self):
             "I hear you are going on a quest?",
         ]
 
-        close = adapter.get("What is your quest?", possible_choices)
+        close = self.adapter.get("What is your quest?", possible_choices)
 
         self.assertEqual("What... is your quest?", close)
+
+    def test_no_choices(self):
+        possible_choices = []
+        close = self.adapter.get("What is your quest?", possible_choices)
+
+        self.assertEqual("What is your quest?", close)
+

tests/test_chatbot_output.py

@@ -43,9 +43,8 @@ def test_get_most_frequent_response(self):
     def test_get_first_response(self):
         statement_list = [
             Statement("What... is your quest?", occurrence=2),
-            Statement("This is a phone.", occurrence=4),
             Statement("A what?", occurrence=2),
-            Statement("A phone.", occurrence=2)
+            Statement("A quest.", occurrence=2)
         ]
 
         output = self.chatbot.get_first_response(
@@ -55,7 +54,17 @@ def test_get_first_response(self):
         self.assertEqual("What... is your quest?", output)
 
     def test_get_random_response(self):
-        pass # TODO
+        statement_list = [
+            Statement("This is a phone.", occurrence=4),
+            Statement("A what?", occurrence=2),
+            Statement("A phone.", occurrence=2)
+        ]
+
+        output = self.chatbot.get_random_response(
+            statement_list
+        )
+
+        self.assertTrue(output)
 
     def test_answer_to_known_input(self):
         """