mirror of
https://github.com/gonum/gonum.git
synced 2025-10-06 23:52:47 +08:00
62 lines
1.6 KiB
Go
62 lines
1.6 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"
|
|
)
|
|
|
|
type Zhpmver interface {
|
|
Zhpmv(uplo blas.Uplo, n int, alpha complex128, ap []complex128, x []complex128, incX int, beta complex128, y []complex128, incY int)
|
|
}
|
|
|
|
func ZhpmvTest(t *testing.T, impl Zhpmver) {
|
|
for tc, test := range zhemvTestCases {
|
|
uplo := test.uplo
|
|
n := len(test.x)
|
|
alpha := test.alpha
|
|
beta := test.beta
|
|
for _, incX := range []int{-11, -2, -1, 1, 2, 7} {
|
|
for _, incY := range []int{-11, -2, -1, 1, 2, 7} {
|
|
x := makeZVector(test.x, incX)
|
|
xCopy := make([]complex128, len(x))
|
|
copy(xCopy, x)
|
|
|
|
y := makeZVector(test.y, incY)
|
|
|
|
ap := zPack(uplo, n, test.a, n)
|
|
apCopy := make([]complex128, len(ap))
|
|
copy(apCopy, ap)
|
|
|
|
impl.Zhpmv(test.uplo, n, alpha, ap, x, incX, beta, y, incY)
|
|
|
|
if !zsame(x, xCopy) {
|
|
t.Errorf("Case %v (incX=%v,incY=%v): unexpected modification of x", tc, incX, incY)
|
|
}
|
|
if !zsame(ap, apCopy) {
|
|
t.Errorf("Case %v (incX=%v,incY=%v): unexpected modification of A", tc, incX, incY)
|
|
}
|
|
|
|
var want []complex128
|
|
switch {
|
|
case incX > 0 && incY > 0:
|
|
want = makeZVector(test.want, incY)
|
|
case incX < 0 && incY > 0:
|
|
want = makeZVector(test.wantXNeg, incY)
|
|
case incX > 0 && incY < 0:
|
|
want = makeZVector(test.wantYNeg, incY)
|
|
default:
|
|
want = makeZVector(test.wantXYNeg, incY)
|
|
}
|
|
if !zsame(y, want) {
|
|
t.Errorf("Case %v (incX=%v,incY=%v): unexpected result\nwant %v\ngot %v", tc, incX, incY, want, y)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|