+++ /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 ztrmvTestCases = []struct {
- uplo blas.Uplo
- a []complex128
- x []complex128
-
- // Results with non-unit diagonal.
- want []complex128
- wantNeg []complex128
- wantTrans []complex128
- wantTransNeg []complex128
- wantConjTrans []complex128
- wantConjTransNeg []complex128
-
- // Results with unit diagonal.
- wantUnit []complex128
- wantUnitNeg []complex128
- wantUnitTrans []complex128
- wantUnitTransNeg []complex128
- wantUnitConjTrans []complex128
- wantUnitConjTransNeg []complex128
-}{
- {uplo: blas.Upper},
- {uplo: blas.Lower},
- {
- uplo: blas.Upper,
- a: []complex128{
- 6 - 8i, -10 + 10i, -6 - 3i, -1 - 8i,
- znan, 7 + 8i, -7 + 9i, 3 + 6i,
- znan, znan, 6 - 4i, -2 - 5i,
- znan, znan, znan, 4 - 8i,
- },
- x: []complex128{
- 10 - 5i,
- -2 + 2i,
- 8 - 1i,
- -7 + 9i,
- },
-
- want: []complex128{
- 48 - 121i,
- -152 + 62i,
- 103 - 21i,
- 44 + 92i,
- },
- wantNeg: []complex128{
- 0 - 100i,
- -49 - 20i,
- 120 + 70i,
- -72 + 119i,
- },
- wantTrans: []complex128{
- 20 - 110i,
- -80 + 148i,
- -35 - 70i,
- -45 - 27i,
- },
- wantTransNeg: []complex128{
- 123 - 2i,
- 18 + 66i,
- 44 - 103i,
- 30 + 110i,
- },
- wantConjTrans: []complex128{
- 100 + 50i,
- -148 - 20i,
- 39 + 90i,
- -75 + 125i,
- },
- wantConjTransNeg: []complex128{
- 27 - 70i,
- -70 - 136i,
- 208 - 91i,
- -114 - 2i,
- },
-
- wantUnit: []complex128{
- 38 - 16i,
- -124 + 66i,
- 67 + 16i,
- -7 + 9i,
- },
- wantUnitNeg: []complex128{
- 10 - 5i,
- -47 - 38i,
- 64 + 12i,
- -109 + 18i,
- },
- wantUnitTrans: []complex128{
- 10 - 5i,
- -52 + 152i,
- -71 - 33i,
- -96 - 110i,
- },
- wantUnitTransNeg: []complex128{
- 133 + 93i,
- 20 + 48i,
- -12 - 161i,
- -7 + 9i,
- },
- wantUnitConjTrans: []complex128{
- 10 - 5i,
- -152 - 48i,
- -5 + 63i,
- 18 + 154i,
- },
- wantUnitConjTransNeg: []complex128{
- -43 - 135i,
- -52 - 138i,
- 168 - 21i,
- -7 + 9i,
- },
- },
- {
- uplo: blas.Lower,
- a: []complex128{
- 10 - 8i, znan, znan, znan,
- 1 - 6i, -4 + 8i, znan, znan,
- 2 - 6i, 4 - 8i, 5 + 3i, znan,
- -7 - 4i, 1 + 3i, -2 - 4i, 9 + 8i,
- },
- x: []complex128{
- 10 + 5i,
- -7 + 1i,
- 3 - 1i,
- 9 + 10i,
- },
-
- want: []complex128{
- 140 - 30i,
- 60 - 115i,
- 48 + 14i,
- -69 + 57i,
- },
- wantNeg: []complex128{
- 51 + 53i,
- 44 - 78i,
- 65 - 16i,
- 170 + 28i,
- },
- wantTrans: []complex128{
- 116 - 113i,
- 3 - 51i,
- 40 - 52i,
- 1 + 162i,
- },
- wantTransNeg: []complex128{
- 50 + 125i,
- -38 - 66i,
- -29 + 123i,
- 109 - 22i,
- },
- wantConjTrans: []complex128{
- -44 + 71i,
- 95 + 55i,
- -46 + 2i,
- 161 + 18i,
- },
- wantConjTransNeg: []complex128{
- 130 - 35i,
- -72 + 56i,
- -31 - 97i,
- -91 + 154i,
- },
-
- wantUnit: []complex128{
- 10 + 5i,
- 33 - 54i,
- 33 + 9i,
- -61 - 95i,
- },
- wantUnitNeg: []complex128{
- 11 - 67i,
- 75 - 61i,
- 72 - 45i,
- 9 + 10i,
- },
- wantUnitTrans: []complex128{
- -14 - 78i,
- -24 + 10i,
- 25 - 57i,
- 9 + 10i,
- },
- wantUnitTransNeg: []complex128{
- 10 + 5i,
- -7 - 49i,
- -22 + 94i,
- -52 - 40i,
- },
- wantUnitConjTrans: []complex128{
- -94 - 54i,
- 52 + 4i,
- -55 + 15i,
- 9 + 10i,
- },
- wantUnitConjTransNeg: []complex128{
- 10 + 5i,
- -47 + 31i,
- -8 - 78i,
- -92 - 8i,
- },
- },
-}
-
-type Ztrmver interface {
- Ztrmv(uplo blas.Uplo, trans blas.Transpose, diag blas.Diag, n int, a []complex128, lda int, x []complex128, incX int)
-}
-
-func ZtrmvTest(t *testing.T, impl Ztrmver) {
- for tc, test := range ztrmvTestCases {
- n := len(test.x)
- uplo := test.uplo
- for _, trans := range []blas.Transpose{blas.NoTrans, blas.Trans, blas.ConjTrans} {
- for _, diag := range []blas.Diag{blas.NonUnit, blas.Unit} {
- for _, incX := range []int{-11, -2, -1, 1, 2, 7} {
- for _, lda := range []int{max(1, n), n + 11} {
- a := makeZGeneral(test.a, n, n, lda)
- if diag == blas.Unit {
- for i := 0; i < n; i++ {
- a[i*lda+i] = znan
- }
- }
- aCopy := make([]complex128, len(a))
- copy(aCopy, a)
-
- x := makeZVector(test.x, incX)
-
- impl.Ztrmv(uplo, trans, diag, n, a, lda, x, incX)
-
- if !zsame(a, aCopy) {
- t.Errorf("Case %v (uplo=%v,trans=%v,diag=%v,lda=%v,incX=%v): unexpected modification of A", tc, uplo, trans, diag, lda, incX)
- }
-
- var want []complex128
- if diag == blas.NonUnit {
- switch {
- case trans == blas.NoTrans && incX > 0:
- want = makeZVector(test.want, incX)
- case trans == blas.NoTrans && incX < 0:
- want = makeZVector(test.wantNeg, incX)
- case trans == blas.Trans && incX > 0:
- want = makeZVector(test.wantTrans, incX)
- case trans == blas.Trans && incX < 0:
- want = makeZVector(test.wantTransNeg, incX)
- case trans == blas.ConjTrans && incX > 0:
- want = makeZVector(test.wantConjTrans, incX)
- case trans == blas.ConjTrans && incX < 0:
- want = makeZVector(test.wantConjTransNeg, incX)
- }
- } else {
- switch {
- case trans == blas.NoTrans && incX > 0:
- want = makeZVector(test.wantUnit, incX)
- case trans == blas.NoTrans && incX < 0:
- want = makeZVector(test.wantUnitNeg, incX)
- case trans == blas.Trans && incX > 0:
- want = makeZVector(test.wantUnitTrans, incX)
- case trans == blas.Trans && incX < 0:
- want = makeZVector(test.wantUnitTransNeg, incX)
- case trans == blas.ConjTrans && incX > 0:
- want = makeZVector(test.wantUnitConjTrans, incX)
- case trans == blas.ConjTrans && incX < 0:
- want = makeZVector(test.wantUnitConjTransNeg, incX)
- }
- }
- if !zsame(x, want) {
- t.Errorf("Case %v (uplo=%v,trans=%v,diag=%v,lda=%v,incX=%v): unexpected result\nwant %v\ngot %v", tc, uplo, trans, diag, lda, incX, want, x)
- }
- }
- }
- }
- }
- }
-}
OSDN Git Service
