Add docstrings to more tests