Deleting temp output directory when done

holdenk · Feb 24, 2014 · 0e6f808 · 0e6f808
1 parent 95c7f67
commit 0e6f808
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/core/src/test/scala/org/apache/spark/storage/LargeIteratorSuite.scala b/core/src/test/scala/org/apache/spark/storage/LargeIteratorSuite.scala
@@ -18,6 +18,8 @@ package org.apache.spark.storage
 
 import org.scalatest.FunSuite
 import org.apache.spark.{LocalSparkContext, SparkContext}
+import org.apache.commons.io.FileUtils
+import java.io.File
 
 class Expander(base:String, count:Int) extends Iterator[String] {
   var i = 0;
@@ -51,5 +53,6 @@ class LargeIteratorSuite extends FunSuite with LocalSparkContext {
     ) );
     val out = seeds.flatMap(Expander.expand(_,10000000));
     out.map(_ + "...").persist(StorageLevel.DISK_ONLY).saveAsTextFile("./test.out")
+    FileUtils.deleteDirectory(new File("./test.out"))
   }
 }