Update test_bucket.py