Fix assertion rewriting of ``all()`` calls to deal with non-generators.