def METHOD_NAME(self):
    """Test that model inference infers model output type."""
    model = """
        <
            ir_version: 7,
            opset_import: [ "" : 17]
        >
        agraph (float[N] x) => (y)
        {
            y = Cast<to=6> (x)
        }
    """
    self._check(model, onnx.TensorProto.INT32)