REF/TST: update boxplot_stats test with outliers changed to fliers

phobson · phobson · commit 614736d6689e · 2013-12-02T20:01:46.000-08:00
ENH/TST: boxplots stats now takes a labels kwarg

MNT: minor PEP8 whitespace fix
diff --git a/lib/matplotlib/cbook.py b/lib/matplotlib/cbook.py
@@ -1832,7 +1832,7 @@ def delete_masked_points(*args):
     return margs
 
 
-def boxplot_stats(X, whis=1.5, bootstrap=None):
+def boxplot_stats(X, whis=1.5, bootstrap=None, labels=None):
     '''
     Returns list of dictionaries of staticists to be use to draw a series of
     box and whisker plots. See the `Returns` section below to the required
@@ -1848,7 +1848,7 @@ def boxplot_stats(X, whis=1.5, bootstrap=None):
 
     whis : float (default = 1.5)
         Determines the reach of the whiskers past the first and third
-        quartiles (e.g., Q3 + whis*IQR). Beyone the whiskers, data are
+        quartiles (e.g., Q3 + whis*IQR). Beyond the whiskers, data are
         considers outliers and are plotted as individual points. Set
         this to an unreasonably high value to force the whiskers to
         show the min and max data. (IQR = interquartile range, Q3-Q1)
@@ -1857,6 +1857,10 @@ def boxplot_stats(X, whis=1.5, bootstrap=None):
         Number of times the confidence intervals around the median should
         be bootstrapped (percentile method).
 
+    labels : sequence
+        Labels for each dataset. Length must be compatible with dimensions
+        of `X`
+
     Returns
     -------
     bxpstats : A list of dictionaries containing the results for each column
@@ -1926,9 +1930,21 @@ def _compute_conf_interval(data, med, iqr, bootstrap):
         X = [X]
 
     ncols = len(X)
-    for ii, x in enumerate(X, start=0):
+    if labels is None:
+        labels = [None] * ncols
+    elif len(labels) != ncols:
+        raise ValueError("Dimensions of labels and X must be compatible")
+
+    for ii, (x, label) in enumerate(zip(X, labels), start=0):
+        # empty dict
         stats = {}
 
+        # set the label
+        if label is not None:
+            stats['label'] = label
+        else:
+            stats['label'] = ii
+
         # arithmetic mean
         stats['mean'] = np.mean(x)
 
diff --git a/lib/matplotlib/tests/test_cbook.py b/lib/matplotlib/tests/test_cbook.py
@@ -105,7 +105,7 @@ def setup(self):
         self.known_keys = sorted([
             'mean', 'med', 'q1', 'q3', 'iqr',
             'cilo', 'cihi', 'whislo', 'whishi',
-            'outliers'
+            'fliers', 'label'
         ])
         self.std_results = cbook.boxplot_stats(self.data)
 
@@ -115,13 +115,14 @@ def setup(self):
             'iqr': 13.492709959447094,
             'mean': 13.00447442387868,
             'med': 3.3335733967038079,
-            'outliers': np.array([
+            'fliers': np.array([
                 92.55467075,  87.03819018,  42.23204914,  39.29390996
             ]),
             'q1': 1.3597529879465153,
             'q3': 14.85246294739361,
             'whishi': 27.899688243699629,
-            'whislo': 0.042143774965502923
+            'whislo': 0.042143774965502923,
+            'label': 0
         }
 
         self.known_bootstrapped_ci = {
@@ -132,7 +133,11 @@ def setup(self):
         self.known_whis3_res = {
             'whishi': 42.232049135969874,
             'whislo': 0.042143774965502923,
-            'outliers': np.array([92.55467075, 87.03819018]),
+            'fliers': np.array([92.55467075, 87.03819018]),
+        }
+
+        self.known_res_with_labels = {
+            'label': 'Test1'
         }
 
     def test_form_main_list(self):
@@ -151,7 +156,7 @@ def test_form_dict_keys(self):
     def test_results_baseline(self):
         res = self.std_results[0]
         for key in list(self.known_nonbootstrapped_res.keys()):
-            if key != 'outliers':
+            if key != 'fliers':
                 assert_statement = assert_approx_equal
             else:
                 assert_statement = assert_array_almost_equal
@@ -174,7 +179,7 @@ def test_results_whiskers(self):
         results = cbook.boxplot_stats(self.data, whis=3)
         res = results[0]
         for key in list(self.known_whis3_res.keys()):
-            if key != 'outliers':
+            if key != 'fliers':
                 assert_statement = assert_approx_equal
             else:
                 assert_statement = assert_array_almost_equal
@@ -183,3 +188,20 @@ def test_results_whiskers(self):
                 res[key],
                 self.known_whis3_res[key]
             )
+
+    def test_results_withlabels(self):
+        labels = ['Test1', 2, 3, 4]
+        results = cbook.boxplot_stats(self.data, labels=labels)
+        res = results[0]
+        for key in list(self.known_res_with_labels.keys()):
+            assert_equal(res[key], self.known_res_with_labels[key])
+
+    @raises(ValueError)
+    def test_label_error(self):
+        labels = [1, 2]
+        results = cbook.boxplot_stats(self.data, labels=labels)
+
+    @raises(ValueError)
+    def test_bad_dims(self):
+        data = np.random.normal(size=(34, 34, 34))
+        results = cbook.boxplot_stats(data)
