initial commit

kiszk · kiszk · commit 3e4438de4faf · 2017-07-23T18:33:05.000+09:00
diff --git a/sql/core/src/test/scala/org/apache/spark/sql/DatasetCacheSuite.scala b/sql/core/src/test/scala/org/apache/spark/sql/DatasetCacheSuite.scala
@@ -43,6 +43,8 @@ class DatasetCacheSuite extends QueryTest with SharedSQLContext {
     // joined Dataset should not be persisted
     val joined = ds1.joinWith(ds2, $"a.value" === $"b.value")
     assert(joined.storageLevel == StorageLevel.NONE)
+    ds1.unpersist()
+    ds2.unpersist()
   }
 
   test("persist and unpersist") {
@@ -59,6 +61,7 @@ class DatasetCacheSuite extends QueryTest with SharedSQLContext {
     // Drop the cache.
     cached.unpersist()
     assert(cached.storageLevel == StorageLevel.NONE, "The Dataset should not be cached.")
+    ds.unpersist()
   }
 
   test("persist and then rebind right encoder when join 2 datasets") {

Original file line number	Diff line number	Diff line change
`@@ -43,6 +43,8 @@ class DatasetCacheSuite extends QueryTest with SharedSQLContext {`
`43`	`43`	`// joined Dataset should not be persisted`
`44`	`44`	`val joined = ds1.joinWith(ds2, $"a.value" === $"b.value")`
`45`	`45`	`assert(joined.storageLevel == StorageLevel.NONE)`
	`46`	`+ ds1.unpersist()`
	`47`	`+ ds2.unpersist()`
`46`	`48`	`}`
`47`	`49`
`48`	`50`	`test("persist and unpersist") {`
`@@ -59,6 +61,7 @@ class DatasetCacheSuite extends QueryTest with SharedSQLContext {`
`59`	`61`	`// Drop the cache.`
`60`	`62`	`cached.unpersist()`
`61`	`63`	`assert(cached.storageLevel == StorageLevel.NONE, "The Dataset should not be cached.")`
	`64`	`+ ds.unpersist()`
`62`	`65`	`}`
`63`	`66`
`64`	`67`	`test("persist and then rebind right encoder when join 2 datasets") {`