Fix docstrings in test_api