change unit test to docstring to make assertions about output
change unit test to docstring to make assertions about output