def testActionSimple(self):
timePre = Prereq(int.__ge__, 30)
a = Action('work', {'time': timePre}, ([{'cash': 100}], [1.]))
self.assertTrue(a.satisfied({'time': 50}))
self.assertFalse(a.satisfied({'dog':
True})) #no satisfying time entry. Fail
def test_plan(self):
utility_funcs = {'cash': lambda x: x}
action = Action('work', {}, ([{
'cash': 100
}, {
'cash': 50
}], [0.5, 0.5]))
# an action not achievable immediately, should become a goal
action_goal = Action('hire help', {'cash': Prereq(operator.ge, 200)},
([{
'cash': 1000
}], [1.]))
goal = Goal('money', {'cash': Prereq(operator.ge, 200)}, ([{
'cash': 1000
}], [1.]))
agent = PlanningAgent({'cash': 0}, [action, action_goal], [goal],
utility_funcs)
plan, goals = agent.plan(agent._state, agent.goals, depth=3)
expected_plan = [(action, ({
'cash': 75.0
}, {goal})), (action, ({
'cash': 150.0
}, {goal})), (action, ({
'cash': 225.0
}, {goal}))]
# goals should contain the desired, but not yet satisfiable, action
self.assertEqual(goals, set([goal, action_goal]))
# returned plan should not contain unsatisfiable actions
self.assertEqual(plan, expected_plan)
def testActionEmpty(self):
a = Action(name=None,prereqs=None,outcomes=None,cost=None)
#no-prereqs, any value OK
self.assertTrue(a.satisfied(None))
self.assertTrue( a.satisfied({'dog':True}) )
self.assertEqual(a.cost(),0, 'None is a 0 cost')
self.assertEqual(str(a), 'Action(untitled action)')
def testActionBrokenStuff(self):
timePre = Prereq(int.__ge__, 30)
a = Action('work', {'time': timePre}, ([{'cash': 100}], [1.]))
#I don't like how this throws, noncompatible compares fail
with self.assertRaises(Exception) as context:
sat = a.satisfied( {'time':None} ) #robust check
self.assertTrue("'descriptor '__ge__' requires a 'int' object but received a 'NoneType'" in context.exception)
def testActionSimple(self):
timePre = Prereq(int.__ge__, 30)
a = Action('work', {'time': timePre}, ([{'cash': 100}], [1.]))
self.assertTrue( a.satisfied({'time':50}) )
self.assertFalse( a.satisfied({'dog':True}) ) #no satisfying time entry
self.assertFalse( a.satisfied(None) ) #robsut check
self.assertEqual(a.cost(), 1)
self.assertEqual(str(a), 'Action(work)')
def test_score_successor(self):
utility_funcs = {'cash': lambda x: x}
action = Action('work', {}, ([{'cash': 100}], [1.]))
goal = Goal('money', {'cash': Prereq(operator.ge, 200)}, ([{
'cash': 1000
}], [1.]))
agent = PlanningAgent({'cash': 0}, [action], [goal], utility_funcs)
from_state = {'cash': 0}
state_a = {'cash': -100}
state_b = {'cash': 100}
state_c = {'cash': 200}
state_d = {'cash': 300}
self.assertTrue(
agent._score_successor(from_state, state_a) <
agent._score_successor(from_state, state_b))
self.assertTrue(
agent._score_successor(from_state, state_b) <
agent._score_successor(from_state, state_c))
self.assertTrue(
agent._score_successor(from_state, state_c) <
agent._score_successor(from_state, state_d))
def testActionEmpty(self):
a = Action(None, None, None)
#no-prereqs, any value OK
self.assertTrue(a.satisfied(None))
self.assertTrue(a.satisfied({'dog': True}))