Improved testing for new pretty print