@@ -469,9 +469,9 @@ def test_mm_prefix_caching():
469
469
# Completed block should have hashes with extra keys.
470
470
assert not computed_blocks
471
471
assert len (req0 .kv_block_hashes ) == 3
472
- assert req0 .kv_block_hashes [0 ].extra_keys == (( "aaa" , 0 ) , )
473
- assert req0 .kv_block_hashes [1 ].extra_keys == (( "aaa" , 5 ), ( "bbb" , 0 ) )
474
- assert req0 .kv_block_hashes [2 ].extra_keys == (( "bbb" , 2 ) , )
472
+ assert req0 .kv_block_hashes [0 ].extra_keys == ("aaa" , )
473
+ assert req0 .kv_block_hashes [1 ].extra_keys == ("aaa" , "bbb" )
474
+ assert req0 .kv_block_hashes [2 ].extra_keys == ("bbb" , )
475
475
476
476
blocks = manager .allocate_slots (req0 , 59 , computed_blocks )
477
477
assert [b .block_id for b in blocks ] == [0 , 1 , 2 , 3 , 4 ]
@@ -485,7 +485,7 @@ def test_mm_prefix_caching():
485
485
486
486
# The just completed block should have hashes with extra keys.
487
487
assert len (req0 .kv_block_hashes ) == 4
488
- assert req0 .kv_block_hashes [3 ].extra_keys == (( "ccc" , 0 ) , )
488
+ assert req0 .kv_block_hashes [3 ].extra_keys == ("ccc" , )
489
489
490
490
# Cache hit.
491
491
unique_token_ids = [- 1 ] * 7 + [200 ] * 5
0 commit comments