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())