"""
Unit tests for the base Agent class.
"""

import pytest
from src.agent import Agent


class DummyAgent(Agent):
    def act(self, state):
        return state


def test_dummy_agent():
    agent = DummyAgent()
    assert agent.act(5) == 5
    assert agent.act("hello") == "hello"