Improve docs in tests