def test_get_inorder(self):
node = Node(3)
node.insert(1)
node.insert(5)
self.assertListEqual([(1, NodeColor.RED), (3, NodeColor.BLACK),
(5, NodeColor.RED)], node.get_inorder())
node.insert(6)
self.assertListEqual([(1, NodeColor.BLACK), (3, NodeColor.BLACK),
(5, NodeColor.BLACK), (6, NodeColor.RED)],
node.get_inorder())
def test_right_left_case(self):
root = Node(15)
root._color = NodeColor.BLACK
root._left = Node(7)
root._left._color = NodeColor.BLACK
root._left._parent = root
root._right = Node(25)
root._right._color = NodeColor.RED
root._right._parent = root
self.assertListEqual([(7, NodeColor.BLACK), (15, NodeColor.BLACK),
(25, NodeColor.RED)], root.get_inorder())
root.insert(24)
root = root._parent
self.assertListEqual([(7, NodeColor.BLACK), (15, NodeColor.RED),
(24, NodeColor.BLACK), (25, NodeColor.RED)],
root.get_inorder())