Set serialization properties in GraphSuite

2013-10-16 19:21:24 -07:00 · 2013-10-16 19:21:24 -07:00 · af8e461841
parent 57ac9073ae
commit af8e461841
1 changed files with 3 additions and 0 deletions
--- a/graph/src/test/scala/org/apache/spark/graph/GraphSuite.scala
+++ b/graph/src/test/scala/org/apache/spark/graph/GraphSuite.scala
@ -10,6 +10,9 @@ class GraphSuite extends FunSuite with LocalSparkContext {

 //  val sc = new SparkContext("local[4]", "test")

+  System.setProperty("spark.serializer", "org.apache.spark.serializer.KryoSerializer")
+  System.setProperty("spark.kryo.registrator", "org.apache.spark.graph.GraphKryoRegistrator")
+
  test("Graph Creation") {
    withSpark(new SparkContext("local", "test")) { sc =>
      val rawEdges = (0L to 100L).zip((1L to 99L) :+ 0L)