Simplify test and add to API docs