From f71650b413be82df00f97cd61b45e55fa10a0f37 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Robert=20H=C3=B6nig?= Date: Fri, 16 Feb 2018 13:24:25 +0100 Subject: [PATCH] Add test for TableDrivenAgentProgram. Fixes #748. --- tests/test_agents.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/tests/test_agents.py b/tests/test_agents.py index d5f63bc48..ded9b7d95 100644 --- a/tests/test_agents.py +++ b/tests/test_agents.py @@ -208,6 +208,20 @@ def test_compare_agents() : assert performance_ReflexVacummAgent <= performance_ModelBasedVacummAgent +def test_TableDrivenAgentProgram(): + table = {(('foo', 1),): 'action1', + (('foo', 2),): 'action2', + (('bar', 1),): 'action3', + (('bar', 2),): 'action1', + (('foo', 1), ('foo', 1),): 'action2', + (('foo', 1), ('foo', 2),): 'action3', + } + agent_program = TableDrivenAgentProgram(table) + assert agent_program(('foo', 1)) == 'action1' + assert agent_program(('foo', 2)) == 'action3' + assert agent_program(('invalid percept',)) == None + + def test_Agent(): def constant_prog(percept): return percept