Hulk did something

[bytom/vapor.git] / vendor / gonum.org / v1 / gonum / blas / testblas / dtrmv.go

diff --git a/vendor/gonum.org/v1/gonum/blas/testblas/dtrmv.go b/vendor/gonum.org/v1/gonum/blas/testblas/dtrmv.go
new file mode 100644 (file)
index 0000000..8f3b6d3
--- /dev/null
+++ b/vendor/gonum.org/v1/gonum/blas/testblas/dtrmv.go
@@ -0,0 +1,147 @@
+package testblas
+
+import (
+       "testing"
+
+       "gonum.org/v1/gonum/blas"
+       "gonum.org/v1/gonum/floats"
+)
+
+type Dtrmver interface {
+       Dtrmv(ul blas.Uplo, tA blas.Transpose, d blas.Diag, n int, a []float64, lda int, x []float64, incX int)
+}
+
+func DtrmvTest(t *testing.T, blasser Dtrmver) {
+       for i, test := range []struct {
+               n   int
+               a   [][]float64
+               x   []float64
+               d   blas.Diag
+               ul  blas.Uplo
+               tA  blas.Transpose
+               ans []float64
+       }{
+               {
+                       n:   1,
+                       a:   [][]float64{{5}},
+                       x:   []float64{2},
+                       d:   blas.NonUnit,
+                       ul:  blas.Upper,
+                       tA:  blas.NoTrans,
+                       ans: []float64{10},
+               },
+               {
+                       n:   1,
+                       a:   [][]float64{{5}},
+                       x:   []float64{2},
+                       d:   blas.Unit,
+                       ul:  blas.Upper,
+                       tA:  blas.NoTrans,
+                       ans: []float64{2},
+               },
+               {
+                       n: 3,
+                       a: [][]float64{
+                               {5, 6, 7},
+                               {0, 9, 10},
+                               {0, 0, 13},
+                       },
+                       x:   []float64{3, 4, 5},
+                       d:   blas.NonUnit,
+                       ul:  blas.Upper,
+                       tA:  blas.NoTrans,
+                       ans: []float64{74, 86, 65},
+               },
+               {
+                       n: 3,
+                       a: [][]float64{
+                               {5, 6, 7},
+                               {0, 9, 10},
+                               {0, 0, 13},
+                       },
+                       x:   []float64{3, 4, 5},
+                       d:   blas.Unit,
+                       ul:  blas.Upper,
+                       tA:  blas.NoTrans,
+                       ans: []float64{62, 54, 5},
+               },
+               {
+                       n: 3,
+                       a: [][]float64{
+                               {5, 0, 0},
+                               {6, 9, 0},
+                               {7, 10, 13},
+                       },
+                       x:   []float64{3, 4, 5},
+                       d:   blas.NonUnit,
+                       ul:  blas.Lower,
+                       tA:  blas.NoTrans,
+                       ans: []float64{15, 54, 126},
+               },
+               {
+                       n: 3,
+                       a: [][]float64{
+                               {1, 0, 0},
+                               {6, 1, 0},
+                               {7, 10, 1},
+                       },
+                       x:   []float64{3, 4, 5},
+                       d:   blas.Unit,
+                       ul:  blas.Lower,
+                       tA:  blas.NoTrans,
+                       ans: []float64{3, 22, 66},
+               },
+               {
+                       n: 3,
+                       a: [][]float64{
+                               {5, 6, 7},
+                               {0, 9, 10},
+                               {0, 0, 13},
+                       },
+                       x:   []float64{3, 4, 5},
+                       d:   blas.NonUnit,
+                       ul:  blas.Upper,
+                       tA:  blas.Trans,
+                       ans: []float64{15, 54, 126},
+               },
+               {
+                       n: 3,
+                       a: [][]float64{
+                               {1, 6, 7},
+                               {0, 1, 10},
+                               {0, 0, 1},
+                       },
+                       x:   []float64{3, 4, 5},
+                       d:   blas.Unit,
+                       ul:  blas.Upper,
+                       tA:  blas.Trans,
+                       ans: []float64{3, 22, 66},
+               },
+               {
+                       n: 3,
+                       a: [][]float64{
+                               {5, 0, 0},
+                               {6, 9, 0},
+                               {7, 10, 13},
+                       },
+                       x:   []float64{3, 4, 5},
+                       d:   blas.NonUnit,
+                       ul:  blas.Lower,
+                       tA:  blas.Trans,
+                       ans: []float64{74, 86, 65},
+               },
+       } {
+               incTest := func(incX, extra int) {
+                       aFlat := flatten(test.a)
+                       x := makeIncremented(test.x, incX, extra)
+                       blasser.Dtrmv(test.ul, test.tA, test.d, test.n, aFlat, test.n, x, incX)
+                       ans := makeIncremented(test.ans, incX, extra)
+                       if !floats.EqualApprox(x, ans, 1e-14) {
+                               t.Errorf("Case %v, idx %v: Want %v, got %v.", i, incX, ans, x)
+                       }
+               }
+               incTest(1, 3)
+               incTest(-3, 3)
+               incTest(4, 3)
+       }
+}