"""
Tests for EpisodicMemory.
"""

import pytest
from dcos.memory.episodic import EpisodicMemory


class TestEpisodicMemory:
    def test_record_and_recall(self):
        mem = EpisodicMemory()
        ep = mem.record({"action": "test"}, "success", "test episode")
        assert ep.summary == "test episode"

    def test_recall_recent(self):
        mem = EpisodicMemory()
        for i in range(5):
            mem.record({"i": i}, f"outcome_{i}")
        recent = mem.recall_recent(3)
        assert len(recent) == 3

    def test_search_by_context(self):
        mem = EpisodicMemory()
        mem.record({"type": "exploration"}, "found gold")
        mem.record({"type": "exploitation"}, "mined ore")
        results = mem.search_by_context("type", "exploration")
        assert len(results) == 1

    def test_count(self):
        mem = EpisodicMemory()
        assert mem.episode_count() == 0
        mem.record({}, "outcome")
        assert mem.episode_count() == 1