[Test] Update LoRA eviction policy tests to match current behavior (#15283)

This commit is contained in:
Alison Shao
2025-12-16 17:55:15 -08:00
committed by GitHub
parent 9d64a7b24f
commit 7ec678eb09
+6 -6
View File
@@ -90,13 +90,13 @@ class TestLoRAEvictionPolicy(unittest.TestCase):
expected_victim="lora2", expected_victim="lora2",
) )
def test_lru_base_model_priority(self): def test_lru_base_model_evicted_last(self):
"""Test LRU prioritizes base model for eviction.""" """Test LRU evicts LoRA adapters before base model (None)."""
self._test_eviction_policy( self._test_eviction_policy(
"lru", "lru",
access_sequence=["lora1", "lora2", "lora3"], access_sequence=["lora1", "lora2", "lora3"],
candidates={None, "lora1", "lora2", "lora3"}, candidates={None, "lora1", "lora2", "lora3"},
expected_victim=None, expected_victim="lora1",
) )
def test_fifo_basic(self): def test_fifo_basic(self):
@@ -135,13 +135,13 @@ class TestLoRAEvictionPolicy(unittest.TestCase):
expected_victim="lora2", expected_victim="lora2",
) )
def test_fifo_base_model_priority(self): def test_fifo_base_model_evicted_last(self):
"""Test FIFO prioritizes base model for eviction.""" """Test FIFO evicts LoRA adapters before base model (None)."""
self._test_eviction_policy( self._test_eviction_policy(
"fifo", "fifo",
access_sequence=["lora1", "lora2", "lora3"], access_sequence=["lora1", "lora2", "lora3"],
candidates={None, "lora1", "lora2", "lora3"}, candidates={None, "lora1", "lora2", "lora3"},
expected_victim=None, expected_victim="lora1",
) )
def test_policy_remove(self): def test_policy_remove(self):