FIX: properly set tz for YearLocator

jklymak · jklymak · commit 57361558d086 · 2018-10-31T08:27:18.000-07:00
diff --git a/lib/matplotlib/tests/test_dates.py b/lib/matplotlib/tests/test_dates.py
@@ -12,6 +12,7 @@
 from matplotlib.cbook import MatplotlibDeprecationWarning
 import matplotlib.dates as mdates
 import matplotlib.ticker as mticker
+from matplotlib import rc_context
 
 
 def __has_pytz():
@@ -572,11 +573,13 @@ def _create_auto_date_locator(date1, date2, tz):
                )
 
     tz = pytz.timezone('US/Pacific')
-    d1 = datetime.datetime(1997, 1, 1).astimezone(tz)
+    d1 = tz.localize(datetime.datetime(1997, 1, 1))
     for t_delta, expected in results:
-        d2 = d1 + t_delta
-        locator = _create_auto_date_locator(d1, d2, tz)
-        assert list(map(str, mdates.num2date(locator(), tz=tz))) == expected
+        with rc_context({'_internal.classic_mode': False}):
+
+            d2 = d1 + t_delta
+            locator = _create_auto_date_locator(d1, d2, tz)
+            assert list(map(str, mdates.num2date(locator(), tz=tz))) == expected
 
 
 @image_comparison(baseline_images=['date_inverted_limit'],
