Add docstrings to more tests

:+1: