def test_put(self): world = World() location = Location(2, 3) world.put_cell(location, CellType.EMPTY) self.assertEqual(CellType.EMPTY, world.get_cell(location)) world.put_cell(location, CellType.FOOD) self.assertEqual(CellType.FOOD, world.get_cell(location)) result = world.put_cell(location, CellType.EMPTY) self.assertFalse(result)
def test_update_when_is_not_free(self): world = World() world.put_cell(Location(1, 1), CellType.FOOD) old_value = world.update_cell(Location(1, 1), CellType.POISON) self.assertEqual(old_value, CellType.FOOD)
def test_put_when_out_of_range(self): world = World() with self.assertRaises(Exception): world.put_cell(Location(999999, 988888), CellType.POISON)
def test_put_when_not_empty(self): world = World() world.put_cell(Location(1, 1), CellType.FOOD) result = world.put_cell(Location(1, 1), CellType.FOOD) self.assertFalse(result)
def test_put_empty_when_empty(self): world = World() location = Location(1, 1) world.put_cell(location, CellType.EMPTY) world.put_cell(location, CellType.EMPTY)
def test_put_on_wall(self): world = World() location = Location(0, 0) result = world.put_cell(location, CellType.POISON) self.assertFalse(result)
def test_get(self): world = World() world.put_cell(Location(1, 3), CellType.POISON) result = world.get_cell(Location(1, 3)) self.assertEqual(result, CellType.POISON)