blas/gonum: test that imaginary part of diagonal is ignored in Zher

gonum · Aug 15, 2017 · eabe408 · eabe408
1 parent 1dc3fdb
commit eabe408
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/blas/testblas/zher.go b/blas/testblas/zher.go
@@ -5,6 +5,7 @@
 package testblas
 
 import (
+	"math"
 	"math/cmplx"
 	"testing"
 
@@ -127,13 +128,15 @@ func ZherTest(t *testing.T, impl Zherer) {
 								a[i*lda+j] = cmplx.NaN()
 								want[i*lda+j] = cmplx.NaN()
 							}
+							a[i*lda+i] = complex(real(a[i*lda+i]), math.NaN())
 						}
 					} else {
 						for i := 0; i < n; i++ {
 							for j := i + 1; j < n; j++ {
 								a[i*lda+j] = cmplx.NaN()
 								want[i*lda+j] = cmplx.NaN()
 							}
+							a[i*lda+i] = complex(real(a[i*lda+i]), math.NaN())
 						}
 					}