blob: b16686a45a321c2050a2f5cef3a25d5ee02e8759 [file] [log] [blame]
// 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)
}
}
}
}
}