Add docstrings to tests

I think this refactor is an improvement, but if you disagree or think it shouldn't be in this pull-request I can easily remove it.