fix cornercase

JuliaPlots · Oct 11, 2024 · 25763d2 · 25763d2
1 parent a6f6f03
commit 25763d2
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/src/common.jl b/src/common.jl
@@ -609,10 +609,9 @@ function colormap_callback(cmap::Symbol)
     cdata = ColorSchemes.colorschemes[cmap]
     (z, minz, maxz) -> begin
         isfinite(z) || return INVALID_COLOR
-        get(
-            cdata,
-            minz == maxz ? zero(z) : (max(minz, min(z, maxz)) - minz) / (maxz - minz),
-        ) |> ansi_color
+        z01 = minz == maxz ? zero(z) : (max(minz, min(z, maxz)) - minz) / (maxz - minz)
+        isfinite(z01) || return INVALID_COLOR
+        get(cdata, z01) |> ansi_color
     end::ColorType
 end
 

diff --git a/test/tst_issues.jl b/test/tst_issues.jl
@@ -107,4 +107,13 @@
         p = scatterplot(rand(1) * u"m", rand(1))
         show(devnull, p)
     end
+
+    @testset "Heatmap ±Inf (#378)" begin
+        p = heatmap([1 2; 3 -Inf])
+        show(devnull, p)
+        p = heatmap([-Inf;;])
+        show(devnull, p)
+        p = heatmap([1 2; 3 Inf])
+        show(devnull, p)
+    end
 end