Enable registrationRequired, which would have caught this bug.

apache · Dec 29, 2014 · d5b409f · d5b409f
1 parent 069d8a7
commit d5b409f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/core/src/test/scala/org/apache/spark/api/python/PythonBroadcastSuite.scala b/core/src/test/scala/org/apache/spark/api/python/PythonBroadcastSuite.scala
@@ -48,8 +48,9 @@ class PythonBroadcastSuite extends FunSuite with Matchers {
       }
       val broadcast = new PythonBroadcast(broadcastDataFile.getAbsolutePath)
       assertBroadcastIsValid(broadcast)
+      val conf = new SparkConf().set("spark.kryo.registrationRequired", "true")
       val deserializedBroadcast =
-        Utils.clone[PythonBroadcast](broadcast, new KryoSerializer(new SparkConf()).newInstance())
+        Utils.clone[PythonBroadcast](broadcast, new KryoSerializer(conf).newInstance())
       assertBroadcastIsValid(deserializedBroadcast)
     } finally {
       Utils.deleteRecursively(tempDir)