change test file

tensnsorflow/test.py

 # -*- coding: utf-8 -*-
 from pyspark.context import SparkContext
 from pyspark.conf import SparkConf
-# import pytispark.pytispark as pti
+import pytispark.pytispark as pti
 from pyspark.sql import SparkSession
 
 
@@ -37,11 +37,11 @@ if __name__ == '__main__':
                 .set("spark.driver.maxResultSize", "8g")
 
         spark = SparkSession.builder.config(conf=sparkConf).enableHiveSupport().getOrCreate()
-        # ti = pti.TiContext(spark)
-        # ti.tidbMapDatabase("jerry_test")
-        # spark.sparkContext.setLogLevel("WARN")
-        # sql = "select stat_date,cid_id,y,ccity_name from esmm_train_data limit 60"
-        # spark.sql(sql).show(6)
+        ti = pti.TiContext(spark)
+        ti.tidbMapDatabase("jerry_test")
+        spark.sparkContext.setLogLevel("WARN")
+        sql = "select stat_date,cid_id,y,ccity_name from esmm_train_data limit 60"
+        spark.sql(sql).show(6)
 
         spark.sql("use online")
         spark.sql("ADD JAR hdfs:///user/hive/share/lib/udf/brickhouse-0.7.1-SNAPSHOT.jar")