test: add tests for agent_factory

CogitoNTNU · Apr 8, 2024 · 1e1a47a · 1e1a47a
1 parent 42b3f62
commit 1e1a47a
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/test/agents/test_agent_factory.py b/test/agents/test_agent_factory.py
@@ -0,0 +1,13 @@
+from src.agents.agent_factory import create_agent
+from src.agents.randomAgent import RandomAgent
+
+def test_create_agent_random():
+    agent = create_agent('random')
+    assert isinstance(agent, RandomAgent)
+
+def test_create_agent_unknown():
+    try:
+        create_agent('unknown')
+        assert False, "Expected ValueError"
+    except ValueError as e:
+        assert str(e) == "Unknown agent type: unknown"