remove TO before TRAIN/ANALYZE in ut

Author: weiguoz

pkg/sql/ir_generator_test.go

@@ -158,9 +158,11 @@ func TestGeneratePredictIR(t *testing.T) {
 	}
 	a := assert.New(t)
 	parser := newParser()
-	predSQL := `SELECT * FROM iris.test
-TO PREDICT iris.predict.class
-USING sqlflow_models.mymodel;`
+	predSQL := `
+	SELECT * FROM iris.test
+	PREDICT iris.predict.class
+	USING sqlflow_models.mymodel;
+	`
 	r, e := parser.Parse(predSQL)
 	a.NoError(e)
 
@@ -170,12 +172,14 @@ USING sqlflow_models.mymodel;`
 	modelDir, e := ioutil.TempDir("/tmp", "sqlflow_models")
 	a.Nil(e)
 	defer os.RemoveAll(modelDir)
-	stream := runExtendedSQL(`SELECT * FROM iris.train
-TO TRAIN DNNClassifier
-WITH model.n_classes=3, model.hidden_units=[10,20]
-COLUMN sepal_length, sepal_width, petal_length, petal_width
-LABEL class
-INTO sqlflow_models.mymodel;`, testDB, modelDir, nil)
+	stream := runExtendedSQL(`
+	SELECT * FROM iris.train
+	TRAIN DNNClassifier
+	WITH model.n_classes=3, model.hidden_units=[10,20]
+	COLUMN sepal_length, sepal_width, petal_length, petal_width
+	LABEL class
+	INTO sqlflow_models.mymodel;
+	`, testDB, modelDir, nil)
 	a.True(goodStream(stream.ReadAll()))
 
 	// Test generate PredicrIR
@@ -206,7 +210,7 @@ func TestGenerateAnalyzeIR(t *testing.T) {
 	stream := runExtendedSQL(`
 	SELECT *
 	FROM iris.train
-	TO TRAIN xgboost.gbtree
+	TRAIN xgboost.gbtree
 	WITH
 	    objective="multi:softprob",
 	    train.num_boost_round = 30,