17 lines
282 B
Python
17 lines
282 B
Python
"""
|
|
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" |