Only rewrite test function if it had the "assert"