ENH: Update view limits on units change

dopplershift · dopplershift · commit e48f14e58d8d · 2017-08-23T15:35:01.000-06:00
This adds a independent callback for units changes that both updates the
data limits (relim) as well as the view limits (autoscale_view).
diff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py
@@ -557,11 +557,11 @@ def __init__(self, fig, rect,
 
         if self.xaxis is not None:
             self._xcid = self.xaxis.callbacks.connect('units finalize',
-                                                      self.relim)
+                                                      self._on_units_changed)
 
         if self.yaxis is not None:
             self._ycid = self.yaxis.callbacks.connect('units finalize',
-                                                      self.relim)
+                                                      self._on_units_changed)
 
         self.tick_params(
             top=rcParams['xtick.top'] and rcParams['xtick.minor.top'],
@@ -1891,6 +1891,15 @@ def add_container(self, container):
         container.set_remove_method(lambda h: self.containers.remove(h))
         return container
 
+    def _on_units_changed(self):
+        """
+        Callback for processing changes to axis units.
+
+        Currently forces updates of data limits and view limits.
+        """
+        self.relim()
+        self.autoscale_view()
+
     def relim(self, visible_only=False):
         """
         Recompute the data limits based on current artists. If you want to
diff --git a/lib/matplotlib/tests/test_units.py b/lib/matplotlib/tests/test_units.py
@@ -76,7 +76,6 @@ def convert(value, unit, axis):
     ax.axvline(Quantity(120, 'minutes'), color='tab:green')
     ax.yaxis.set_units('inches')
     ax.xaxis.set_units('seconds')
-    ax.autoscale_view()
 
     assert qc.convert.called
     assert qc.axisinfo.called
diff --git a/lib/mpl_toolkits/mplot3d/axes3d.py b/lib/mpl_toolkits/mplot3d/axes3d.py
@@ -115,7 +115,7 @@ def __init__(self, fig, rect=None, *args, **kwargs):
 
         if self.zaxis is not None :
             self._zcid = self.zaxis.callbacks.connect('units finalize',
-                                                      self.relim)
+                                                      self._on_units_changed)
         else :
             self._zcid = None
 
