def test_multiple_templates(self):
node1 = PatternTemplateNode(TemplateNode())
node2 = PatternTemplateNode(TemplateNode())
with self.assertRaises(ParserException) as raised:
node1.can_add(node2)
self.assertEqual(str(raised.exception), "Cannot add template node to template node")
def test_equivalent(self):
node1 = PatternTemplateNode(TemplateNode())
node2 = PatternTemplateNode(TemplateNode())
node3 = PatternTemplateNode(TemplateNode(), userid="testid")
self.assertTrue(node1.equivalent(node2))
self.assertFalse(node1.equivalent(node3))
def test_init(self):
node = PatternTemplateNode(TemplateNode())
self.assertIsNotNone(node)
self.assertFalse(node.is_root())
self.assertFalse(node.is_priority())
self.assertFalse(node.is_wildcard())
self.assertFalse(node.is_zero_or_more())
self.assertFalse(node.is_one_or_more())
self.assertFalse(node.is_set())
self.assertFalse(node.is_bot())
self.assertTrue(node.is_template())
self.assertFalse(node.is_that())
self.assertFalse(node.is_topic())
self.assertFalse(node.is_wildcard())
self.assertIsNotNone(node.children)
self.assertFalse(node.has_children())
self.assertTrue(node.equivalent(PatternTemplateNode(TemplateNode())))
self.assertEqual(
node.to_string(),
"PTEMPLATE [*] [P(0)^(0)#(0)C(0)_(0)*(0)To(0)Th(0)Te(1)]")
self.assertEqual(node.to_string(verbose=False), "PTEMPLATE")
self.assertEqual('<template></template>\n',
node.to_xml(self._client_context))
self.assertFalse(node.equivalent(PatternTopicNode()))
def test_to_string(self):
node1 = PatternTemplateNode(TemplateNode())
self.assertEqual(node1.to_string(verbose=False), "PTEMPLATE")
self.assertEqual(
node1.to_string(),
"PTEMPLATE [*] [P(0)^(0)#(0)C(0)_(0)*(0)To(0)Th(0)Te(1)]")
node2 = PatternTemplateNode(TemplateNode(), userid="testid")
self.assertEqual(node2.to_string(verbose=False), "PTEMPLATE")
self.assertEqual(
node2.to_string(),
"PTEMPLATE [testid] [P(0)^(0)#(0)C(0)_(0)*(0)To(0)Th(0)Te(1)]")
def test_to_xml(self):
node1 = PatternTemplateNode(TemplateNode())
self.assertEqual('<template></template>\n',
node1.to_xml(self._client_context))
self.assertEqual('<template userid="*"></template>\n',
node1.to_xml(self._client_context, include_user=True))
node2 = PatternTemplateNode(TemplateNode(), userid="testid")
self.assertEqual('<template></template>\n',
node2.to_xml(self._client_context))
self.assertEqual('<template userid="testid"></template>\n',
node2.to_xml(self._client_context, include_user=True))
def test_template_node_exists(self):
node = PatternNode()
self.assertIsNotNone(node)
new_node1 = PatternTemplateNode(TemplateNode())
self.assertIsNone(node._template_node_exist(new_node1))
node.add_template(new_node1)
self.assertIsNotNone(node._template_node_exist(new_node1))
new_node2 = PatternTemplateNode(TemplateNode(), userid="testid2")
self.assertIsNotNone(node._template_node_exist(new_node2))
node.add_template(new_node2)
self.assertIsNotNone(node._template_node_exist(new_node2))
def test_simple_hash_and_star_at_end(self):
graph = PatternGraph(self._client_context.brain.aiml_parser)
pattern_element = ET.fromstring("<pattern>A # *</pattern>")
topic_element = ET.fromstring("<topic>*</topic>")
that_element = ET.fromstring("<that>*</that>")
template_node = TemplateNode()
graph.add_pattern_to_graph(pattern_element, topic_element,
that_element, template_node)
self.assertIsNotNone(graph.root)
self.assertIsNotNone(graph.root.children)
self.assertEqual(len(graph.root.children), 1)
word_node = graph.root.children[0]
self.assertIsInstance(word_node, PatternWordNode)
self.assertEqual(word_node.word, "A")
self.assertTrue(word_node.has_zero_or_more())
if word_node.arrow is not None:
wildcard_node = word_node.arrow
elif word_node.hash is not None:
wildcard_node = word_node.hash
self.assertIsNotNone(wildcard_node)
self.assertTrue(wildcard_node.has_one_or_more())
if word_node.star is not None:
wildcard_node = word_node.star
elif word_node.underline is not None:
wildcard_node = word_node.underline
self.assertIsNotNone(wildcard_node)
def test_carousel_node(self):
root = TemplateNode()
self.assertIsNotNone(root)
self.assertIsNotNone(root.children)
self.assertEqual(len(root.children), 0)
carousel = TemplateCarouselNode()
card = TemplateCardNode()
card._image = TemplateWordNode("http://Servusai.com")
card._title = TemplateWordNode("Servusai.com")
card._subtitle = TemplateWordNode("The home of programr")
button = TemplateButtonNode()
button._text = TemplateWordNode("More...")
button._url = TemplateWordNode("http://Servusai.com/aiml")
card._buttons.append(button)
carousel._cards.append(card)
root.append(carousel)
resolved = root.resolve(self._client_context)
self.assertIsNotNone(resolved)
self.assertEquals(
"<carousel><card><image>http://Servusai.com</image><title>Servusai.com</title><subtitle>The home of programr</subtitle><button><text>More...</text><url>http://Servusai.com/aiml</url></button></card></carousel>",
resolved)
self.assertEquals(
"<carousel><card><image>http://Servusai.com</image><title>Servusai.com</title><subtitle>The home of programr</subtitle><button><text>More...</text><url>http://Servusai.com/aiml</url></button></card></carousel>",
root.to_xml(self._client_context))
def test_resolve_with_no_defaults(self):
root = TemplateNode()
self.assertIsNotNone(root)
self.assertIsNotNone(root.children)
self.assertEqual(len(root.children), 0)
node = TemplateRequestNode(index=1)
self.assertIsNotNone(node)
root.append(node)
self.assertEqual(len(root.children), 1)
self.assertEqual(1, node.index)
conversation = Conversation(self._client_context)
self._client_context.bot._conversations["testid"] = conversation
question = Question.create_from_text(
self._client_context.brain.nlp.tokenizer, "Hello world")
question.current_sentence()._response = "Hello matey"
conversation._questions.append(question)
question = Question.create_from_text(
self._client_context.brain.nlp.tokenizer, "What did you say")
question.current_sentence()._response = "Hello matey"
conversation._questions.append(question)
response = root.resolve(self._client_context)
self.assertIsNotNone(response)
self.assertEqual(response, "Hello world")
def test_resolve_no_defaults(self):
root = TemplateNode()
self.assertIsNotNone(root)
self.assertIsNotNone(root.children)
self.assertEqual(len(root.children), 0)
node = TemplateInputNode(index=1)
self.assertIsNotNone(node)
root.append(node)
self.assertEqual(len(root.children), 1)
self.assertEqual(1, node.index)
conversation = Conversation(self._client_context)
question = Question.create_from_text(
self._client_context.brain.nlp.tokenizer, "Hello world")
question.current_sentence()._response = "Hello matey"
conversation.record_question(question)
question = Question.create_from_text(
self._client_context.brain.nlp.tokenizer,
"How are you. Are you well")
question.current_sentence()._response = "Fine thanks"
conversation.record_question(question)
self._client_context.bot._conversations["testid"] = conversation
response = root.resolve(self._client_context)
self.assertIsNotNone(response)
self.assertEqual(response, "How are you")
def test_type2_node_bot(self):
root = TemplateNode()
self.assertIsNotNone(root)
self.assertIsNotNone(root.children)
self.assertEqual(len(root.children), 0)
node = TemplateConditionNode("cond1",
var_type=TemplateConditionNode.BOT,
condition_type=2)
self.assertIsNotNone(node)
cond1 = TemplateConditionListItemNode(
value=TemplateWordNode("value1"),
var_type=TemplateConditionNode.BOT)
cond1.append(TemplateWordNode("Word1"))
node.append(cond1)
cond2 = TemplateConditionListItemNode(
value=TemplateWordNode("value2"),
var_type=TemplateConditionNode.BOT)
cond2.append(TemplateWordNode("Word2"))
node.append(cond2)
cond3 = TemplateConditionListItemNode()
cond3.append(TemplateWordNode("Word3"))
node.append(cond3)
root.append(node)
self.assertEqual(len(root.children), 1)
self._client_context.brain.properties.add_property('cond1', "value2")
result = root.resolve(self._client_context)
self.assertIsNotNone(result)
self.assertEqual("Word2", result)
def test_type2_to_xml_bot(self):
root = TemplateNode()
node = TemplateConditionNode("cond1",
var_type=TemplateConditionNode.BOT,
condition_type=2)
self.assertIsNotNone(node)
cond1 = TemplateConditionListItemNode(value=TemplateWordNode("value1"))
cond1.append(TemplateWordNode("Word1"))
node.append(cond1)
cond2 = TemplateConditionListItemNode(value=TemplateWordNode("value2"))
cond2.append(TemplateWordNode("Word2"))
node.append(cond2)
cond3 = TemplateConditionListItemNode()
cond3.append(TemplateWordNode("Word3"))
node.append(cond3)
root.append(node)
xml = root.xml_tree(self._client_context)
self.assertIsNotNone(xml)
xml_str = ET.tostring(xml, "utf-8").decode("utf-8")
self.assertEqual(
'<template><condition bot="cond1"><li><value>value1</value>Word1</li> <li><value>value2</value>Word2</li> <li>Word3</li></condition></template>',
xml_str)
def test_type2_node_global(self):
root = TemplateNode()
self.assertIsNotNone(root)
self.assertIsNotNone(root.children)
self.assertEqual(len(root.children), 0)
node = TemplateConditionNode("cond1", condition_type=2)
self.assertIsNotNone(node)
cond1 = TemplateConditionListItemNode(value=TemplateWordNode("value1"))
cond1.append(TemplateWordNode("Word1"))
node.append(cond1)
cond2 = TemplateConditionListItemNode(value=TemplateWordNode("value2"))
cond2.append(TemplateWordNode("Word2"))
node.append(cond2)
cond3 = TemplateConditionListItemNode()
cond3.append(TemplateWordNode("Word3"))
node.append(cond3)
root.append(node)
self.assertEqual(len(root.children), 1)
self._client_context.bot.conversation(
self._client_context).set_property('cond1', "value2")
question = Question.create_from_text(
self._client_context.brain.nlp.tokenizer, "Hello")
self._client_context.bot.conversation(
self._client_context).record_question(question)
self._client_context.bot.conversation(
self._client_context).current_question().set_property(
"cond1", "value2")
result = root.resolve(self._client_context)
self.assertIsNotNone(result)
self.assertEqual("Word2", result)
def test_check_child_is_wildcard_hash(self):
wildcard = MockPatternWildCardNode("*")
self.assertIsNotNone(wildcard)
wildcard._0ormore_hash = PatternZeroOrMoreWildCardNode('#')
wildcard._0ormore_hash._template = PatternTemplateNode(TemplateNode())
context = MatchContext(
max_search_depth=100,
max_search_timeout=-1,
tokenizer=self._client_context.brain.nlp.tokenizer)
sentence = Sentence(self._client_context.brain.nlp.tokenizer,
"TEST SENTENCE")
match = wildcard.check_child_is_wildcard("", self._client_context,
context, sentence, 1,
Match.WORD, 0)
self.assertIsNotNone(match)
context = MatchContext(
max_search_depth=100,
max_search_timeout=-1,
tokenizer=self._client_context.brain.nlp.tokenizer)
sentence = Sentence(self._client_context.brain.nlp.tokenizer, "TEST")
match = wildcard.check_child_is_wildcard("", self._client_context,
context, sentence, 0,
Match.WORD, 0)
self.assertIsNotNone(match)
def test_node(self):
learnf_path = self.get_os_specific_path() + os.sep + "testid.aiml"
if os.path.exists(learnf_path):
os.remove(learnf_path)
self.assertFalse(os.path.exists(learnf_path))
root = TemplateNode()
self.assertIsNotNone(root)
learn = TemplateLearnfNode()
self.assertIsNotNone(learn)
learn_cat = LearnCategory(
ET.fromstring("<pattern>HELLO LEARN</pattern>"),
ET.fromstring("<topic>*</topic>"), ET.fromstring("<that>*</that>"),
TemplateWordNode("LEARN"))
learn.append(learn_cat)
root.append(learn)
self.assertEqual(1, len(root.children))
self._client_context.brain.configuration.defaults._learn_path = self.get_os_specific_path(
)
resolved = root.resolve(self._client_context)
self.assertIsNotNone(resolved)
self.assertEqual("", resolved)
self.assertTrue(os.path.exists(learnf_path))
if os.path.exists(learnf_path):
os.remove(learnf_path)
self.assertFalse(os.path.exists(learnf_path))
def test_resolve_with_no_defaults(self):
root = TemplateNode()
self.assertIsNotNone(root)
self.assertIsNotNone(root.children)
self.assertEqual(len(root.children), 0)
node = TemplateThatNode(question=1, sentence=1)
self.assertIsNotNone(node)
root.append(node)
self.assertEqual(len(root.children), 1)
self.assertEqual(1, node.question)
self.assertEqual(1, node.sentence)
conversation = Conversation(self._client_context)
question = Question.create_from_text(
self._client_context.brain.nlp.tokenizer, "Hello world")
question.current_sentence()._response = "Hello matey"
conversation.record_question(question)
question = Question.create_from_text(
self._client_context.brain.nlp.tokenizer, "How are you")
question.current_sentence()._response = "Very well thanks"
conversation.record_question(question)
self._client_context.bot._conversations["testid"] = conversation
self.assertEqual("Hello matey", node.resolve(self._client_context))
def test_resolve_with_no_defaults(self):
root = TemplateNode()
self.assertIsNotNone(root)
self.assertIsNotNone(root.children)
self.assertEqual(len(root.children), 0)
node = TemplateResponseNode(index=1)
self.assertIsNotNone(node)
self.assertEqual(1, node.index)
root.append(node)
self.assertEqual(len(root.children), 1)
conversation = Conversation(self._client_context)
question = Question.create_from_text(
self._client_context.brain.nlp.tokenizer, "Hello1 question")
question.current_sentence()._response = "Hello1 response"
conversation.record_question(question)
question = Question.create_from_text(
self._client_context.brain.nlp.tokenizer, "Hello quesiton2")
question.current_sentence()._response = "Hello2 response"
conversation.record_question(question)
self._client_context.bot._conversations["testid"] = conversation
response = root.resolve(self._client_context)
self.assertIsNotNone(response)
self.assertEqual(response, "Hello1 response")
def test_type1_node_local_nomatch(self):
root = TemplateNode()
self.assertIsNotNone(root)
self.assertIsNotNone(root.children)
self.assertEqual(len(root.children), 0)
node = TemplateConditionNode("var1",
TemplateWordNode("value1"),
var_type=TemplateConditionNode.LOCAL)
self.assertIsNotNone(node)
node.append(TemplateWordNode("Hello"))
root.append(node)
self.assertEqual(len(root.children), 1)
question = Question.create_from_text(
self._client_context.brain.nlp.tokenizer, "Hello")
self._client_context.bot.conversation(
self._client_context).record_question(question)
self._client_context.bot.conversation(
self._client_context).current_question().set_property(
"var1", "value2")
result = root.resolve(self._client_context)
self.assertIsNotNone(result)
self.assertEqual(result, "")
def test_xml_tree_simple(self):
node = TemplateNode()
node.append(TemplateWordNode("Word1"))
node.append(TemplateWordNode("Word2"))
xml = node.xml_tree(self._client_context)
xml_str = ET.tostring(xml, "utf-8").decode("utf-8")
self.assertEqual("<template>Word1 Word2</template>", xml_str)
def test_node(self):
topic_element = ET.fromstring('<topic>*</topic>')
that_element = ET.fromstring('<that>*</that>')
pattern_element = ET.fromstring("<pattern>hello world</pattern>")
self._client_context.brain._aiml_parser.pattern_parser.add_pattern_to_graph(
pattern_element, topic_element, that_element, None)
loader = SetLoader()
self._client_context.brain.sets.add_set(
"testset",
loader.load_from_text("""
val1
val2
val3
"""))
root = TemplateNode()
self.assertIsNotNone(root)
self.assertIsNotNone(root.children)
self.assertEqual(len(root.children), 0)
node = TemplateVocabularyNode()
self.assertIsNotNone(node)
root.append(node)
self.assertEqual(len(root.children), 1)
self.assertEquals(root.resolve(self._client_context), '5')
def test_node_with_timeout(self):
self._client_context.brain.configuration.overrides._allow_system_aiml = True
root = TemplateNode()
self.assertIsNotNone(root)
self.assertIsNotNone(root.children)
self.assertEqual(len(root.children), 0)
node = TemplateSystemNode()
node.timeout = 1000
self.assertIsNotNone(node)
node.append(TemplateWordNode('echo "Hello World"'))
root.append(node)
self.assertEqual(len(root.children), 1)
response = root.resolve(self._client_context)
self.assertIsNotNone(response)
if os.name == 'posix':
self.assertEqual(response, "Hello World")
elif os.name == 'nt':
self.assertEqual(response, '"Hello World"')
else:
self.assertFalse(True)
def test_node_with_star_with_none(self):
root = TemplateNode()
node = TemplateThatStarNode()
root.append(node)
conversation = Conversation(self._client_context)
question = Question.create_from_text(
self._client_context.brain.nlp.tokenizer, "Hello world")
question.current_sentence()._response = "Hello matey"
conversation.record_question(question)
question = Question.create_from_text(
self._client_context.brain.nlp.tokenizer, "How are you")
question.current_sentence()._response = "Very well thanks"
conversation.record_question(question)
match = PatternOneOrMoreWildCardNode("*")
context = MatchContext(
max_search_depth=100,
max_search_timeout=-1,
tokenizer=self._client_context.brain.nlp.tokenizer)
context.add_match(Match(Match.THAT, match, None))
question.current_sentence()._matched_context = context
conversation.record_question(question)
self._client_context.bot._conversations["testid"] = conversation
self.assertEqual("", root.resolve(self._client_context))
def test_local_set(self):
root = TemplateNode()
self.assertIsNotNone(root)
self.assertIsNotNone(root.children)
self.assertEqual(len(root.children), 0)
node = TemplateSetNode()
self.assertIsNotNone(node)
node.name = TemplateWordNode("name")
node.local = True
node.append(TemplateWordNode("keith"))
root.append(node)
self.assertEqual(len(root.children), 1)
conversation = self._client_context.bot.get_conversation(self._client_context)
self.assertIsNotNone(conversation)
question = Question.create_from_text(self._client_context.brain.nlp.tokenizer, "Hello")
conversation.record_question(question)
self.assertIsNotNone(conversation.current_question())
result = node.resolve(self._client_context)
self.assertIsNotNone(result)
self.assertEqual("keith", result)
self.assertEqual("keith", question.property("name"))
def test_global_set_allow_overrides_with_default(self):
root = TemplateNode()
self.assertIsNotNone(root)
self.assertIsNotNone(root.children)
self.assertEqual(len(root.children), 0)
self._client_context.bot.configuration.override_propertys = True
self._client_context.brain.properties.pairs.append(["name", "fred"])
node = TemplateSetNode()
self.assertIsNotNone(node)
node.name = TemplateWordNode("name")
node.local = False
node.append(TemplateWordNode("keith"))
root.append(node)
self.assertEqual(len(root.children), 1)
conversation = self._client_context.bot.get_conversation(self._client_context)
self.assertIsNotNone(conversation)
question = Question.create_from_text(self._client_context.brain.nlp.tokenizer, "Hello")
conversation.record_question(question)
self.assertIsNotNone(conversation.current_question())
result = node.resolve(self._client_context)
self.assertIsNotNone(result)
self.assertEqual("keith", result)
self.assertEqual("keith", conversation.property("name"))
def test_local_no_value(self):
root = TemplateNode()
self.assertIsNotNone(root)
self.assertIsNotNone(root.children)
self.assertEqual(len(root.children), 0)
node = TemplateGetNode()
node.name = TemplateWordNode("name")
node.local = True
node.append(TemplateWordNode("Fred"))
self.assertIsNotNone(node)
root.append(node)
self.assertEqual(len(root.children), 1)
self._client_context.brain.properties.add_property(
"default-get", "unknown")
conversation = self._client_context.bot.get_conversation(
self._client_context)
self.assertIsNotNone(conversation)
question = Question.create_from_text(
self._client_context.brain.nlp.tokenizer, "Hello")
conversation.record_question(question)
result = root.resolve(self._client_context)
self.assertIsNotNone(result)
self.assertEqual("unknown", result)
def test_node_exception_handling(self):
root = TemplateNode()
node = MockTemplateSRAIXNode()
root.append(node)
result = root.resolve(self._client_context)
self.assertIsNotNone(result)
self.assertEquals("", result)
def test_node(self):
root = TemplateNode()
node = TemplateAddTripleNode(subj=TemplateWordNode("S"), pred=TemplateWordNode("P"), obj=TemplateWordNode("O"))
root.append(node)
result = root.resolve(self._client_context)
self.assertIsNotNone(result)
self.assertEquals("", result)
def test_root_to_template(self):
node1 = PatternTemplateNode(TemplateNode())
node2 = PatternRootNode()
with self.assertRaises(ParserException) as raised:
node1.can_add(node2)
self.assertEqual(str(raised.exception),
"Cannot add root node to template node")
def test_node_defaults(self):
root = TemplateNode()
node = TemplateUniqNode()
root.append(node)
result = root.resolve(self._client_context)
self.assertIsNotNone(result)
self.assertEquals("", result)
def test_to_xml(self):
root = TemplateNode()
root.append(TemplateIdNode())
xml = root.xml_tree(self._client_context)
self.assertIsNotNone(xml)
xml_str = ET.tostring(xml, "utf-8").decode("utf-8")
self.assertEqual("<template><id /></template>", xml_str)
