File size: 545 Bytes
5a2d63f | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 | from cicd_debug_env.memory.failure_bank import FailureMemoryBank
from cicd_debug_env.models import Action
def test_memory():
bank = FailureMemoryBank(store="dict")
act = Action("fail_act", {}, 0.1, "bad")
bank.store("Error 404", act, "Fail", 0.0)
assert len(bank.memory) == 1
# We used dict fallback implicitly in query logic when passing top matching queries
res = bank.query("Error 404", 1)
assert bank.should_avoid("Error 404", "fail_act")
assert not bank.should_avoid("Error 404", "good_act")
|