aboutsummaryrefslogtreecommitdiff
path: root/python/pyspark/serializers.py
diff options
context:
space:
mode:
Diffstat (limited to 'python/pyspark/serializers.py')
-rw-r--r--python/pyspark/serializers.py6
1 files changed, 6 insertions, 0 deletions
diff --git a/python/pyspark/serializers.py b/python/pyspark/serializers.py
index 33aa55f7f1..bd08c9a6d2 100644
--- a/python/pyspark/serializers.py
+++ b/python/pyspark/serializers.py
@@ -463,6 +463,9 @@ class CompressedSerializer(FramedSerializer):
def loads(self, obj):
return self.serializer.loads(zlib.decompress(obj))
+ def __eq__(self, other):
+ return isinstance(other, CompressedSerializer) and self.serializer == other.serializer
+
class UTF8Deserializer(Serializer):
@@ -489,6 +492,9 @@ class UTF8Deserializer(Serializer):
except EOFError:
return
+ def __eq__(self, other):
+ return isinstance(other, UTF8Deserializer) and self.use_unicode == other.use_unicode
+
def read_long(stream):
length = stream.read(8)