[SPARK-8715] ArrayOutOfBoundsException fixed for DataFrameStatSuite.crosstab

brkyvz · yhuai · commit ffc793a6caaf · 2015-06-29T18:48:38.000-07:00
cc yhuai Author: Burak Yavuz <brkyvz@gmail.com> Closes #7100 from brkyvz/ct-flakiness-fix and squashes the following commits: abc299a [Burak Yavuz] change 'to' to until 7e96d7c [Burak Yavuz] ArrayOutOfBoundsException fixed for DataFrameStatSuite.crosstab (cherry picked from commit ecacb1e) Signed-off-by: Yin Huai <yhuai@databricks.com>
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/DataFrameStatSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/DataFrameStatSuite.scala
@@ -78,7 +78,7 @@ class DataFrameStatSuite extends SparkFunSuite  {
     val rows = crosstab.collect()
     rows.foreach { row =>
       val i = row.getString(0).toInt
-      for (col <- 1 to 9) {
+      for (col <- 1 until columnNames.length) {
         val j = columnNames(col).toInt
         assert(row.getLong(col) === expected.getOrElse((i, j), 0).toLong)
       }

Original file line number	Diff line number	Diff line change
`@@ -78,7 +78,7 @@ class DataFrameStatSuite extends SparkFunSuite {`
`78`	`78`	`val rows = crosstab.collect()`
`79`	`79`	`rows.foreach { row =>`
`80`	`80`	`val i = row.getString(0).toInt`
`81`		`- for (col <- 1 to 9) {`
	`81`	`+ for (col <- 1 until columnNames.length) {`
`82`	`82`	`val j = columnNames(col).toInt`
`83`	`83`	`assert(row.getLong(col) === expected.getOrElse((i, j), 0).toLong)`
`84`	`84`	`}`