def test_add_context(self):
c1 = Context().add_entry_phrase("a", 0).add_response("b", 1)
c2 = Context().add_entry_phrase("x", 2).add_response("y", 3)
c1.add_context(c2)
r = Responder().add_context(c1).add_response("q", 4)
# Verify we're not in a context
self.assertEqual(r.get_response("q")[0], 4)
self.assertEqual(r.get_response("b")[0], NoResponse)
self.assertEqual(r.get_response("y")[0], NoResponse)
# Enter first context, verify we're in it
self.assertEqual(r.get_response("a")[0], 0)
self.assertEqual(r.get_response("b")[0], 1)
# Verify we're not in the subcontext
self.assertEqual(r.get_response("y")[0], NoResponse)
# Verify we're still in the first context
self.assertEqual(r.get_response("b")[0], 1)
# Enter subcontext contained in first context, verify we're in it
self.assertEqual(r.get_response("x")[0], 2)
self.assertEqual(r.get_response("y")[0], 3)
# Verify first context isn't active anymore
self.assertEqual(r.get_response("b")[0], NoResponse)
# ... but we should be able to re-enter it, since it's a top-level
# context added to the responder
self.assertEqual(r.get_response("a")[0], 0)
self.assertEqual(r.get_response("b")[0], 1)
def test_add_contexts_invalid(self):
c = Context()
self.assertRaises(ValueError, c.add_contexts, 5)
self.assertRaises(ValueError, c.add_contexts, True)
self.assertRaises(ValueError, c.add_contexts, "test")
self.assertRaises(ValueError, c.add_contexts, Responder())
self.assertRaises(ValueError, c.add_contexts, Context(), 0)
def test_add_entry_phrases_invalid(self):
c = Context()
self.assertRaises(ValueError, c.add_entry_phrases, (0, 0))
self.assertRaises(ValueError, c.add_entry_phrases, (Context(), 0))
self.assertRaises(ValueError, c.add_entry_phrases, (False, 0))
self.assertRaises(ValueError, c.add_entry_phrases, (None, 0))
self.assertRaises(ValueError, c.add_entry_phrases, ("valid", 0),
(None, 0))
def test_add_responses_list_or_string(self):
c1 = Context().add_responses(("ab|c+", 1), ("xy|z*", 2))
c2 = Context().add_responses((["ab", "c+"], 1), (["xy", "z*"], 2))
for c in [c1, c2]:
self.assertEqual(c.get_response("ab")[0], 1)
self.assertEqual(c.get_response("c")[0], 1)
self.assertEqual(c.get_response("cccc")[0], 1)
self.assertEqual(c.get_response("xy")[0], 2)
self.assertEqual(c.get_response("zzzz")[0], 2)
def test_compile(self):
r = Responder().add_response("f?", 0)
c1 = Context().add_entry_phrase("a+", 1).add_response("b+", 2)
c1.add_chained_phrases(("x", 3), ("y", 4), ("z", 5))
c2 = Context().add_entry_phrase("q+", 6).add_response("t*", 7)
c2.add_chained_phrases(("x", 3), ("y", 4), ("z", 5))
r.add_contexts(c1, c2)
# Verify nothing's been compiled yet
for responsedict in iterate_redicts(r):
self.assertFalse(responsedict.compiled)
# Get type of compiled regex
retype = type(re.compile("abc+"))
# Compile responder
r.compile()
# Verify everything's compiled now
for responsedict in iterate_redicts(r):
self.assertTrue(len(responsedict.compiled) > 0)
for compiled in responsedict.compiled:
self.assertEqual(type(compiled), retype)
def test_add_responses(self):
c1 = Context().add_responses(("abc+", 1), ("xyz*", 2))
self.assertEqual(c1.get_response("abc")[0], 1)
self.assertEqual(c1.get_response("abcc")[0], 1)
self.assertEqual(c1.get_response("abcccc")[0], 1)
self.assertEqual(c1.get_response("xy")[0], 2)
self.assertEqual(c1.get_response("xyz")[0], 2)
self.assertEqual(c1.get_response("xyzzzz")[0], 2)
def test_add_context(self):
num_tests = 100
r = Responder()
for i in range(num_tests):
r.add_context(Context())
self.assertEqual(len(r.contexts), i + 1)
def test_add_entry_phrase_list_or_string(self):
c1 = Context().add_entry_phrase("a|b|c", 1).add_response("d", 2)
c2 = Context().add_entry_phrase(["a", "b", "c"],
3).add_response("e", 4)
r1 = Responder().add_response("x", 5).add_context(c1)
r2 = Responder().add_response("y", 6).add_context(c2)
self.assertEqual(r1.get_response("x")[0], 5)
self.assertEqual(r1.get_response("d")[0], NoResponse)
self.assertEqual(r2.get_response("y")[0], 6)
self.assertEqual(r1.get_response("e")[0], NoResponse)
# Verify we can enter context c1 with all 3 version of entry phrase
self.assertEqual(r1.get_response("a")[0], 1)
self.assertEqual(r1.get_response("d")[0], 2)
self.assertEqual(r1.get_response("x")[0], 5)
self.assertEqual(r1.get_response("b")[0], 1)
self.assertEqual(r1.get_response("d")[0], 2)
self.assertEqual(r1.get_response("x")[0], 5)
self.assertEqual(r1.get_response("c")[0], 1)
self.assertEqual(r1.get_response("d")[0], 2)
self.assertEqual(r1.get_response("x")[0], 5)
# Verify we can enter context c2 with all 3 version of entry phrase
self.assertEqual(r2.get_response("a")[0], 3)
self.assertEqual(r2.get_response("e")[0], 4)
self.assertEqual(r2.get_response("y")[0], 6)
self.assertEqual(r2.get_response("b")[0], 3)
self.assertEqual(r2.get_response("e")[0], 4)
self.assertEqual(r2.get_response("y")[0], 6)
self.assertEqual(r2.get_response("c")[0], 3)
self.assertEqual(r2.get_response("e")[0], 4)
self.assertEqual(r2.get_response("y")[0], 6)
def test_add_entry_phrases(self):
c = Context().add_entry_phrases(("a", 1), ("b", 2))
c.add_response("c", 3)
r = Responder().add_response("d", 4).add_context(c)
# Verify not in context
self.assertEqual(r.get_response("d")[0], 4)
self.assertEqual(r.get_response("c")[0], NoResponse)
self.assertFalse(r.context)
# Enter context
self.assertEqual(r.get_response("a")[0], 1)
self.assertEqual(r.get_response("c")[0], 3)
self.assertIs(r.context, c)
# Exit context
self.assertEqual(r.get_response("d")[0], 4)
# Enter context
self.assertEqual(r.get_response("b")[0], 2)
self.assertEqual(r.get_response("c")[0], 3)
self.assertIs(r.context, c)
def test_add_entry_phrase(self):
c = Context().add_entry_phrase("e", 1).add_response("f", 2)
r = Responder().add_response("a", 3).add_context(c)
# Verify not in context
self.assertEqual(r.get_response("a")[0], 3)
self.assertEqual(r.get_response("f")[0], NoResponse)
self.assertFalse(r.context)
# Enter context
self.assertEqual(r.get_response("e")[0], 1)
self.assertEqual(r.get_response("f")[0], 2)
self.assertIs(r.context, c)
# Exit context, verify we have exited
self.assertEqual(r.get_response("a")[0], 3)
self.assertEqual(r.get_response("f")[0], NoResponse)
self.assertFalse(r.context)
# Verify we can re-enter
self.assertEqual(r.get_response("e")[0], 1)
self.assertEqual(r.get_response("f")[0], 2)
self.assertIs(r.context, c)
def test_add_contexts(self):
c1 = Context().add_entry_phrase("a", 0).add_response("b", 1)
c2 = Context().add_entry_phrase("c", 2).add_response("d", 3)
c3 = Context().add_entry_phrase("e", 4).add_response("f", 5)
c1.add_contexts(c2, c3)
r = Responder().add_context(c1).add_response("q", 6)
# Verify we're not in a context
self.assertEqual(r.get_response("q")[0], 6)
self.assertEqual(r.get_response("b")[0], NoResponse)
self.assertEqual(r.get_response("d")[0], NoResponse)
self.assertEqual(r.get_response("f")[0], NoResponse)
# Verify we can't enter either of the subcontexts
self.assertEqual(r.get_response("c")[0], NoResponse)
self.assertEqual(r.get_response("e")[0], NoResponse)
# Enter first context
self.assertEqual(r.get_response("a")[0], 0)
self.assertEqual(r.get_response("b")[0], 1)
# Verify two subcontexts are not active
self.assertEqual(r.get_response("d")[0], NoResponse)
self.assertEqual(r.get_response("f")[0], NoResponse)
# Enter first subcontext
self.assertEqual(r.get_response("c")[0], 2)
self.assertEqual(r.get_response("d")[0], 3)
# Verify we can't enter 2nd subcontext from here
self.assertEqual(r.get_response("e")[0], NoResponse)
self.assertEqual(r.get_response("b")[0], NoResponse)
# Go back to first context
self.assertEqual(r.get_response("a")[0], 0)
self.assertEqual(r.get_response("b")[0], 1)
# Enter second subcontext
self.assertEqual(r.get_response("e")[0], 4)
self.assertEqual(r.get_response("f")[0], 5)
# Verify we can't enter 1st subcontext from here
self.assertEqual(r.get_response("c")[0], NoResponse)
self.assertEqual(r.get_response("b")[0], NoResponse)
# > do you have one?
# "No, computer programs can't have dogs."
# > what's your favourite thing about them?
# "They are loyal"
import random
import time
from chatbot_utils.responder import Responder, Context
random.seed(time.time())
responder = Responder()
# Add a context for talking about cats
cat_context = Context()
cat_context.add_entry_phrases(
(["(.* )?(talk about|tell( me)? about) cats?.*"], ["Sure, I love cats"]))
cat_context.add_responses(
(["(.* )?favou?rite thing about (them|cats?).*"], ["They are fuzzy"]),
(["(.* )?(do )?you have (one|(a )?cat).*"
], ["No, computer programs can't have cats."]))
# Add a context for talking about cats
dog_context = Context()
dog_context.add_entry_phrases((["(.* )?(talk about|tell( me)? about) dogs?.*"],
["Sure, I think dogs are great"]))
dog_context.add_responses(
(["(.* )?favou?rite thing about (them|dogs?).*"], ["They are loyal"]),
def test_add_entry_phrase_invalid(self):
c = Context()
self.assertRaises(ValueError, c.add_entry_phrase, 0, 0)
self.assertRaises(ValueError, c.add_entry_phrase, Context(), 0)
self.assertRaises(ValueError, c.add_entry_phrase, False, 0)
self.assertRaises(ValueError, c.add_entry_phrase, None, 0)
def test_add_chained_phrases(self):
c = Context().add_chained_phrases(("0", 0), ("1", 1), ("2", 2),
("3", 3))
c.add_response("q", "q")
# Walk through chain
self.assertEqual(c.get_response("0")[0], 0)
self.assertEqual(c.get_response("1")[0], 1)
self.assertEqual(c.get_response("2")[0], 2)
self.assertEqual(c.get_response("3")[0], 3)
# Should be able to go back one step
self.assertEqual(c.get_response("2")[0], 2)
self.assertEqual(c.get_response("3")[0], 3)
# Invalid input (no response) shouldn't exit the chain
self.assertEqual(c.get_response("h")[0], NoResponse)
self.assertEqual(c.get_response("3")[0], 3)
# Exit chain with valid response from context.responses
self.assertEqual(c.get_response("q")[0], "q")
self.assertEqual(c.get_response("3")[0], NoResponse)
# Make sure we can re-enter the chain
self.assertEqual(c.get_response("0")[0], 0)
self.assertEqual(c.get_response("1")[0], 1)
self.assertEqual(c.get_response("2")[0], 2)
self.assertEqual(c.get_response("3")[0], 3)
def test_add_response_invalid(self):
c = Context()
self.assertRaises(ValueError, c.add_response, True, 1)
self.assertRaises(ValueError, c.add_response, 1, 1)
self.assertRaises(ValueError, c.add_response, Context(), 1)
def test_add_contexts(self):
num_tests = 100
r = Responder()
contexts = [Context() for _ in range(num_tests)]
r.add_contexts(*contexts)
self.assertEqual(len(contexts), len(r.contexts))
def test_add_responses_invalid(self):
c = Context()
self.assertRaises(ValueError, c.add_responses, (True, 1))
self.assertRaises(ValueError, c.add_responses, (1, 1))
self.assertRaises(ValueError, c.add_responses, (Context(), 1))
self.assertRaises(ValueError, c.add_responses, ("valid", 1), (1, 1))