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