Added docstrings to unit tests