def test_update_explosion_in_matrix(self):
game_display = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
arena = Arena()
arena.arena_matrix = self.get_default_arena()
bomb = Bomb((22, 0), 2, arena)
bomb.state = BombState.EXPLODING
arena.update_explosion_in_matrix(bomb)
matrix = arena.get_arena()
self.assertEqual(matrix[1][1], TileType.FLAME)
def test_get_arena(self):
game_display = pygame.display.set_mode((SCREEN_WIDTH, SCREEN_HEIGHT))
arena = Arena()
arena_matrix = arena.arena_matrix
self.assertEqual(arena.get_arena(), arena_matrix)
