add test for analyzeIR

Author: weiguoz

pkg/sql/ir_generator.go

@@ -72,6 +72,21 @@ func generateTrainIR(slct *extendedSelect, connStr string) (*codegen.TrainIR, er
 	}, nil
 }
 
+func generateAnalyzeIR(slct *extendedSelect, connStr string) (*codegen.AnalyzeIR, error) {
+	attrs, err := generateAttributeIR(&slct.analyzeAttrs)
+	if err != nil {
+		return nil, err
+	}
+	return &codegen.AnalyzeIR{
+		DataSource: connStr,
+		Select:     slct.standardSelect.String(),
+		Attributes: attrs,
+		Explainer:  slct.explainer,
+		// TrainIR is the TrainIR used for generating the training job of the corresponding model
+		// TrainIR TrainIR
+	}, nil
+}
+
 func generateAttributeIR(attrs *attrs) (map[string]interface{}, error) {
 	ret := make(map[string]interface{})
 	for k, v := range *attrs {

pkg/sql/ir_generator_test.go

@@ -148,3 +148,25 @@ INTO mymodel;`
 	a.Equal(10000, catCol.FieldMeta.Shape[0])
 	a.Equal(",", catCol.FieldMeta.Delimiter)
 }
+
+func TestGenerateAnalyzeIR(t *testing.T) {
+	a := assert.New(t)
+	stmt := `
+	SELECT *
+	FROM iris.train
+	ANALYZE sqlflow_models.my_xgboost_model
+	WITH
+	    shap_summary.plot_type="bar",
+	    shap_summary.alpha=1,
+	    shap_summary.sort=True
+	USING TreeExplainer;
+	`
+	pr, e := newParser().Parse(stmt)
+	a.NoError(e)
+
+	connStr := "mysql://root:root@tcp(localhost)"
+	ir, e := generateAnalyzeIR(pr, connStr)
+	a.NoError(e)
+	a.Equal(ir.Explainer, "TreeExplainer")
+	a.Equal(ir.DataSource, connStr)
+}