diff --git a/sql/core/src/test/scala/org/apache/spark/sql/execution/UnsafeExternalSortSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/execution/UnsafeExternalSortSuite.scala index f4b8782e39b03..c5ab74ad72266 100644 --- a/sql/core/src/test/scala/org/apache/spark/sql/execution/UnsafeExternalSortSuite.scala +++ b/sql/core/src/test/scala/org/apache/spark/sql/execution/UnsafeExternalSortSuite.scala @@ -40,12 +40,12 @@ class UnsafeExternalSortSuite extends SparkPlanTest with BeforeAndAfterAll { // TODO: randomized spilling to ensure that merging is tested at least once for every data type. for ( dataType <- DataTypeTestUtils.atomicTypes ++ Set(NullType); - nullable <- Seq(true, false); - sortOrder <- Seq('a.asc :: Nil, 'a.desc :: Nil); + nullable <- Seq(false); + sortOrder <- Seq('a.asc :: Nil); randomDataGenerator <- RandomDataGenerator.forType(dataType, nullable) ) { test(s"sorting on $dataType with nullable=$nullable, sortOrder=$sortOrder") { - val inputData = Seq.fill(3)(randomDataGenerator()).filter { + val inputData = Seq.fill(10)(randomDataGenerator()).filter { case d: Double => !d.isNaN case f: Float => !java.lang.Float.isNaN(f) case x => true @@ -57,7 +57,7 @@ class UnsafeExternalSortSuite extends SparkPlanTest with BeforeAndAfterAll { assert(UnsafeExternalSort.supportsSchema(inputDf.schema)) checkAnswer( inputDf, - UnsafeExternalSort(sortOrder, global = false, _: SparkPlan, testSpillFrequency = 2), + UnsafeExternalSort(sortOrder, global = false, _: SparkPlan, testSpillFrequency = 3), Sort(sortOrder, global = false, _: SparkPlan), sortAnswers = false )