from snake_state import SnakeState class SnakeUpMovementState(SnakeState): def _get_next_pos(self, old_pos: tuple, radius: float) -> tuple: return (old_pos[0], old_pos[1]-2*radius)