TST: add test of bad pandas indexing

tacaswell · tacaswell · commit ab6d338c876e · 2015-12-28T19:02:08.000-05:00
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -7,6 +7,7 @@
 import io
 
 from nose.tools import assert_equal, assert_raises, assert_false, assert_true
+from nose.plugins.skip import SkipTest
 
 import datetime
 
@@ -4183,6 +4184,23 @@ def test_broken_barh_empty():
     ax.broken_barh([], (.1, .5))
 
 
+@cleanup
+def test_pandas_indexing_dates():
+    try:
+        import pandas as pd
+    except ImportError:
+        raise SkipTest("Pandas not installed")
+
+    dates = np.arange('2005-02', '2005-03', dtype='datetime64[D]')
+    values = np.sin(np.array(range(len(dates))))
+    df = pd.DataFrame({'dates': dates, 'values': values})
+
+    ax = plt.gca()
+
+    without_zero_index = df[np.array(df.index) % 2 == 1].copy()
+    ax.plot('dates', 'values', data=without_zero_index)
+
+
 if __name__ == '__main__':
     import nose
     import sys
