Make backend_pdf.Name comparable and hashable

jkseppan · jkseppan · commit ee609055f865 · 2016-05-15T16:37:25.000+03:00
diff --git a/lib/matplotlib/backends/backend_pdf.py b/lib/matplotlib/backends/backend_pdf.py
@@ -18,6 +18,7 @@
 import warnings
 import zlib
 from io import BytesIO
+from functools import total_ordering
 
 import numpy as np
 from matplotlib.externals.six import unichr
@@ -243,6 +244,7 @@ def write(self, contents, file):
         write(b"\nendobj\n")
 
 
+@total_ordering
 class Name(object):
     """PDF name object."""
     __slots__ = ('name',)
@@ -262,6 +264,15 @@ def __repr__(self):
     def __str__(self):
         return '/' + six.text_type(self.name)
 
+    def __eq__(self, other):
+        return isinstance(other, Name) and self.name == other.name
+
+    def __lt__(self, other):
+        return isinstance(other, Name) and self.name < other.name
+
+    def __hash__(self):
+        return hash(self.name)
+
     @staticmethod
     def hexify(match):
         return '#%02x' % ord(match.group())
