Merge pull request #13255 from NelleV/13187_adding_test

dopplershift · web-flow · commit 4f6db14133c3 · 2019-01-21T13:31:45.000-07:00
TST pandas support bar
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -1524,6 +1524,26 @@ def test_bar_timedelta():
                    (10, 20))
 
 
+def test_bar_pandas(pd):
+    # Smoke test for pandas
+
+    fig, ax = plt.subplots()
+
+    df = pd.DataFrame(
+        {'year': [2018, 2018, 2018],
+         'month': [1, 1, 1],
+         'day': [1, 2, 3],
+         'value': [1, 2, 3]})
+    df['date'] = pd.to_datetime(df[['year', 'month', 'day']])
+
+    monthly = df[['date', 'value']].groupby(['date']).sum()
+    dates = monthly.index
+    forecast = monthly['value']
+    baseline = monthly['value']
+    ax.bar(dates, forecast, width=10, align='center')
+    ax.plot(dates, baseline, color='orange', lw=4)
+
+
 @image_comparison(baseline_images=['hist_log'],
                   remove_text=True)
 def test_hist_log():
