mirror of
https://github.com/gonum/gonum.git
synced 2025-11-01 02:52:49 +08:00
314 lines
6.3 KiB
Go
314 lines
6.3 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 (
|
|
"testing"
|
|
|
|
"gonum.org/v1/gonum/blas"
|
|
)
|
|
|
|
var zher2TestCases = []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: 1,
|
|
incY: 1,
|
|
x: []complex128{
|
|
-6 + 2i,
|
|
-2 - 4i,
|
|
0 + 0i,
|
|
0 + 7i,
|
|
},
|
|
y: []complex128{
|
|
2 - 5i,
|
|
-8 - 9i,
|
|
0 + 0i,
|
|
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, 213 - 17i, 3 + 11i, -173 + 55i,
|
|
213 + 17i, 64 + 0i, -5 + 2i, 30 + 93i,
|
|
3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
|
|
-173 - 55i, 30 - 93i, 2 + 1i, -66 + 0i,
|
|
},
|
|
},
|
|
{
|
|
alpha: 1 + 2i,
|
|
incX: 2,
|
|
incY: 4,
|
|
x: []complex128{
|
|
-6 + 2i,
|
|
-2 - 4i,
|
|
0 + 0i,
|
|
0 + 7i,
|
|
},
|
|
y: []complex128{
|
|
2 - 5i,
|
|
-8 - 9i,
|
|
0 + 0i,
|
|
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, 213 - 17i, 3 + 11i, -173 + 55i,
|
|
213 + 17i, 64 + 0i, -5 + 2i, 30 + 93i,
|
|
3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
|
|
-173 - 55i, 30 - 93i, 2 + 1i, -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,
|
|
},
|
|
},
|
|
}
|
|
|
|
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 zher2TestCases {
|
|
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] = znan
|
|
want[i*lda+j] = znan
|
|
}
|
|
}
|
|
} else {
|
|
for i := 0; i < n; i++ {
|
|
for j := i + 1; j < n; j++ {
|
|
a[i*lda+j] = znan
|
|
want[i*lda+j] = znan
|
|
}
|
|
}
|
|
}
|
|
|
|
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)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|