Add docstrings to tests

> I think this refactor is an improvement

So do I, thanks `:)`