Merge pull request #2787 from tacaswell/empty_event_loop

pelson · pelson · commit d2f58cbe1985 · 2014-02-10T11:53:52.000Z
Empty eventplot.
diff --git a/lib/matplotlib/axes.py b/lib/matplotlib/axes.py
@@ -3988,18 +3988,24 @@ def eventplot(self, positions, orientation='horizontal', lineoffsets=1,
             colls.append(coll)
 
         if len(positions) > 0:
-            minpos = min(position.min() for position in positions)
-            maxpos = max(position.max() for position in positions)
-
-            minline = (lineoffsets - linelengths).min()
-            maxline = (lineoffsets + linelengths).max()
-
-            if colls[0].is_horizontal():
-                corners = (minpos, minline), (maxpos, maxline)
-            else:
-                corners = (minline, minpos), (maxline, maxpos)
-            self.update_datalim(corners)
-            self.autoscale_view()
+            # try to get min/max
+            min_max = [(np.min(_p), np.max(_p)) for _p in positions
+                       if len(_p) > 0]
+            # if we have any non-empty positions, try to autoscale
+            if len(min_max) > 0:
+                mins, maxes = zip(*min_max)
+                minpos = np.min(mins)
+                maxpos = np.max(maxes)
+
+                minline = (lineoffsets - linelengths).min()
+                maxline = (lineoffsets + linelengths).max()
+
+                if colls[0].is_horizontal():
+                    corners = (minpos, minline), (maxpos, maxline)
+                else:
+                    corners = (minline, minpos), (maxline, maxpos)
+                self.update_datalim(corners)
+                self.autoscale_view()
 
         return colls
 
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -1297,6 +1297,14 @@ def test_eventplot():
     num_collections = len(colls)
     np.testing.assert_equal(num_collections, num_datasets)
 
+
+@cleanup
+def test_empty_eventplot():
+    fig, ax = plt.subplots(1, 1)
+    ax.eventplot([[]], colors=[(0.0, 0.0, 0.0, 0.0)])
+    plt.draw()
+
+
 @image_comparison(baseline_images=['vertex_markers'], extensions=['png'],
                   remove_text=True)
 def test_vertex_markers():
@@ -1389,6 +1397,7 @@ def test_mixed_collection():
     ax.set_xlim(0, 16)
     ax.set_ylim(0, 16)
 
+
 @cleanup
 def test_subplot_key_hash():
     ax = plt.subplot(np.float64(5.5), np.int64(1), np.float64(1.2))
