def setUp(self):
super(LowConfidenceAdapterTestCase, self).setUp()
self.adapter = LowConfidenceAdapter()
# Add a mock storage adapter to the logic adapter
self.adapter.set_chatbot(self.chatbot)
possible_choices = [
Statement(
'Who do you love?',
in_response_to=[Response('I hear you are going on a quest?')]),
Statement('What is the meaning of life?',
in_response_to=[
Response('Yuck, black licorice jelly beans.')
]),
Statement('I am Iron Man.',
in_response_to=[Response('What... is your quest?')]),
Statement('What... is your quest?',
in_response_to=[Response('I am Iron Man.')]),
Statement(
'Yuck, black licorice jelly beans.',
in_response_to=[Response('What is the meaning of life?')]),
Statement('I hear you are going on a quest?',
in_response_to=[Response('Who do you love?')]),
]
self.adapter.chatbot.storage.filter = MagicMock(
return_value=possible_choices)
def setUp(self):
super().setUp()
self.adapter = LowConfidenceAdapter(self.chatbot)
possible_choices = [
Statement(
text='Who do you love?',
in_response_to='I hear you are going on a quest?'
),
Statement(
text='What is the meaning of life?',
in_response_to='Yuck, black licorice jelly beans.'
),
Statement(
text='I am Iron Man.',
in_response_to='What... is your quest?'
),
Statement(
text='What... is your quest?',
in_response_to='I am Iron Man.'
),
Statement(
text='Yuck, black licorice jelly beans.',
in_response_to='What is the meaning of life?'
),
Statement(
text='I hear you are going on a quest?',
in_response_to='Who do you love?'
),
]
self.adapter.chatbot.storage.filter = MagicMock(return_value=possible_choices)
class LowConfidenceAdapterTestCase(ChatBotTestCase):
"""
Test cases for the LowConfidenceAdapter.
"""
def setUp(self):
super(LowConfidenceAdapterTestCase, self).setUp()
self.adapter = LowConfidenceAdapter()
# Add a mock storage adapter to the logic adapter
self.adapter.set_chatbot(self.chatbot)
possible_choices = [
Statement('Who do you love?',
in_response_to='I hear you are going on a quest?'),
Statement('What is the meaning of life?',
in_response_to='Yuck, black licorice jelly beans.'),
Statement('I am Iron Man.',
in_response_to='What... is your quest?'),
Statement('What... is your quest?',
in_response_to='I am Iron Man.'),
Statement('Yuck, black licorice jelly beans.',
in_response_to='What is the meaning of life?'),
Statement('I hear you are going on a quest?',
in_response_to='Who do you love?'),
]
self.adapter.chatbot.storage.filter = MagicMock(
return_value=possible_choices)
def test_high_confidence(self):
"""
Test the case that a high confidence response is known.
"""
statement = Statement('What is your quest?')
match = self.adapter.process(statement)
self.assertEqual(match.confidence, 0)
self.assertEqual(match, self.adapter.default_responses[0])
def test_low_confidence(self):
"""
Test the case that a high confidence response is not known.
"""
statement = Statement('Is this a tomato?')
match = self.adapter.process(statement)
self.assertEqual(match.confidence, 1)
self.assertEqual(match, self.adapter.default_responses[0])
def test_low_confidence_options_list(self):
"""
Test the case that a high confidence response is not known.
"""
self.adapter.default_responses = [Statement(text='No')]
statement = Statement('Is this a tomato?')
match = self.adapter.process(statement)
self.assertEqual(match.confidence, 1)
self.assertEqual(match, 'No')
def test_low_confidence(self):
from chatterbot.logic import LowConfidenceAdapter
adapter = LowConfidenceAdapter(self.chatbot)
statement = Statement(text='Why is the sky blue?')
response = adapter.process(statement)
self.assertEqual(response.text, adapter.default_responses[0])
def test_low_confidence(self):
from chatterbot.logic import LowConfidenceAdapter
adapter = LowConfidenceAdapter(self.chatbot)
statement = Statement(text='Why is the sky blue?')
response = adapter.process(statement)
self.assertEqual(response.text, adapter.default_responses[0])
class LowConfidenceAdapterTestCase(ChatBotTestCase):
"""
Test cases for the LowConfidenceAdapter.
"""
def setUp(self):
super(LowConfidenceAdapterTestCase, self).setUp()
self.adapter = LowConfidenceAdapter()
# Add a mock storage adapter to the logic adapter
self.adapter.set_chatbot(self.chatbot)
possible_choices = [
Statement('Who do you love?', in_response_to=[
Response('I hear you are going on a quest?')
]),
Statement('What is the meaning of life?', in_response_to=[
Response('Yuck, black licorice jelly beans.')
]),
Statement('I am Iron Man.', in_response_to=[
Response('What... is your quest?')
]),
Statement('What... is your quest?', in_response_to=[
Response('I am Iron Man.')
]),
Statement('Yuck, black licorice jelly beans.', in_response_to=[
Response('What is the meaning of life?')
]),
Statement('I hear you are going on a quest?', in_response_to=[
Response('Who do you love?')
]),
]
self.adapter.chatbot.storage.filter = MagicMock(return_value=possible_choices)
def test_high_confidence(self):
"""
Test the case that a high confidence response is known.
"""
statement = Statement('What is your quest?')
confidence, match = self.adapter.process(statement)
self.assertEqual(confidence, 0)
self.assertEqual(match.confidence, 0)
self.assertEqual(match, self.adapter.default_response)
def test_low_confidence(self):
"""
Test the case that a high confidence response is not known.
"""
statement = Statement('Is this a tomato?')
confidence, match = self.adapter.process(statement)
self.assertEqual(confidence, 1)
self.assertEqual(match.confidence, 1)
self.assertEqual(match, self.adapter.default_response)
def setUp(self):
super(LowConfidenceAdapterTestCase, self).setUp()
self.adapter = LowConfidenceAdapter()
# Add a mock storage adapter to the logic adapter
self.adapter.set_chatbot(self.chatbot)
possible_choices = [
Statement('Who do you love?', in_response_to=[
Response('I hear you are going on a quest?')
]),
Statement('What is the meaning of life?', in_response_to=[
Response('Yuck, black licorice jelly beans.')
]),
Statement('I am Iron Man.', in_response_to=[
Response('What... is your quest?')
]),
Statement('What... is your quest?', in_response_to=[
Response('I am Iron Man.')
]),
Statement('Yuck, black licorice jelly beans.', in_response_to=[
Response('What is the meaning of life?')
]),
Statement('I hear you are going on a quest?', in_response_to=[
Response('Who do you love?')
]),
]
self.adapter.chatbot.storage.filter = MagicMock(return_value=possible_choices)
def setUp(self):
super().setUp()
self.adapter = LowConfidenceAdapter(self.chatbot)
self.chatbot.storage.create(
text='Who do you love?',
in_response_to='I hear you are going on a quest?')
self.chatbot.storage.create(
text='What is the meaning of life?',
in_response_to='Yuck, black licorice jelly beans.')
self.chatbot.storage.create(text='I am Iron Man.',
in_response_to='What... is your quest?')
self.chatbot.storage.create(text='What... is your quest?',
in_response_to='I am Iron Man.')
self.chatbot.storage.create(
text='Yuck, black licorice jelly beans.',
in_response_to='What is the meaning of life?')
self.chatbot.storage.create(text='I hear you are going on a quest?',
in_response_to='Who do you love?')
