Adapt doctest output to new version of numpy