Fix tests

dstansby · dstansby · commit fba6973fa150 · 2019-08-17T11:35:51.000+01:00
diff --git a/lib/matplotlib/axes/_base.py b/lib/matplotlib/axes/_base.py
@@ -3045,7 +3045,7 @@ def set_xbound(self, lower=None, upper=None):
         if upper is None and np.iterable(lower):
             lower, upper = lower
 
-        old_lower, old_upper = self.get_xbound()
+        old_lower, old_upper = self.get_xbound(units=False)
 
         if lower is None:
             lower = old_lower
@@ -3087,7 +3087,6 @@ def get_xlim(self, units=True):
         -----
         The x-axis may be inverted, in which case the *left* value will
         be greater than the *right* value.
-
         """
         if units:
             return self._convert_back_lim(self.viewLim.intervalx, self.xaxis)
@@ -3425,7 +3424,7 @@ def yaxis_inverted(self):
         """
         return self.yaxis.get_inverted()
 
-    def get_ybound(self):
+    def get_ybound(self, units=True):
         """
         Return the lower and upper y-axis bounds, in increasing order.
 
@@ -3440,7 +3439,7 @@ def get_ybound(self):
         get_ylim, set_ylim
         invert_yaxis, yaxis_inverted
         """
-        bottom, top = self.get_ylim()
+        bottom, top = self.get_ylim(units=units)
         if bottom < top:
             return bottom, top
         else:
@@ -3468,7 +3467,7 @@ def set_ybound(self, lower=None, upper=None):
         if upper is None and np.iterable(lower):
             lower, upper = lower
 
-        old_lower, old_upper = self.get_ybound()
+        old_lower, old_upper = self.get_ybound(units=False)
 
         if lower is None:
             lower = old_lower
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -6471,7 +6471,10 @@ def test_datetime_masked():
     fig, ax = plt.subplots()
     ax.plot(x, m)
     # these are the default viewlim
-    assert ax.get_xlim() == (730120.0, 733773.0)
+    assert ax.get_xlim(units=False) == (730120.0, 733773.0)
+    assert (ax.get_xlim(units=True) ==
+            (datetime.datetime(2000, 1, 1, tzinfo=datetime.timezone.utc),
+             datetime.datetime(2010, 1, 1, tzinfo=datetime.timezone.utc)))
 
 
 def test_hist_auto_bins():
