From d5b409f78adcf3498f48728bc66066a478816d73 Mon Sep 17 00:00:00 2001 From: Josh Rosen Date: Mon, 29 Dec 2014 12:15:53 -0800 Subject: [PATCH] Enable registrationRequired, which would have caught this bug. --- .../org/apache/spark/api/python/PythonBroadcastSuite.scala | 3 ++- 1 file changed, 2 insertions(+), 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 index de07b2c177afe..2805ac91b7eef 100644 --- 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)