TST: add a test for pcolormesh underflow

jklymak · jklymak · commit fdfeba2032c9 · 2024-02-11T18:16:37.000-08:00
diff --git a/lib/matplotlib/tests/test_axes.py b/lib/matplotlib/tests/test_axes.py
@@ -1517,6 +1517,22 @@ def test_pcolorargs():
         ax.pcolormesh(X, Y, Z, shading='auto')
 
 
+def test_pcolormesh_underflow_error():
+    """
+    Test that underflow errors don't crop up in pcolormesh.  Probably
+    a numpy bug, but...
+    """
+    oldsettings = np.geterr()
+    np.seterr(under="raise")
+    x = np.arange(0, 3, 0.1)
+    y = np.arange(0, 6, 0.1)
+    z = np.random.randn(len(y), len(x))
+    fig, ax = plt.subplots()
+    ax.pcolormesh(x, y, z)
+
+    np.seterr(**oldsettings)
+
+
 def test_pcolorargs_with_read_only():
     x = np.arange(6).reshape(2, 3)
     xmask = np.broadcast_to([False, True, False], x.shape)  # read-only array
