def draw_board(dis: pygame.display, size: int, my_snake: Snake, my_snack: Snack): dis.fill((30, 32, 40)) # dark blue snake_pos_list = my_snake.get_pos() snack_pos = my_snack.get_pos() increment = min(display.get_size()) / size for y in range(size): for x in range(size): tile_size = increment - 2 tile_pos = (x * increment, (-y + size - 1) * increment) tile = Tile(dis, tile_size, tile_pos) if (x, y) in snake_pos_list: tile.change_state('SNAKE') elif (x, y) == snack_pos: tile.change_state('SNACK') else: tile.change_state('BLANK') tile.draw() pygame.display.update()
def test_change_state(): """test Tile change_state method""" tile = Tile(1, 1) tile.change_state("PLAYER") assert tile.current_state == "PLAYER"