Updated tests with new KV assertions