Единственное что я бы добавил это если у тебя отчеты в аллюр, и ты не хочешь чтобы при рейзе от недождавшегося wait у тебя тест красился желтым (broken), сделай обертку над вейтом и возвращай True/False и дергай ассерт на это, тогда будет красить в красный