Add missing docstring to unit test