Use TestOptimizationMixin to provide better error messages when the op is not properly optimized.