Add docstrings for tests