Using docstrings for testing