Apply code suggestions

Reference: #799 (review)
Signed-off-by: Eng Zer Jun <[email protected]>

Author: Juneezee

palette/moreland/luminance.go

@@ -176,9 +176,16 @@ func (l luminance) Palette(n int) palette.Palette {
 		l.SetMax(1)
 	}
 	delta := (l.max - l.min) / float64(n-1)
+	var v float64
 	c := make([]color.Color, n)
 	for i := range n {
-		v := min(l.min+float64(delta*float64(i)), l.max)
+		if i == n-1 {
+			// Avoid potential overflow on last element
+			// due to floating point error.
+			v = l.max
+		} else {
+			v = l.min + float64(delta*float64(i))
+		}
 		var err error
 		c[i], err = l.At(v)
 		if err != nil {

palette/moreland/luminance_test.go

@@ -162,10 +162,46 @@ func BenchmarkLuminance_At(b *testing.B) {
 	}
 }
 
+// See https://github.com/gonum/plot/issues/798
 func TestIssue798Kindlmann(t *testing.T) {
-	// https://github.com/gonum/plot/issues/798
-	colors := Kindlmann()
-	colors.SetMin(0.3402859786606234)
-	colors.SetMax(15.322841335211892)
-	colors.Palette(15)
+	for _, test := range []struct {
+		n        int
+		min, max float64
+	}{
+		0: {n: 2, min: 0, max: 1},
+		1: {n: 15, min: 0.3402859786606234, max: 15.322841335211892},
+	} {
+		t.Run("", func(t *testing.T) {
+			defer func() {
+				r := recover()
+				if r != nil {
+					t.Errorf("unexpected panic with n=%d min=%f max=%f: %v", test.n, test.min, test.max, r)
+				}
+			}()
+			colors := Kindlmann()
+			colors.SetMin(test.min)
+			colors.SetMax(test.max)
+			col := colors.Palette(test.n).Colors()
+			min, err := colors.At(test.min)
+			if err != nil {
+				t.Fatalf("unexpected error calling colors.At(min): %v", err)
+			}
+			if !sameColor(min, col[0]) {
+				t.Errorf("unexpected min color %#v != %#v", min, col[0])
+			}
+			max, err := colors.At(test.max)
+			if err != nil {
+				t.Fatalf("unexpected error calling colors.At(max): %v", err)
+			}
+			if !sameColor(max, col[len(col)-1]) {
+				t.Errorf("unexpected max color %#v != %#v", max, col[len(col)-1])
+			}
+		})
+	}
+}
+
+func sameColor(a, b color.Color) bool {
+	ar, ag, ab, aa := a.RGBA()
+	br, bg, bb, ba := b.RGBA()
+	return ar == br && ag == bg && ab == bb && aa == ba
 }

palette/moreland/smooth.go

@@ -171,9 +171,16 @@ func (p smoothDiverging) Palette(n int) palette.Palette {
 		p.SetMax(1)
 	}
 	delta := (p.max - p.min) / float64(n-1)
+	var v float64
 	c := make([]color.Color, n)
 	for i := range c {
-		v := min(p.min+float64(delta*float64(i)), p.max)
+		if i == n-1 {
+			// Avoid potential overflow on last element
+			// due to floating point error.
+			v = p.max
+		} else {
+			v = p.min + float64(delta*float64(i))
+		}
 		var err error
 		c[i], err = p.At(v)
 		if err != nil {

palette/moreland/smooth_test.go

@@ -247,10 +247,40 @@ func similar(a, b color.Color, tolerance float64) bool {
 	return true
 }
 
+// See https://github.com/gonum/plot/issues/798
 func TestIssue798SmoothBlueRed(t *testing.T) {
-	// https://github.com/gonum/plot/issues/798
-	colors := SmoothBlueRed()
-	colors.SetMin(0.3402859786606234)
-	colors.SetMax(15.322841335211892)
-	colors.Palette(15)
+	for _, test := range []struct {
+		n        int
+		min, max float64
+	}{
+		0: {n: 2, min: 0, max: 1},
+		1: {n: 15, min: 0.3402859786606234, max: 15.322841335211892},
+	} {
+		t.Run("", func(t *testing.T) {
+			defer func() {
+				r := recover()
+				if r != nil {
+					t.Errorf("unexpected panic with n=%d min=%f max=%f: %v", test.n, test.min, test.max, r)
+				}
+			}()
+			colors := SmoothBlueRed()
+			colors.SetMin(test.min)
+			colors.SetMax(test.max)
+			col := colors.Palette(test.n).Colors()
+			min, err := colors.At(test.min)
+			if err != nil {
+				t.Fatalf("unexpected error calling colors.At(min): %v", err)
+			}
+			if !sameColor(min, col[0]) {
+				t.Errorf("unexpected min color %#v != %#v", min, col[0])
+			}
+			max, err := colors.At(test.max)
+			if err != nil {
+				t.Fatalf("unexpected error calling colors.At(max): %v", err)
+			}
+			if !sameColor(max, col[len(col)-1]) {
+				t.Errorf("unexpected max color %#v != %#v", max, col[len(col)-1])
+			}
+		})
+	}
 }