def test_node_shape(self):
tree = Node()
children = {1:Node(),2:Node()}
tree.children = children
self.assertEqual((3,1),tree.shape())
tree.children[1].children = children = {3:Node(),4:Node()}
self.assertEqual((5,2),tree.shape())
tree.children[1].children[3].children = children = {5:Node(),6:Node()}
self.assertEqual((7,3),tree.shape())
def test_node_shape(self):
tree = Node()
children = {1: Node(), 2: Node()}
tree.children = children
self.assertEqual((3, 1), tree.shape())
tree.children[1].children = children = {3: Node(), 4: Node()}
self.assertEqual((5, 2), tree.shape())
tree.children[1].children[3].children = children = {
5: Node(),
6: Node()
}
self.assertEqual((7, 3), tree.shape())
