Add test ax.set_yticks for math.round issue of np.inf

doronbehar · doronbehar · commit 1ae9888dfd5e · 2024-05-29T16:22:04.000+03:00
diff --git a/lib/matplotlib/tests/test_ticker.py b/lib/matplotlib/tests/test_ticker.py
@@ -1774,6 +1774,10 @@ def test_small_range_loglocator(numticks):
         ticks = ll.tick_values(.5, top)
         assert (np.diff(np.log10(ll.tick_values(6, 150))) == 1).all()
 
+def test_yticks_with_inf():
+    fig, ax = plt.subplots()
+    ax.loglog(np.logspace(0, 10, 10), np.logspace(0, 10, 10)**2)
+    ax.set_yticks([1, 10, np.inf])
 
 def test_NullFormatter():
     formatter = mticker.NullFormatter()
