Skip to content

Commit

Permalink
Add test ids for boolean parametrizations
Browse files Browse the repository at this point in the history
  • Loading branch information
vshampor committed Jan 22, 2025
1 parent bea9823 commit 1e0bfc9
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions tests/python_tests/test_kv_cache_eviction.py
Original file line number Diff line number Diff line change
Expand Up @@ -99,8 +99,9 @@ class CacheOptTestStruct:
avg_cache_usage_optimization_ratio=1.1),
], ids=lambda x: x.test_id)
@pytest.mark.parametrize("enable_prefix_caching", [True, False]) # prefix caching shouldn't impact similarity
@pytest.mark.parametrize("apply_rotation", [True, False]) # rotation should improve similarity
@pytest.mark.parametrize("enable_prefix_caching", [True, False],
ids=["with_prefix_caching", "no_prefix_caching"]) # prefix caching shouldn't impact similarity
@pytest.mark.parametrize("apply_rotation", [True, False], ids=["with_rotation", "no_rotation"]) # rotation should improve similarity
def test_cache_optimized_generation_is_similar_to_unoptimized(converted_model, test_struct, enable_prefix_caching, apply_rotation):
import whowhatbench

Expand Down

0 comments on commit 1e0bfc9

Please sign in to comment.