Improved interaction of center and frame arguments

kpalin · kpalin · commit a4048ae22157 · 2014-09-22T13:39:28.000+03:00
diff --git a/lib/matplotlib/axes/_axes.py b/lib/matplotlib/axes/_axes.py
@@ -2521,8 +2521,11 @@ def pie(self, x, explode=None, labels=None, colors=None,
 
         if not frame:
             self.set_frame_on(False)
-            self.set_xlim((-1.25, 1.25))
-            self.set_ylim((-1.25, 1.25))
+
+            self.set_xlim((-1.25 + center[0],
+                            1.25 + center[0]))
+            self.set_ylim((-1.25 + center[1],
+                            1.25 + center[1]))
             self.set_xticks([])
             self.set_yticks([])
 
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -3314,6 +3314,25 @@ def test_pie_linewidth_0():
     plt.axis('equal')
 
 
+@image_comparison(baseline_images=['pie_center_radius'], extensions=['png'])
+def test_pie_center_radius():
+    # The slices will be ordered and plotted counter-clockwise.
+    labels = 'Frogs', 'Hogs', 'Dogs', 'Logs'
+    sizes = [15, 30, 45, 10]
+    colors = ['yellowgreen', 'gold', 'lightskyblue', 'lightcoral']
+    explode = (0, 0.1, 0, 0) # only "explode" the 2nd slice (i.e. 'Hogs')
+
+    plt.pie(sizes, explode=explode, labels=labels, colors=colors,
+            autopct='%1.1f%%', shadow=True, startangle=90,
+            wedgeprops={'linewidth': 0}, center=(1,2), radius=1.5)
+
+    plt.annotate("Center point", xy=(1,2), xytext=(1,1.5),
+                 arrowprops=dict(arrowstyle="->",
+                                 connectionstyle="arc3"))
+    # Set aspect ratio to be equal so that pie is drawn as a circle.
+    plt.axis('equal')
+
+
 @image_comparison(baseline_images=['pie_linewidth_2'], extensions=['png'])
 def test_pie_linewidth_2():
     # The slices will be ordered and plotted counter-clockwise.
