add more test

wayneguow · Jul 26, 2024 · 5fb5e62 · 5fb5e62
1 parent 2f54c99
commit 5fb5e62
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/StringFunctionsSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/StringFunctionsSuite.scala
@@ -426,6 +426,7 @@ class StringFunctionsSuite extends QueryTest with SharedSparkSession {
     )
 
     val testTable = "test_substring_index"
+    // count is int
     withTempPath { path =>
       withTable(testTable) {
         sql(s"CREATE TABLE $testTable (num int) USING parquet LOCATION '${path.toURI.toString}'")
@@ -434,6 +435,18 @@ class StringFunctionsSuite extends QueryTest with SharedSparkSession {
         checkAnswer(sql(query), Seq(Row(1, "a"), Row(2, "a_a"), Row(3, "a_a_a"), Row(null, null)))
       }
     }
+    // count is string
+    withTempPath { path =>
+      withTable(testTable) {
+        sql(s"CREATE TABLE $testTable (num string) USING parquet " +
+          s"LOCATION '${path.toURI.toString}'")
+        sql(s"INSERT INTO $testTable VALUES ('1'), ('2'), ('3'), (NULL)")
+        val query = s"SELECT num, SUBSTRING_INDEX('a_a_a', '_', num) as sub_str FROM $testTable"
+        checkAnswer(
+          sql(query),
+          Seq(Row("1", "a"), Row("2", "a_a"), Row("3", "a_a_a"), Row(null, null)))
+      }
+    }
   }
 
   test("string locate function") {