def test_add_pattern_to_node(self):
graph = PatternGraph()
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_pattern_to_node(self):
graph = PatternGraph(self._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_pattern_to_node_whitespaces(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_pattern_to_node_whitespaces(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_that_to_node_children(self):
graph = PatternGraph(self._client_context.brain.aiml_parser)
pattern = ET.fromstring("""
<pattern>
HELLO
</pattern>""")
pattern_node = graph.add_pattern_to_node(pattern)
that = ET.fromstring("""
<that>
THIS <set name="other" /> OTHER
</that>""")
that_node = graph.add_that_to_node(that, pattern_node)
self.assertIsNotNone(that_node)
self.assertIsInstance(that_node, PatternWordNode)
def test_add_topic_to_node_multi_children(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>
THIS <set name="ubertopic" /> OTHER
</topic>""")
topic_node = graph.add_topic_to_node(topic, pattern_node)
self.assertIsNotNone(topic_node)
self.assertIsInstance(topic_node, PatternWordNode)
def test_add_topic_to_node_word_with_multiple_words(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>""")
last_node = graph.add_topic_to_node(topic, pattern_node)
self.assertIsNotNone(last_node)
self.assertIsInstance(last_node, PatternOneOrMoreWildCardNode)
def test_add_topic_to_node_wildcard(self):
graph = PatternGraph(self._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_topic_to_node_word(self):
graph = PatternGraph(self._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)
self.assertIsNotNone(topic_node)
self.assertIsInstance(topic_node, PatternWordNode)
self.assertEqual(topic_node.word, "THERE")
def test_add_that_to_node_word_multiple_words(self):
graph = PatternGraph(self._client_context.brain.aiml_parser)
pattern = ET.fromstring("""
<pattern>
HELLO
</pattern>""")
pattern_node = graph.add_pattern_to_node(pattern)
that = ET.fromstring("""
<that>
HERE *
</that>""")
that_node = graph.add_that_to_node(that, pattern_node)
self.assertIsNotNone(that_node)
self.assertIsInstance(that_node, PatternOneOrMoreWildCardNode)
self.assertEqual(that_node.wildcard, "*")
def test_add_topic_to_node_wildcard(self):
graph = PatternGraph()
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_topic_to_node_word(self):
graph = PatternGraph()
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)
self.assertIsNotNone(topic_node)
self.assertIsInstance(topic_node, PatternWordNode)
self.assertEqual(topic_node.word, "THERE")
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>
*
</that>""")
that_node = graph.add_that_to_node(that, topic_node)
self.assertIsNotNone(that_node)
self.assertIsInstance(that_node, PatternOneOrMoreWildCardNode)
self.assertEqual(that_node.wildcard, "*")
def test_add_template_to_node(self):
graph = PatternGraph(self._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)
def test_add_template_to_node(self):
graph = PatternGraph()
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)
def test_add_pattern_to_node_invalid(self):
graph = PatternGraph(self._client_context.brain.aiml_parser)
pattern = ET.fromstring('<pattern><get something="XXX" /></pattern>')
with self.assertRaises(ParserException):
_ = graph.add_pattern_to_node(pattern)