mirror of
https://github.com/gonum/gonum.git
synced 2025-10-23 15:13:31 +08:00
d4b3b7957c
Previously, the test condition for same-ness of two complex numbers was wrong. The presence of even a single NaN in the real or imaginary component of either number made both numbers to be considered same. This fix changes the logic so that two complex numbers are considered same if: * both numbers have both components NaN * both numbers have one component NaN and the other is exactly equal * both numbers have both components exactly equal
255 lines
5.4 KiB
Go
255 lines
5.4 KiB
Go
// Copyright ©2017 The Gonum Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package testblas
|
|
|
|
import (
|
|
"math/cmplx"
|
|
"testing"
|
|
|
|
"gonum.org/v1/gonum/blas"
|
|
)
|
|
|
|
type Zher2er interface {
|
|
Zher2(uplo blas.Uplo, n int, alpha complex128, x []complex128, incX int, y []complex128, incY int, a []complex128, lda int)
|
|
}
|
|
|
|
func Zher2Test(t *testing.T, impl Zher2er) {
|
|
for tc, test := range []struct {
|
|
alpha complex128
|
|
incX, incY int
|
|
x []complex128
|
|
y []complex128
|
|
a []complex128
|
|
|
|
want []complex128
|
|
}{
|
|
{
|
|
alpha: 1 + 2i,
|
|
incX: 1,
|
|
incY: 1,
|
|
},
|
|
{
|
|
alpha: 1 + 2i,
|
|
incX: 1,
|
|
incY: 1,
|
|
x: []complex128{
|
|
-6 + 2i,
|
|
-2 - 4i,
|
|
0 + 0i,
|
|
0 + 7i,
|
|
},
|
|
y: []complex128{
|
|
2 - 5i,
|
|
0 + 0i,
|
|
-8 - 9i,
|
|
6 + 6i,
|
|
},
|
|
a: []complex128{
|
|
2 + 0i, -9 + 7i, 3 + 11i, 10 - 1i,
|
|
-9 - 7i, 16 + 0i, -5 + 2i, -7 - 5i,
|
|
3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
|
|
10 + 1i, -7 + 5i, 2 + 1i, 18 + 0i,
|
|
},
|
|
want: []complex128{
|
|
62 + 0i, 43 - 7i, 173 + 1i, -173 + 55i,
|
|
43 + 7i, 16 + 0i, 19 + 120i, -19 - 89i,
|
|
173 - 1i, 19 - 120i, 14 + 0i, 51 + 181i,
|
|
-173 - 55i, -19 + 89i, 51 - 181i, -66 + 0i,
|
|
},
|
|
},
|
|
{
|
|
alpha: 1 + 2i,
|
|
incX: 3,
|
|
incY: 7,
|
|
x: []complex128{
|
|
-6 + 2i,
|
|
-2 - 4i,
|
|
0 + 0i,
|
|
0 + 7i,
|
|
},
|
|
y: []complex128{
|
|
2 - 5i,
|
|
0 + 0i,
|
|
-8 - 9i,
|
|
6 + 6i,
|
|
},
|
|
a: []complex128{
|
|
2 + 0i, -9 + 7i, 3 + 11i, 10 - 1i,
|
|
-9 - 7i, 16 + 0i, -5 + 2i, -7 - 5i,
|
|
3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
|
|
10 + 1i, -7 + 5i, 2 + 1i, 18 + 0i,
|
|
},
|
|
want: []complex128{
|
|
62 + 0i, 43 - 7i, 173 + 1i, -173 + 55i,
|
|
43 + 7i, 16 + 0i, 19 + 120i, -19 - 89i,
|
|
173 - 1i, 19 - 120i, 14 + 0i, 51 + 181i,
|
|
-173 - 55i, -19 + 89i, 51 - 181i, -66 + 0i,
|
|
},
|
|
},
|
|
{
|
|
alpha: 1 + 2i,
|
|
incX: -3,
|
|
incY: 7,
|
|
x: []complex128{
|
|
0 + 7i,
|
|
0 + 0i,
|
|
-2 - 4i,
|
|
-6 + 2i,
|
|
},
|
|
y: []complex128{
|
|
2 - 5i,
|
|
0 + 0i,
|
|
-8 - 9i,
|
|
6 + 6i,
|
|
},
|
|
a: []complex128{
|
|
2 + 0i, -9 + 7i, 3 + 11i, 10 - 1i,
|
|
-9 - 7i, 16 + 0i, -5 + 2i, -7 - 5i,
|
|
3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
|
|
10 + 1i, -7 + 5i, 2 + 1i, 18 + 0i,
|
|
},
|
|
want: []complex128{
|
|
62 + 0i, 43 - 7i, 173 + 1i, -173 + 55i,
|
|
43 + 7i, 16 + 0i, 19 + 120i, -19 - 89i,
|
|
173 - 1i, 19 - 120i, 14 + 0i, 51 + 181i,
|
|
-173 - 55i, -19 + 89i, 51 - 181i, -66 + 0i,
|
|
},
|
|
},
|
|
{
|
|
alpha: 1 + 2i,
|
|
incX: 3,
|
|
incY: -7,
|
|
x: []complex128{
|
|
-6 + 2i,
|
|
-2 - 4i,
|
|
0 + 0i,
|
|
0 + 7i,
|
|
},
|
|
y: []complex128{
|
|
6 + 6i,
|
|
-8 - 9i,
|
|
0 + 0i,
|
|
2 - 5i,
|
|
},
|
|
a: []complex128{
|
|
2 + 0i, -9 + 7i, 3 + 11i, 10 - 1i,
|
|
-9 - 7i, 16 + 0i, -5 + 2i, -7 - 5i,
|
|
3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
|
|
10 + 1i, -7 + 5i, 2 + 1i, 18 + 0i,
|
|
},
|
|
want: []complex128{
|
|
62 + 0i, 43 - 7i, 173 + 1i, -173 + 55i,
|
|
43 + 7i, 16 + 0i, 19 + 120i, -19 - 89i,
|
|
173 - 1i, 19 - 120i, 14 + 0i, 51 + 181i,
|
|
-173 - 55i, -19 + 89i, 51 - 181i, -66 + 0i,
|
|
},
|
|
},
|
|
{
|
|
alpha: 1 + 2i,
|
|
incX: -3,
|
|
incY: -7,
|
|
x: []complex128{
|
|
0 + 7i,
|
|
0 + 0i,
|
|
-2 - 4i,
|
|
-6 + 2i,
|
|
},
|
|
y: []complex128{
|
|
6 + 6i,
|
|
-8 - 9i,
|
|
0 + 0i,
|
|
2 - 5i,
|
|
},
|
|
a: []complex128{
|
|
2 + 0i, -9 + 7i, 3 + 11i, 10 - 1i,
|
|
-9 - 7i, 16 + 0i, -5 + 2i, -7 - 5i,
|
|
3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
|
|
10 + 1i, -7 + 5i, 2 + 1i, 18 + 0i,
|
|
},
|
|
want: []complex128{
|
|
62 + 0i, 43 - 7i, 173 + 1i, -173 + 55i,
|
|
43 + 7i, 16 + 0i, 19 + 120i, -19 - 89i,
|
|
173 - 1i, 19 - 120i, 14 + 0i, 51 + 181i,
|
|
-173 - 55i, -19 + 89i, 51 - 181i, -66 + 0i,
|
|
},
|
|
},
|
|
{
|
|
alpha: 0,
|
|
incX: 1,
|
|
incY: 1,
|
|
x: []complex128{
|
|
-6 + 2i,
|
|
-2 - 4i,
|
|
0 + 0i,
|
|
0 + 7i,
|
|
},
|
|
y: []complex128{
|
|
2 - 5i,
|
|
0 + 0i,
|
|
-8 - 9i,
|
|
6 + 6i,
|
|
},
|
|
a: []complex128{
|
|
2 + 0i, -9 + 7i, 3 + 11i, 10 - 1i,
|
|
-9 - 7i, 16 + 0i, -5 + 2i, -7 - 5i,
|
|
3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
|
|
10 + 1i, -7 + 5i, 2 + 1i, 18 + 0i,
|
|
},
|
|
want: []complex128{
|
|
2 + 0i, -9 + 7i, 3 + 11i, 10 - 1i,
|
|
-9 - 7i, 16 + 0i, -5 + 2i, -7 - 5i,
|
|
3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
|
|
10 + 1i, -7 + 5i, 2 + 1i, 18 + 0i,
|
|
},
|
|
},
|
|
} {
|
|
n := len(test.x)
|
|
incX := test.incX
|
|
incY := test.incY
|
|
for _, uplo := range []blas.Uplo{blas.Lower, blas.Upper} {
|
|
for _, lda := range []int{max(1, n), n + 11} {
|
|
x := makeZVector(test.x, incX)
|
|
xCopy := make([]complex128, len(x))
|
|
copy(xCopy, x)
|
|
|
|
y := makeZVector(test.y, incY)
|
|
yCopy := make([]complex128, len(y))
|
|
copy(yCopy, y)
|
|
|
|
a := makeZGeneral(test.a, n, n, lda)
|
|
want := makeZGeneral(test.want, n, n, lda)
|
|
|
|
if uplo == blas.Upper {
|
|
for i := 0; i < n; i++ {
|
|
for j := 0; j < i; j++ {
|
|
a[i*lda+j] = cmplx.NaN()
|
|
want[i*lda+j] = cmplx.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()
|
|
}
|
|
}
|
|
}
|
|
|
|
impl.Zher2(uplo, n, test.alpha, x, incX, y, incY, a, lda)
|
|
|
|
if !zsame(x, xCopy) {
|
|
t.Errorf("Case %v (uplo=%v,incX=%v,incY=%v,lda=%v: unexpected modification of x", tc, uplo, incX, incY, lda)
|
|
}
|
|
if !zsame(y, yCopy) {
|
|
t.Errorf("Case %v (uplo=%v,incX=%v,incY=%v,lda=%v: unexpected modification of y", tc, uplo, incX, incY, lda)
|
|
}
|
|
if !zsame(want, a) {
|
|
t.Errorf("Case %v (uplo=%v,incX=%v,incY=%v,lda=%v: unexpected result\nwant: %v\ngot: %v", tc, uplo, incX, incY, lda, want, a)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|