Merge pull request #2447 from phobson/bug-boxplots-with-manual-CIarray

mdboom · mdboom · commit 5dce348db9b2 · 2013-09-30T14:00:40.000-07:00
BUG: Fix boxplots with manual confidence intervals passed as a numpy array
diff --git a/lib/matplotlib/axes.py b/lib/matplotlib/axes.py
@@ -5990,7 +5990,7 @@ def computeConfInterval(data, med, iq, bootstrap):
                     raise ValueError(msg1)
                 elif conf_intervals.shape[0] != col:
                     raise ValueError(msg2)
-                elif conf_intervals.shape[1] == 2:
+                elif conf_intervals.shape[1] != 2:
                     raise ValueError(msg3)
             else:
                 if len(conf_intervals) != col:
diff --git a/lib/matplotlib/tests/baseline_images/test_axes/boxplot_with_CIarray.png b/lib/matplotlib/tests/baseline_images/test_axes/boxplot_with_CIarray.png
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -985,6 +985,21 @@ def test_boxplot():
                conf_intervals=[None, (-1.0, 3.5)], notch=1)
     ax.set_ylim((-30, 30))
 
+@image_comparison(baseline_images=['boxplot_with_CIarray'],
+                  remove_text=True, extensions=['png'],
+                  savefig_kwarg={'dpi': 40})
+def test_boxplot_with_CIarray():
+    x = np.linspace(-7, 7, 140)
+    x = np.hstack([-25, x, 25])
+    fig = plt.figure()
+    ax = fig.add_subplot(111)
+    CIs = np.array([[-1.5, 3.], [-1., 3.5]])
+
+    # show 1 boxplot with mpl medians/conf. interfals, 1 with manual values
+    ax.boxplot([x, x], bootstrap=10000, usermedians=[None, 1.0],
+               conf_intervals=CIs, notch=1)
+    ax.set_ylim((-30, 30))
+
 @image_comparison(baseline_images=['boxplot_no_inverted_whisker'],
                   remove_text=True, extensions=['png'],
                   savefig_kwarg={'dpi': 40})
