Fix list test

gonum · Jan 18, 2017 · 933011a · 933011a
1 parent 3ab5199
commit 933011a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/mat64/list_test.go b/mat64/list_test.go
@@ -420,9 +420,10 @@ func equalApprox(a, b Matrix, tol float64, ignoreNaN bool) bool {
 	for i := 0; i < ar; i++ {
 		for j := 0; j < ac; j++ {
 			if !floats.EqualWithinAbsOrRel(a.At(i, j), b.At(i, j), tol, tol) {
-				if !ignoreNaN || math.IsNaN(a.At(i, j)) != math.IsNaN(b.At(i, j)) {
-					return false
+				if ignoreNaN && math.IsNaN(a.At(i, j)) && math.IsNaN(b.At(i, j)) {
+					continue
 				}
+				return false
 			}
 		}
 	}