TST: add test for finite bbox on logit

jklymak · jklymak · commit e213f4bc48ab · 2018-06-12T10:27:24.000-07:00
diff --git a/lib/matplotlib/tests/test_scale.py b/lib/matplotlib/tests/test_scale.py
@@ -17,15 +17,19 @@ def test_log_scales():
 @image_comparison(baseline_images=['logit_scales'], remove_text=True,
                   extensions=['png'])
 def test_logit_scales():
-    ax = plt.figure().add_subplot(111, xscale='logit')
+    fig, ax = plt.subplots()
 
     # Typical extinction curve for logit
     x = np.array([0.001, 0.003, 0.01, 0.03, 0.1, 0.2, 0.3, 0.4, 0.5,
                   0.6, 0.7, 0.8, 0.9, 0.97, 0.99, 0.997, 0.999])
     y = 1.0 / x
 
     ax.plot(x, y)
+    ax.set_xscale('logit')
     ax.grid(True)
+    bbox = ax.get_tightbbox(fig.canvas.get_renderer())
+    assert np.isfinite(bbox.x0)
+    assert np.isfinite(bbox.y0)
 
 
 def test_log_scatter():
