Думаю, в тестах можно себе позволить использовать `with`.