FIX: _safe_first_finite on all nan array

rcomer · rcomer · commit 237e492bf8f4 · 2023-03-25T15:40:59.000Z
diff --git a/lib/matplotlib/cbook.py b/lib/matplotlib/cbook.py
@@ -1657,7 +1657,7 @@ def safe_isfinite(val):
         raise RuntimeError("matplotlib does not "
                            "support generators as input")
     else:
-        return next(val for val in obj if safe_isfinite(val))
+        return next((val for val in obj if safe_isfinite(val)), np.nan)
 
 
 def sanitize_sequence(data):
diff --git a/lib/matplotlib/tests/test_cbook.py b/lib/matplotlib/tests/test_cbook.py
@@ -608,6 +608,12 @@ def test_flatiter():
     assert 1 == next(it)
 
 
+def test__safe_first_finite_all_nan():
+    arr = np.full(2, np.nan)
+    ret = cbook._safe_first_finite(arr)
+    assert ret is np.nan
+
+
 def test_reshape2d():
 
     class Dummy:
