+++ /dev/null
-// 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 zherTestCases = []struct {
- alpha float64
- x []complex128
- a []complex128
-
- want []complex128
- wantRev []complex128 // Result when incX is negative.
-}{
- {
- alpha: 1,
- },
- {
- alpha: 3,
- x: []complex128{
- 0 - 3i,
- 6 + 10i,
- -2 - 7i,
- },
- a: []complex128{
- -2 + 3i, -3 - 11i, 0 + 4i,
- -3 + 11i, -6 + 3i, 7 + 2i,
- 0 - 4i, 7 - 2i, 18 + 3i,
- },
- want: []complex128{
- 25 + 0i, -93 - 65i, 63 + 22i,
- -93 + 65i, 402 + 0i, -239 + 68i,
- 63 - 22i, -239 - 68i, 177 + 0i},
- wantRev: []complex128{
- 157 + 0i, -249 - 77i, 63 - 14i,
- -249 + 77i, 402 + 0i, -83 + 56i,
- 63 + 14i, -83 - 56i, 45 + 0i,
- },
- },
- {
- alpha: 3,
- x: []complex128{
- -6 + 2i,
- -2 - 4i,
- 0 + 0i,
- 0 + 7i,
- },
- a: []complex128{
- 2 + 3i, -9 + 7i, 3 + 11i, 10 - 1i,
- -9 - 7i, 16 + 3i, -5 + 2i, -7 - 5i,
- 3 - 11i, -5 - 2i, 14 + 3i, 2 - 1i,
- 10 + 1i, -7 + 5i, 2 + 1i, 18 + 3i,
- },
- want: []complex128{
- 122 + 0i, 3 - 77i, 3 + 11i, 52 + 125i,
- 3 + 77i, 76 + 0i, -5 + 2i, -91 + 37i,
- 3 - 11i, -5 - 2i, 14 + 0i, 2 - 1i,
- 52 - 125i, -91 - 37i, 2 + 1i, 165 + 0i,
- },
- wantRev: []complex128{
- 149 + 0i, -9 + 7i, -81 - 31i, 52 - 127i,
- -9 - 7i, 16 + 0i, -5 + 2i, -7 - 5i,
- -81 + 31i, -5 - 2i, 74 + 0i, 14 + 83i,
- 52 + 127i, -7 + 5i, 14 - 83i, 138 + 0i,
- },
- },
- {
- alpha: 0,
- x: []complex128{
- -6 + 2i,
- -2 - 4i,
- 0 + 0i,
- 0 + 7i,
- },
- 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,
- },
- wantRev: []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 Zherer interface {
- Zher(uplo blas.Uplo, n int, alpha float64, x []complex128, incX int, a []complex128, lda int)
-}
-
-func ZherTest(t *testing.T, impl Zherer) {
- for tc, test := range zherTestCases {
- n := len(test.x)
- for _, uplo := range []blas.Uplo{blas.Lower, blas.Upper} {
- for _, incX := range []int{-11, -2, -1, 1, 2, 7} {
- for _, lda := range []int{max(1, n), n + 11} {
- x := makeZVector(test.x, incX)
- xCopy := make([]complex128, len(x))
- copy(xCopy, x)
-
- a := makeZGeneral(test.a, n, n, lda)
-
- var want []complex128
- if incX > 0 {
- want = makeZGeneral(test.want, n, n, lda)
- } else {
- want = makeZGeneral(test.wantRev, 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.Zher(uplo, n, test.alpha, x, incX, a, lda)
-
- if !zsame(x, xCopy) {
- t.Errorf("Case %v (uplo=%v,incX=%v,lda=%v,alpha=%v): unexpected modification of x", tc, uplo, incX, test.alpha, lda)
- }
- if !zsame(want, a) {
- t.Errorf("Case %v (uplo=%v,incX=%v,lda=%v,alpha=%v): unexpected result\nwant: %v\ngot: %v", tc, uplo, incX, lda, test.alpha, want, a)
- }
- }
- }
- }
- }
-}
OSDN Git Service
