Improve documentation tests