graphframes · SemyonSinchenko · Oct 4, 2025 · Oct 4, 2025
diff --git a/core/src/test/scala/org/graphframes/lib/AggregateMessagesSuite.scala b/core/src/test/scala/org/graphframes/lib/AggregateMessagesSuite.scala
@@ -54,6 +54,7 @@ class AggregateMessagesSuite extends SparkFunSuite with GraphFrameTestSparkConte
         case _: Row => throw new GraphFramesUnreachableException()
       }
       .toMap
+    agg.unpersist()
     // Compute the truth via brute force for comparison.
     val trueAgg: Map[String, Int] = {
       val user2age = g.vertices
@@ -97,6 +98,7 @@ class AggregateMessagesSuite extends SparkFunSuite with GraphFrameTestSparkConte
         case _: Row => throw new GraphFramesUnreachableException()
       }
       .toMap
+    agg2.unpersist()
     // Compare to the true values.
     agg2Map.keys.foreach { case user =>
       assert(agg2Map(user) === trueAgg(user), s"Failure on user $user")
@@ -171,5 +173,8 @@ class AggregateMessagesSuite extends SparkFunSuite with GraphFrameTestSparkConte
     assert(output1 === expectedValues)
     assert(output2 === expectedValues)
     assert(output3 === expectedValues)
+    agg.unpersist()
+    agg2.unpersist()
+    agg3.unpersist()
   }
 }
diff --git a/core/src/test/scala/org/graphframes/lib/DetectingCyclesSuite.scala b/core/src/test/scala/org/graphframes/lib/DetectingCyclesSuite.scala
@@ -28,6 +28,7 @@ class DetectingCyclesSuite extends SparkFunSuite with GraphFrameTestSparkContext
     assert(collected(0) == Seq(1, 2, 3, 1))
     assert(collected(1) == Seq(2, 3, 1, 2))
     assert(collected(2) == Seq(3, 1, 2, 3))
+    res.unpersist()
   }
 
   test("test no cycles") {
@@ -40,6 +41,7 @@ class DetectingCyclesSuite extends SparkFunSuite with GraphFrameTestSparkContext
         .toDF("src", "dst"))
     val res = graph.detectingCycles.setUseLocalCheckpoints(true).run()
     assert(res.count() == 0)
+    res.unpersist()
   }
 
   test("test multiple cycles from one source") {
@@ -65,5 +67,6 @@ class DetectingCyclesSuite extends SparkFunSuite with GraphFrameTestSparkContext
     assert(collected(4) == Seq(2, 5, 1, 2))
     assert(collected(5) == Seq(3, 1, 3))
     assert(collected(6) == Seq(5, 1, 2, 5))
+    res.unpersist()
   }
 }
diff --git a/core/src/test/scala/org/graphframes/lib/LabelPropagationSuite.scala b/core/src/test/scala/org/graphframes/lib/LabelPropagationSuite.scala
@@ -39,5 +39,6 @@ class LabelPropagationSuite extends SparkFunSuite with GraphFrameTestSparkContex
       labels.filter(s"id >= $n").select("label").collect().toSeq.map(_.getLong(0)).toSet
     assert(clique2.size === 1)
     assert(clique1 !== clique2)
+    labels.unpersist()
   }
 }
diff --git a/core/src/test/scala/org/graphframes/lib/PageRankSuite.scala b/core/src/test/scala/org/graphframes/lib/PageRankSuite.scala
@@ -39,6 +39,7 @@ class PageRankSuite extends SparkFunSuite with GraphFrameTestSparkContext {
     TestUtils.testSchemaInvariants(g, pr)
     TestUtils.checkColumnType(pr.vertices.schema, "pagerank", DataTypes.DoubleType)
     TestUtils.checkColumnType(pr.edges.schema, "weight", DataTypes.DoubleType)
+    pr.unpersist()
   }
 
   test("friends graph with personalized PageRank") {
@@ -48,6 +49,7 @@ class PageRankSuite extends SparkFunSuite with GraphFrameTestSparkContext {
     assert(
       gRank === 0.0,
       s"User g (Gabby) doesn't connect with a. So its pagerank should be 0 but we got $gRank.")
+    results.unpersist()
   }
 
   test("graph with three disconnected components") {
@@ -75,5 +77,6 @@ class PageRankSuite extends SparkFunSuite with GraphFrameTestSparkContext {
     val originalIds = v.select("id").collect().map(_.getLong(0)).toSet
     val resultIds = results.vertices.select("id").collect().map(_.getLong(0)).toSet
     assert(originalIds === resultIds, "PageRank results should preserve all vertex IDs")
+    results.unpersist()
   }
 }
diff --git a/core/src/test/scala/org/graphframes/lib/ParallelPersonalizedPageRankSuite.scala b/core/src/test/scala/org/graphframes/lib/ParallelPersonalizedPageRankSuite.scala
@@ -65,6 +65,7 @@ class ParallelPersonalizedPageRankSuite extends SparkFunSuite with GraphFrameTes
     TestUtils.testSchemaInvariants(g, pr)
     TestUtils.checkColumnType(pr.vertices.schema, "pageranks", SQLDataTypes.VectorType)
     TestUtils.checkColumnType(pr.edges.schema, "weight", DataTypes.DoubleType)
+    pr.unpersist()
   }
 
   test("friends graph with parallel personalized PageRank") {
@@ -96,5 +97,6 @@ class ParallelPersonalizedPageRankSuite extends SparkFunSuite with GraphFrameTes
     assert(
       gRank.numNonzeros === 0,
       s"User g (Gabby) doesn't connect with a. So its pagerank should be 0 but we got ${gRank.numNonzeros}.")
+    pr.unpersist()
   }
 }
