Fix up test docstrings