def setUp(self): self.obj = Object('Test') self.body = Big(2) self.obj.add_component(self.body) create_celltype('Floor', False, False) create_celltype('Wall', True, False) self.map = create_map('Test', 10, 10, 'Floor')
class SimpleTest(unittest.TestCase): def setUp(self): self.obj = Object('Test') self.body = Big(2) self.obj.add_component(self.body) create_celltype('Floor', False, False) create_celltype('Wall', True, False) self.map = create_map('Test', 10, 10, 'Floor') def check(self, x, y): self.assertEqual(self.body.x, x) self.assertEqual(self.body.y, y) self.assertEquals(len(self.body.occupied_cells), 4) for j in range(0, 9): for k in range(0, 9): cell = self.map.get_cell(j, k) if (j is x or j is x + 1) and (k is y or k is y + 1): self.assertTrue(cell in self.body.occupied_cells) self.assertTrue(self.obj in cell.objects) else: self.assertFalse(cell in self.body.occupied_cells) self.assertFalse(self.obj in cell.objects) def test_get_occupied_cells(self): self.map = create_map('Test', 10, 10, 'Floor') for x in range(0, 9): for y in range(0, 9): self.body.add_to_map(self.map, x, y) cells = self.body.get_occupied_cells() self.assertEquals(len(cells), 4) self.assertTrue(self.map.get_cell(x, y) in cells) self.assertTrue(self.map.get_cell(x, y + 1) in cells) self.assertTrue(self.map.get_cell(x + 1, y) in cells) self.assertTrue(self.map.get_cell(x + 1, y + 1) in cells) # add def test_add_to_map(self): self.map = create_map('Test', 10, 10, 'Floor') # inside for x in range(0, 9): for y in range(0, 9): self.assertTrue(self.body.add_to_map(self.map, x, y)) self.check(x, y) # outside self.assertFalse(self.body.add_to_map(self.map, -1, -1)) self.assertFalse(self.body.add_to_map(self.map, 9, -1)) self.assertFalse(self.body.add_to_map(self.map, -1, 9)) self.assertFalse(self.body.add_to_map(self.map, 9, 9)) self.check(8, 8) # solid self.map.get_cell(5, 5).celltype = get_celltype('Wall') self.assertFalse(self.body.add_to_map(self.map, 5, 5)) self.assertFalse(self.body.add_to_map(self.map, 4, 5)) self.assertFalse(self.body.add_to_map(self.map, 5, 4)) self.assertFalse(self.body.add_to_map(self.map, 4, 4)) self.check(8, 8) def test_map_add_object(self): self.map = create_map('Test', 10, 10, 'Floor') # inside for x in range(0, 9): for y in range(0, 9): self.assertTrue(self.map.add_object(self.obj, x, y)) self.check(x, y) # outside self.assertFalse(self.map.add_object(self.obj, -1, 0)) self.assertFalse(self.map.add_object(self.obj, 0, -1)) self.assertFalse(self.map.add_object(self.obj, 10, 9)) self.assertFalse(self.map.add_object(self.obj, 9, 10)) self.check(8, 8) # solid self.map.get_cell(5, 5).celltype = get_celltype('Wall') self.assertFalse(self.map.add_object(self.obj, 5, 5)) self.assertFalse(self.map.add_object(self.obj, 4, 5)) self.assertFalse(self.map.add_object(self.obj, 5, 4)) self.assertFalse(self.map.add_object(self.obj, 4, 4)) self.check(8, 8) # move def test_move(self): self.map = create_map('Test', 10, 10, 'Floor') self.map.add_object(self.obj, 0, 0) # inside self.assertTrue(self.body.move(1)) self.check(1, 0) self.assertTrue(self.body.move(0)) self.check(1, 1) self.assertTrue(self.body.move(3)) self.check(0, 1) self.assertTrue(self.body.move(2)) self.check(0, 0) # outside self.assertFalse(self.body.move(3)) self.check(0, 0) self.assertFalse(self.body.move(2)) self.check(0, 0) self.map.add_object(self.obj, 8, 8) self.assertFalse(self.body.move(1)) self.check(8, 8) self.assertFalse(self.body.move(0)) self.check(8, 8) # solid self.map.get_cell(5, 5).celltype = get_celltype('Wall') self.map.add_object(self.obj, 3, 5) self.assertFalse(self.body.move(1)) self.check(3, 5) self.map.add_object(self.obj, 3, 4) self.assertFalse(self.body.move(1)) self.check(3, 4) self.map.add_object(self.obj, 6, 5) self.assertFalse(self.body.move(3)) self.check(6, 5) self.map.add_object(self.obj, 6, 4) self.assertFalse(self.body.move(3)) self.check(6, 4) self.map.add_object(self.obj, 5, 3) self.assertFalse(self.body.move(0)) self.check(5, 3) self.map.add_object(self.obj, 4, 3) self.assertFalse(self.body.move(0)) self.check(4, 3) self.map.add_object(self.obj, 5, 6) self.assertFalse(self.body.move(2)) self.check(5, 6) self.map.add_object(self.obj, 4, 6) self.assertFalse(self.body.move(2)) self.check(4, 6) """def test_objects_move(self): self.map = create_map('Test', 10, 10, 'Floor') self.map.add_object(self.obj, 0, 0) # inside self.assertTrue(self.obj.move(1)) self.check(1, 0) self.assertTrue(self.obj.move(0)) self.check(1, 1) self.assertTrue(self.obj.move(3)) self.check(0, 1) self.assertTrue(self.obj.move(2)) self.check(0, 0) # outside self.assertFalse(self.obj.move(3)) self.check(0, 0) self.assertFalse(self.obj.move(2)) self.check(0, 0) self.map.add_object(self.obj, 8, 8) self.assertFalse(self.obj.move(1)) self.check(8, 8) self.assertFalse(self.obj.move(0)) self.check(8, 8) # solid self.map.get_cell(5, 5).celltype = get_celltype('Wall') self.map.add_object(self.obj, 3, 5) self.assertFalse(self.obj.move(1)) self.check(3, 5) self.map.add_object(self.obj, 3, 4) self.assertFalse(self.obj.move(1)) self.check(3, 4) self.map.add_object(self.obj, 6, 5) self.assertFalse(self.obj.move(3)) self.check(6, 5) self.map.add_object(self.obj, 6, 4) self.assertFalse(self.obj.move(3)) self.check(6, 4) self.map.add_object(self.obj, 5, 3) self.assertFalse(self.obj.move(0)) self.check(5, 3) self.map.add_object(self.obj, 4, 3) self.assertFalse(self.obj.move(0)) self.check(4, 3) self.map.add_object(self.obj, 5, 6) self.assertFalse(self.obj.move(2)) self.check(5, 6) self.map.add_object(self.obj, 4, 6) self.assertFalse(self.obj.move(2)) self.check(4, 6)""" # remove def test_remove_from_map(self): self.map = create_map('Test', 10, 10, 'Floor') for x in range(0, 9): for y in range(0, 9): self.body.add_to_map(self.map, x, y) self.body.remove_from_map() self.assertEqual(self.body.map, None) self.assertEqual(self.body.x, None) self.assertEqual(self.body.y, None) self.assertEquals(len(self.body.occupied_cells), 0) self.assertFalse(self.obj in self.map.get_cell(x, y).objects) self.assertFalse(self.obj in self.map.get_cell(x, y + 1).objects) self.assertFalse(self.obj in self.map.get_cell(x + 1, y).objects) self.assertFalse(self.obj in self.map.get_cell(x + 1, y + 1).objects) for x in range(0, 10): for y in range(0, 10): self.assertFalse(self.obj in self.map.get_cell(x, y).objects)