Skip to content

Commit 0649987

Browse files
committed
add test
1 parent 4756fad commit 0649987

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

sql/core/src/test/scala/org/apache/spark/sql/CachedTableSuite.scala

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,9 +22,19 @@ import org.apache.spark.sql.columnar.{InMemoryRelation, InMemoryColumnarTableSca
2222
import org.apache.spark.sql.test.TestSQLContext
2323
import org.apache.spark.sql.test.TestSQLContext._
2424

25+
case class BigData(s: String)
26+
2527
class CachedTableSuite extends QueryTest {
2628
TestData // Load test tables.
2729

30+
test("too big for memory") {
31+
val data = "*" * 10000
32+
sparkContext.parallelize(1 to 1000000, 1).map(_ => BigData(data)).registerTempTable("bigData")
33+
cacheTable("bigData")
34+
assert(table("bigData").count() === 1000000L)
35+
uncacheTable("bigData")
36+
}
37+
2838
test("SPARK-1669: cacheTable should be idempotent") {
2939
assume(!table("testData").logicalPlan.isInstanceOf[InMemoryRelation])
3040

0 commit comments

Comments
 (0)