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

   1 package testblas
   2
   3 import (
   4         "testing"
   5
   6         "gonum.org/v1/gonum/blas"
   7 )
   8
   9 type Dsprer interface {
  10         Dspr(ul blas.Uplo, n int, alpha float64, x []float64, incX int, a []float64)
  11 }
  12
  13 func DsprTest(t *testing.T, blasser Dsprer) {
  14         for i, test := range []struct {
  15                 ul    blas.Uplo
  16                 n     int
  17                 a     [][]float64
  18                 x     []float64
  19                 alpha float64
  20                 ans   [][]float64
  21         }{
  22                 {
  23                         ul: blas.Upper,
  24                         n:  4,
  25                         a: [][]float64{
  26                                 {10, 2, 0, 1},
  27                                 {0, 1, 2, 3},
  28                                 {0, 0, 9, 15},
  29                                 {0, 0, 0, -6},
  30                         },
  31                         x:     []float64{1, 2, 0, 5},
  32                         alpha: 8,
  33                         ans: [][]float64{
  34                                 {18, 18, 0, 41},
  35                                 {0, 33, 2, 83},
  36                                 {0, 0, 9, 15},
  37                                 {0, 0, 0, 194},
  38                         },
  39                 },
  40                 {
  41                         ul: blas.Lower,
  42                         n:  3,
  43                         a: [][]float64{
  44                                 {10, 2, 0},
  45                                 {4, 1, 2},
  46                                 {2, 7, 9},
  47                         },
  48                         x:     []float64{3, 0, 5},
  49                         alpha: 8,
  50                         ans: [][]float64{
  51                                 {82, 2, 0},
  52                                 {4, 1, 2},
  53                                 {122, 7, 209},
  54                         },
  55                 },
  56         } {
  57                 incTest := func(incX, extra int) {
  58                         xnew := makeIncremented(test.x, incX, extra)
  59                         aFlat := flattenTriangular(test.a, test.ul)
  60                         ans := flattenTriangular(test.ans, test.ul)
  61                         blasser.Dspr(test.ul, test.n, test.alpha, xnew, incX, aFlat)
  62                         if !dSliceTolEqual(aFlat, ans) {
  63                                 t.Errorf("Case %v, idx %v: Want %v, got %v.", i, incX, ans, aFlat)
  64                         }
  65                 }
  66                 incTest(1, 3)
  67                 incTest(1, 0)
  68                 incTest(3, 2)
  69                 incTest(-2, 2)
  70         }
  71 }