Fixed unit tests by making them clean up the SparkContext after use and

thus clean up the various singletons (RDDCache, MapOutputTracker, etc).
This isn't perfect yet (ideally we shouldn't use singleton objects at
all) but we can fix that later.

1 files changed, 2 insertions, 0 deletions

diff --git a/bagel/src/test/scala/bagel/BagelSuite.scala b/bagel/src/test/scala/bagel/BagelSuite.scala
index 1b47fc9ed5..9e64d3f136 100644
--- a/bagel/src/test/scala/bagel/BagelSuite.scala
+++ b/bagel/src/test/scala/bagel/BagelSuite.scala
@@ -28,6 +28,7 @@ class BagelSuite extends FunSuite with Assertions {
     })
     for (vert <- result.collect)
       assert(vert.age === numSupersteps)
+    sc.stop()
   }
 
   test("halting by message silence") {
@@ -49,5 +50,6 @@ class BagelSuite extends FunSuite with Assertions {
     })
     for (vert <- result.collect)
       assert(vert.age === numSupersteps)
+    sc.stop()
   }
 }