ну у меня был какой-то такой косяк, но то вроде только на симуляторе глючило.
в любом случае, был такой кусок кода которым вроде получалось переключать - я правда уже забыл нафиг как это работало :))
let keyboard = app.keyboards.element(boundBy: 0)
let nextKeyboardButton = app/*
@START_MENU_TOKEN@*/.buttons["Next keyboard"]/*[[".keyboards.buttons[\"Next keyboard\"]",".buttons[\"Next keyboard\"]"],[[[-1,1],[-1,0]]],[0]]
@END_MENU_TOKEN@*/
while keyboard.buttons.element(boundBy: keyboard.buttons.count - 1).label != "done" {
nextKeyboardButton.tap()
}