TST: better assertion messages on test failures

CLN: pep8