add docstring to test