def test_bucket(): """Should add stuff & give it back based on keys""" bucket = Bucket() bucket.add("key1", "value1") assert bucket.retrieve("key1") == "value1" bucket.add("key2", "value2") assert bucket.retrieve("key1") == "value1" assert bucket.retrieve("key2") == "value2" # Resets stuff, too bucket.add("key1", "value3") assert bucket.retrieve("key1") == "value3"