def testGetLeafCount(self): bst = BinarySearchTree() l = [8, 4, 2, 1, 3, 6, 5, 7, 12, 10, 9, 11, 14, 13, 15] for d in l: bst.insert(d) self.assertEqual(bst.getLeafCount(), 8) bst = BinarySearchTree() bst.insert(2) bst.insert(1) self.assertEqual(bst.getLeafCount(), 1) bst.insert(3) self.assertEqual(bst.getLeafCount(), 2)