Add tests for the documentation