better message for assert.

apache · Aug 10, 2016 · 288b699 · 288b699
1 parent 14f4959
commit 288b699
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/SQLQueryTestSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/SQLQueryTestSuite.scala
@@ -150,7 +150,11 @@ class SQLQueryTestSuite extends QueryTest with SharedSQLContext {
     val expectedOutputs: Seq[QueryOutput] = {
       val goldenOutput = fileToString(new File(testCase.resultFile))
       val segments = goldenOutput.split("-- !query.+\n")
-      assert(segments.size == outputs.size * 3 + 1)  // each query has 3 segments, plus the header
+
+      // each query has 3 segments, plus the header
+      assert(segments.size == outputs.size * 3 + 1,
+        s"Expected ${outputs.size * 3 + 1} blocks in result file but got ${segments.size}. " +
+        s"Try regenerate the result files.")
       Seq.tabulate(outputs.size) { i =>
         QueryOutput(
           sql = segments(i * 3 + 1).trim,