def test_insertion(): rbt = RedBlackTree() for i in xrange(256): rbt.insert(random.randint(0, 1e6)) assert rbt.size() == 256 assert abs(rbt.balance()) <= 1 assert rbt.depth() <= 10 check_validity(rbt)