Improving documentation in tests