def test_GetAddedNodes(self):
tree = KeyedTree()
tree.add_pair("parent", "child")
self.assertEqual(tree.get_node("parent"), KeyedNode("parent"))
self.assertEqual(tree.get_node("child"), KeyedNode("child"))
self.assertEqual(
tree.get_node("child").get_parent(), tree.get_node("parent"))
def test_ParentIsAlreadyKnown(self):
tree = KeyedTree()
tree.add_pair("parent", "child")
tree.add_pair("parent", "brother")
self.assertEqual(tree.get_node("parent"), KeyedNode("parent"))
self.assertEqual(tree.get_node("child"), KeyedNode("child"))
self.assertEqual(tree.get_node("brother"), KeyedNode("brother"))
self.assertListEqual(
tree.get_node("parent").get_child_nodes(),
[KeyedNode("child"), KeyedNode("brother")])
def test_ChildIsAlreadyKnown(self):
tree = KeyedTree()
tree.add_pair("parent", "child")
tree.add_pair("grandparent", "parent")
self.assertEqual(tree.get_node("parent"), KeyedNode("parent"))
self.assertEqual(tree.get_node("grandparent"),
KeyedNode("grandparent"))
self.assertEqual(tree.get_node("child"), KeyedNode("child"))
self.assertListEqual(
tree.get_node("parent").get_child_nodes(), [KeyedNode("child")])
self.assertEqual(
tree.get_node("parent").get_parent(), tree.get_node("grandparent"))