External unsafe row dequeue test extension

apache · Aug 10, 2018 · 62a8071 · 62a8071
1 parent 6a8ff5d
commit 62a8071
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/...src/test/scala/org/apache/spark/sql/execution/ExternalAppendOnlyUnsafeRowArraySuite.scala b/...src/test/scala/org/apache/spark/sql/execution/ExternalAppendOnlyUnsafeRowArraySuite.scala
@@ -292,7 +292,7 @@ class ExternalAppendOnlyUnsafeRowArraySuite extends SparkFunSuite with LocalSpar
   test(s"test dequeue with spill") {
     withExternalArray(inMemoryThreshold = 2, spillThreshold = 3, true) { array =>
       // insert 2 rows, iterate until the first row
-      populateRows(array, 10)
+      populateRows(array, 5)
       assertSpill()
 
       var iterator = array.generateIterator()
@@ -308,14 +308,16 @@ class ExternalAppendOnlyUnsafeRowArraySuite extends SparkFunSuite with LocalSpar
       val third2 = array.dequeue().get
       assert(third.equals(third2))
 
-      assert(array.length == 7)
+      assert(array.length == 2)
+
+      array.dequeue()
 
       populateRows(array, 10)
 
       array.dequeue()
       array.dequeue()
 
-      assert(array.length == 15)
+      assert(array.length == 9)
     }
   }