diff --git a/core/src/test/scala/org/graphframes/lib/SVDPlusPlusSuite.scala b/core/src/test/scala/org/graphframes/lib/SVDPlusPlusSuite.scala
@@ -53,5 +53,6 @@ class SVDPlusPlusSuite extends SparkFunSuite with GraphFrameTestSparkContext {
       }
       .reduce(_ + _) / g.edges.count()
     assert(err <= svdppErr)
+    v2.unpersist()
   }
 }
diff --git a/core/src/test/scala/org/graphframes/lib/ShortestPathsSuite.scala b/core/src/test/scala/org/graphframes/lib/ShortestPathsSuite.scala
@@ -69,6 +69,7 @@ class ShortestPathsSuite extends SparkFunSuite with GraphFrameTestSparkContext {
       DataTypes.createMapType(v2.schema("id").dataType, DataTypes.IntegerType, false))
     val results = v2.collect().toSet
     assert(results === expected)
+    v2.unpersist()
     ()
   }
 
@@ -105,6 +106,7 @@ class ShortestPathsSuite extends SparkFunSuite with GraphFrameTestSparkContext {
       case _ => throw new GraphFramesUnreachableException()
     }
     assert(results.toSet === shortestPaths)
+    v2.unpersist()
   }
 
   test("friends graph") {
@@ -128,6 +130,7 @@ class ShortestPathsSuite extends SparkFunSuite with GraphFrameTestSparkContext {
       }
       .toSet
     assert(results === expected)
+    v.unpersist()
   }
 
   test("friends graph with GraphFrames") {
@@ -151,6 +154,7 @@ class ShortestPathsSuite extends SparkFunSuite with GraphFrameTestSparkContext {
       }
       .toSet
     assert(results === expected)
+    v.unpersist()
   }
   test("Test vertices with column name") {
     val verticeSeq =

diff --git a/core/src/test/scala/org/graphframes/lib/StronglyConnectedComponentsSuite.scala b/core/src/test/scala/org/graphframes/lib/StronglyConnectedComponentsSuite.scala
@@ -37,5 +37,6 @@ class StronglyConnectedComponentsSuite extends SparkFunSuite with GraphFrameTest
     for (Row(id: Long, component: Long, _) <- c.select("id", "component", "value").collect()) {
       assert(id === component)
     }
+    c.unpersist()
   }
 }
diff --git a/core/src/test/scala/org/graphframes/lib/TriangleCountSuite.scala b/core/src/test/scala/org/graphframes/lib/TriangleCountSuite.scala
@@ -43,6 +43,7 @@ class TriangleCountSuite extends SparkFunSuite with GraphFrameTestSparkContext {
         case Row(_: Long, count: Long, _) => assert(count === 1)
         case _: Row => throw new GraphFramesUnreachableException()
       }
+    v2.unpersist()
   }
 
   test("Count two triangles") {
@@ -62,6 +63,7 @@ class TriangleCountSuite extends SparkFunSuite with GraphFrameTestSparkContext {
         }
       case _: Row => throw new GraphFramesUnreachableException()
     }
+    v2.unpersist()
   }
 
   test("Count one triangles with bi-directed edges") {
@@ -80,6 +82,7 @@ class TriangleCountSuite extends SparkFunSuite with GraphFrameTestSparkContext {
         }
       case _: Row => throw new GraphFramesUnreachableException()
     }
+    v2.unpersist()
   }
 
   test("Count a single triangle with duplicate edges") {
@@ -95,6 +98,7 @@ class TriangleCountSuite extends SparkFunSuite with GraphFrameTestSparkContext {
         assert(count === 1)
       case _: Row => throw new GraphFramesUnreachableException()
     }
+    v2.unpersist()
   }
 
   test("Count with dot column name") {
@@ -112,6 +116,7 @@ class TriangleCountSuite extends SparkFunSuite with GraphFrameTestSparkContext {
         case Row(_: Long, count: Long, _) => assert(count === 1)
         case _: Row => throw new GraphFramesUnreachableException()
       }
+    v2.unpersist()
   }
 
   test("Count with backquote in column name") {
@@ -129,6 +134,7 @@ class TriangleCountSuite extends SparkFunSuite with GraphFrameTestSparkContext {
         case Row(_: Long, count: Long, _) => assert(count === 1)
         case _: Row => throw new GraphFramesUnreachableException()
       }
+    v2.unpersist()
   }
 
   test("no triangle") {
@@ -140,5 +146,6 @@ class TriangleCountSuite extends SparkFunSuite with GraphFrameTestSparkContext {
         assert(count === 0)
       case _: Row => throw new GraphFramesUnreachableException()
     }
+    v2.unpersist()
   }
 }