vendor/gonum.org/v1/gonum/blas/testblas/dtbmv.go

   1 package testblas
   2
   3 import (
   4         "testing"
   5
   6         "gonum.org/v1/gonum/blas"
   7 )
   8
   9 type Dtbmver interface {
  10         Dtbmv(ul blas.Uplo, tA blas.Transpose, d blas.Diag, n, k int, a []float64, lda int, x []float64, incX int)
  11 }
  12
  13 func DtbmvTest(t *testing.T, blasser Dtbmver) {
  14         for i, test := range []struct {
  15                 ul  blas.Uplo
  16                 tA  blas.Transpose
  17                 d   blas.Diag
  18                 n   int
  19                 k   int
  20                 a   [][]float64
  21                 x   []float64
  22                 ans []float64
  23         }{
  24                 {
  25                         ul: blas.Upper,
  26                         tA: blas.NoTrans,
  27                         d:  blas.Unit,
  28                         n:  3,
  29                         k:  1,
  30                         a: [][]float64{
  31                                 {1, 2, 0},
  32                                 {0, 1, 4},
  33                                 {0, 0, 1},
  34                         },
  35                         x:   []float64{2, 3, 4},
  36                         ans: []float64{8, 19, 4},
  37                 },
  38                 {
  39                         ul: blas.Upper,
  40                         tA: blas.NoTrans,
  41                         d:  blas.NonUnit,
  42                         n:  5,
  43                         k:  1,
  44                         a: [][]float64{
  45                                 {1, 3, 0, 0, 0},
  46                                 {0, 6, 7, 0, 0},
  47                                 {0, 0, 2, 1, 0},
  48                                 {0, 0, 0, 12, 3},
  49                                 {0, 0, 0, 0, -1},
  50                         },
  51                         x:   []float64{1, 2, 3, 4, 5},
  52                         ans: []float64{7, 33, 10, 63, -5},
  53                 },
  54                 {
  55                         ul: blas.Lower,
  56                         tA: blas.NoTrans,
  57                         d:  blas.NonUnit,
  58                         n:  5,
  59                         k:  1,
  60                         a: [][]float64{
  61                                 {7, 0, 0, 0, 0},
  62                                 {3, 6, 0, 0, 0},
  63                                 {0, 7, 2, 0, 0},
  64                                 {0, 0, 1, 12, 0},
  65                                 {0, 0, 0, 3, -1},
  66                         },
  67                         x:   []float64{1, 2, 3, 4, 5},
  68                         ans: []float64{7, 15, 20, 51, 7},
  69                 },
  70                 {
  71                         ul: blas.Upper,
  72                         tA: blas.Trans,
  73                         d:  blas.NonUnit,
  74                         n:  5,
  75                         k:  2,
  76                         a: [][]float64{
  77                                 {7, 3, 9, 0, 0},
  78                                 {0, 6, 7, 10, 0},
  79                                 {0, 0, 2, 1, 11},
  80                                 {0, 0, 0, 12, 3},
  81                                 {0, 0, 0, 0, -1},
  82                         },
  83                         x:   []float64{1, 2, 3, 4, 5},
  84                         ans: []float64{7, 15, 29, 71, 40},
  85                 },
  86                 {
  87                         ul: blas.Lower,
  88                         tA: blas.Trans,
  89                         d:  blas.NonUnit,
  90                         n:  5,
  91                         k:  2,
  92                         a: [][]float64{
  93                                 {7, 0, 0, 0, 0},
  94                                 {3, 6, 0, 0, 0},
  95                                 {9, 7, 2, 0, 0},
  96                                 {0, 10, 1, 12, 0},
  97                                 {0, 0, 11, 3, -1},
  98                         },
  99                         x:   []float64{1, 2, 3, 4, 5},
 100                         ans: []float64{40, 73, 65, 63, -5},
 101                 },
 102         } {
 103                 extra := 0
 104                 var aFlat []float64
 105                 if test.ul == blas.Upper {
 106                         aFlat = flattenBanded(test.a, test.k, 0)
 107                 } else {
 108                         aFlat = flattenBanded(test.a, 0, test.k)
 109                 }
 110                 incTest := func(incX, extra int) {
 111                         xnew := makeIncremented(test.x, incX, extra)
 112                         ans := makeIncremented(test.ans, incX, extra)
 113                         lda := test.k + 1
 114                         blasser.Dtbmv(test.ul, test.tA, test.d, test.n, test.k, aFlat, lda, xnew, incX)
 115                         if !dSliceTolEqual(ans, xnew) {
 116                                 t.Errorf("Case %v, Inc %v: Want %v, got %v", i, incX, ans, xnew)
 117                         }
 118                 }
 119                 incTest(1, extra)
 120                 incTest(3, extra)
 121                 incTest(-2, extra)
 122         }
 123 }