Ещё вариант -- можно было использовать ваши декораторы `with_setup`, `with_teardown`.