Mixin version

jklymak · jklymak · commit 01049f1158e5 · 2021-01-05T15:32:55.000-08:00
diff --git a/lib/matplotlib/scale.py b/lib/matplotlib/scale.py
@@ -79,14 +79,18 @@ def limit_range_for_scale(self, vmin, vmax, minpos):
         return vmin, vmax
 
 
-class TransformScaleBase(Transform):
-    """Base transform for scales to allow a deep copy"""
+class CopyableTransformMixin():
+    """
+    Mixin to support copy and deep copy on transforms.  This alows scales,
+    and hence norms, to be copyable.
+    """
     def __deepcopy__(self, memo):
         return self.frozen()
 
+    __copy__ = __deepcopy__
+
 
-class LinearTransform(TransformScaleBase, IdentityTransform):
-    """Mixed class for Identity"""
+class LinearTransform(CopyableTransformMixin, IdentityTransform):
     pass
 
 
@@ -124,9 +128,9 @@ def get_transform(self):
         return LinearTransform()
 
 
-class FuncTransform(TransformScaleBase):
+class FuncTransform(CopyableTransformMixin, Transform):
     """
-    A simple transform that takes and arbitrary function for the
+    A transform that takes and arbitrary function for the
     forward and inverse transform.
     """
 
@@ -202,7 +206,7 @@ def set_default_locators_and_formatters(self, axis):
             axis.set_minor_locator(NullLocator())
 
 
-class LogTransform(TransformScaleBase):
+class LogTransform(CopyableTransformMixin, Transform):
     input_dims = output_dims = 1
 
     @_api.rename_parameter("3.3", "nonpos", "nonpositive")
@@ -244,7 +248,7 @@ def inverted(self):
         return InvertedLogTransform(self.base)
 
 
-class InvertedLogTransform(TransformScaleBase):
+class InvertedLogTransform(CopyableTransformMixin, Transform):
     input_dims = output_dims = 1
 
     def __init__(self, base):
@@ -370,7 +374,7 @@ def get_transform(self):
         return self._transform
 
 
-class SymmetricalLogTransform(TransformScaleBase):
+class SymmetricalLogTransform(CopyableTransformMixin, Transform):
     input_dims = output_dims = 1
 
     def __init__(self, base, linthresh, linscale):
@@ -402,7 +406,7 @@ def inverted(self):
                                                self.linscale)
 
 
-class InvertedSymmetricalLogTransform(TransformScaleBase):
+class InvertedSymmetricalLogTransform(CopyableTransformMixin, Transform):
     input_dims = output_dims = 1
 
     def __init__(self, base, linthresh, linscale):
@@ -505,7 +509,7 @@ def get_transform(self):
         return self._transform
 
 
-class LogitTransform(TransformScaleBase):
+class LogitTransform(CopyableTransformMixin, Transform):
     input_dims = output_dims = 1
 
     @_api.rename_parameter("3.3", "nonpos", "nonpositive")
@@ -531,7 +535,7 @@ def __str__(self):
         return "{}({!r})".format(type(self).__name__, self._nonpositive)
 
 
-class LogisticTransform(TransformScaleBase):
+class LogisticTransform(CopyableTransformMixin, Transform):
     input_dims = output_dims = 1
 
     @_api.rename_parameter("3.3", "nonpos", "nonpositive")
diff --git a/lib/matplotlib/tests/test_colors.py b/lib/matplotlib/tests/test_colors.py
@@ -1269,7 +1269,6 @@ def test_repr_html():
     png = cmap._repr_png_()
     assert base64.b64encode(png).decode('ascii') in html
     assert cmap.name in html
-<<<<<<< HEAD
     assert html.startswith('<div')
     assert html.endswith('</div>')
 
