Improve docstring in tests