def testLookAround(self): w = World(4, [], [(1, 1), (33, 2)], [(2, 0), (11, 11)]) r = Robot('bob') r.setWorld(w) r.lookAround() r.act() self.assertEqual(r.brain.boxes.keys(), [(1, 1)]) self.assertEqual(r.brain.holes.keys(), [(2, 0)])
def testInit(self): r = Robot('bob') self.assertEqual(r.name, 'bob') self.assertEqual(r.neighbors, []) self.assertEqual(r.position, (0, 0)) r.addNeighbor(3) self.assert_(3 in r.neighbors) r.setWorld('boo') self.assertEqual(r.world, 'boo')
def testWanderingState(self): r = Robot('bob') w = World(4, [], [], []) r.setWorld(w) r.act() self.assertNotEqual(r.position, (0, 0)) r = Robot('bob') w = World(4, [], [(1, 1)], []) r.setWorld(w) r.lookAround() r.act() self.assertEqual(r.brain.isBoxAssigned((1, 1)), True) self.assertEqual(isinstance(r.state, MovingToBoxState), True)