More docstrings in tests

This will need to be fixed because of #40