Improve tests for documentation