def test_add_pattern_to_node(self): graph = PatternGraph(self._client_context.brain.aiml_parser) pattern = ET.fromstring('<pattern>HELLO</pattern>') node = graph.add_pattern_to_node(pattern) self.assertIsNotNone(node) self.assertIsInstance(node, PatternWordNode) self.assertEqual(node.word, "HELLO")
def test_add_topic_to_node_wildcard(self): graph = PatternGraph(self._client_context.brain.aiml_parser) pattern = ET.fromstring(""" <pattern> HELLO </pattern>""") pattern_node = graph.add_pattern_to_node(pattern) topic = ET.fromstring(""" <topic> * </topic>""") topic_node = graph.add_that_to_node(topic, pattern_node) self.assertIsNotNone(topic_node) self.assertIsInstance(topic_node, PatternOneOrMoreWildCardNode) self.assertEqual(topic_node.wildcard, "*")
def test_add_that_to_node_word(self): graph = PatternGraph(self._client_context.brain.aiml_parser) pattern = ET.fromstring(""" <pattern> HELLO </pattern>""") pattern_node = graph.add_pattern_to_node(pattern) topic = ET.fromstring(""" <topic> THERE </topic>""") topic_node = graph.add_that_to_node(topic, pattern_node) that = ET.fromstring(""" <that> HERE </that>""") that_node = graph.add_that_to_node(that, topic_node) self.assertIsNotNone(that_node) self.assertIsInstance(that_node, PatternWordNode) self.assertEqual(that_node.word, "HERE")
def test_add_template_to_node(self): graph = PatternGraph(self._client_context.brain.aiml_parser) pattern = ET.fromstring(""" <pattern> HELLO </pattern>""") pattern_node = graph.add_pattern_to_node(pattern) topic = ET.fromstring(""" <topic> THERE </topic>""") topic_node = graph.add_that_to_node(topic, pattern_node) that = ET.fromstring(""" <that> * </that>""") that_node = graph.add_that_to_node(that, topic_node) template = TemplateWordNode("TEST") template_node = graph.add_template_to_node(template, that_node) self.assertIsNotNone(template_node) self.assertIsInstance(template_node, PatternTemplateNode)