more fuzz setup

alicebob · Oct 4, 2023 · e9a11f3 · e9a11f3
1 parent 330352d
commit e9a11f3
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/fpconv/fuzz_test.go b/fpconv/fuzz_test.go
@@ -4,6 +4,7 @@
 package fpconv
 
 import (
+	"math"
 	"strconv"
 	"testing"
 )
@@ -13,6 +14,9 @@ func FuzzDtoa(f *testing.F) {
 	f.Add(0.0)
 	f.Add(2.5339988685347402e-65)
 	f.Add(3.1415)
+	f.Add(math.Inf(1))
+	f.Add(math.Inf(-1))
+	f.Add(math.NaN())
 
 	f.Fuzz(func(t *testing.T, orig float64) {
 		s := Dtoa(orig)
@@ -23,6 +27,16 @@ func FuzzDtoa(f *testing.F) {
 		if err != nil {
 			t.Errorf("parse failed: %s", err)
 		}
+		if math.IsNaN(orig) {
+			if !math.IsNaN(n) {
+				t.Error("not NaN")
+			}
+			return
+		}
+
+		if math.IsNaN(n) {
+			t.Error("got NaN")
+		}
 		if n != orig {
 			t.Errorf("changed %f -> %f", n, orig)
 		}