def setUp(self):
from chatterbot.adapters import Adaptation
from chatterbot.adapters.storage import JsonDatabaseAdapter
self.mixin = KnownResponseMixin()
context = Adaptation()
adapter = JsonDatabaseAdapter(context)
# Simulate a storage adapter
setattr(self.mixin, "context", context)
setattr(self.mixin.context, "storage", adapter)
class ResponseSelectionMixinTests(TestCase):
def setUp(self):
from chatterbot.adapters import Adaptation
from chatterbot.adapters.storage import JsonDatabaseAdapter
self.mixin = KnownResponseMixin()
context = Adaptation()
adapter = JsonDatabaseAdapter(context)
# Simulate a storage adapter
setattr(self.mixin, "context", context)
setattr(self.mixin.context, "storage", adapter)
def test_get_statements_with_known_responses(self):
statement_list = [
Statement("What... is your quest?"),
Statement("This is a phone."),
Statement("A what?", in_response_to=[Response("This is a phone.")]),
Statement("A phone.", in_response_to=[Response("A what?")])
]
for statement in statement_list:
self.mixin.context.storage.update(statement)
responses = self.mixin.get_statements_with_known_responses()
self.assertEqual(len(responses), 2)
class ResponseSelectionMixinTests(TestCase):
def setUp(self):
from chatterbot.adapters import Adaptation
from chatterbot.adapters.storage import JsonDatabaseAdapter
self.mixin = KnownResponseMixin()
context = Adaptation()
adapter = JsonDatabaseAdapter()
adapter.set_context(context)
# Simulate a storage adapter
setattr(self.mixin, "context", context)
setattr(self.mixin.context, "storage", adapter)
def test_get_statements_with_known_responses(self):
statement_list = [
Statement("What... is your quest?"),
Statement("This is a phone."),
Statement("A what?", in_response_to=[Response("This is a phone.")]),
Statement("A phone.", in_response_to=[Response("A what?")])
]
for statement in statement_list:
self.mixin.context.storage.update(statement)
responses = self.mixin.get_statements_with_known_responses()
self.assertEqual(len(responses), 2)
def setUp(self):
from chatterbot.adapters import Adaptation
from chatterbot.adapters.storage import JsonDatabaseAdapter
self.mixin = KnownResponseMixin()
context = Adaptation()
adapter = JsonDatabaseAdapter()
adapter.set_context(context)
# Simulate a storage adapter
setattr(self.mixin, "context", context)
setattr(self.mixin.context, "storage", adapter)
