def test_to_xml(self):
root = TemplateNode()
node = TemplateVocabularyNode()
root.append(node)
node.append(TemplateWordNode("Test"))
xml = root.xml_tree(self.bot, self.clientid)
self.assertIsNotNone(xml)
xml_str = ET.tostring(xml, "utf-8").decode("utf-8")
self.assertEqual("<template><vocabulary>Test</vocabulary></template>", xml_str)
def test_to_xml(self):
root = TemplateNode()
node = TemplateVocabularyNode()
root.append(node)
node.append(TemplateWordNode("Test"))
xml = root.xml_tree(self.bot, self.clientid)
self.assertIsNotNone(xml)
xml_str = ET.tostring(xml, "utf-8").decode("utf-8")
self.assertEqual("<template><vocabulary>Test</vocabulary></template>", xml_str)
def test_node_with_priority_word(self):
topic_element = ET.fromstring('<topic>*</topic>')
that_element = ET.fromstring('<that>*</that>')
pattern_element = ET.fromstring("<pattern>hello $world yes</pattern>")
self._client_context.brain._aiml_parser.pattern_parser.add_pattern_to_graph(
pattern_element, topic_element, that_element, None)
self._client_context.brain.sets._sets["TESTSET"] = {
"VAL1": ["VAL1"],
"VAL2": ["VAL2"],
"VAL3": ["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.assertEqual(root.resolve(self._client_context), '6')
def test_node(self):
test_bot = Bot(Brain(BrainConfiguration()), BotConfiguration())
topic_element = ET.fromstring('<topic>*</topic>')
that_element = ET.fromstring('<that>*</that>')
pattern_element = ET.fromstring("<pattern>hello world</pattern>")
test_bot.brain._aiml_parser.pattern_parser.add_pattern_to_graph(pattern_element, topic_element, that_element, None)
loader = SetLoader()
test_bot.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(test_bot, "testid"), '5')
def __init__(self):
TemplateVocabularyNode.__init__(self)
