Hulk did something

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

diff --git a/vendor/gonum.org/v1/gonum/blas/testblas/dsyr2k.go b/vendor/gonum.org/v1/gonum/blas/testblas/dsyr2k.go
new file mode 100644 (file)
index 0000000..7c8600b
--- /dev/null
+++ b/vendor/gonum.org/v1/gonum/blas/testblas/dsyr2k.go
@@ -0,0 +1,201 @@
+package testblas
+
+import (
+       "testing"
+
+       "gonum.org/v1/gonum/blas"
+       "gonum.org/v1/gonum/floats"
+)
+
+type Dsyr2ker interface {
+       Dsyr2k(ul blas.Uplo, tA blas.Transpose, n, k int, alpha float64, a []float64, lda int, b []float64, ldb int, beta float64, c []float64, ldc int)
+}
+
+func Dsyr2kTest(t *testing.T, blasser Dsyr2ker) {
+       for i, test := range []struct {
+               ul    blas.Uplo
+               tA    blas.Transpose
+               n     int
+               k     int
+               alpha float64
+               a     [][]float64
+               b     [][]float64
+               c     [][]float64
+               beta  float64
+               ans   [][]float64
+       }{
+               {
+                       ul:    blas.Upper,
+                       tA:    blas.NoTrans,
+                       n:     3,
+                       k:     2,
+                       alpha: 0,
+                       a: [][]float64{
+                               {1, 2},
+                               {3, 4},
+                               {5, 6},
+                       },
+                       b: [][]float64{
+                               {7, 8},
+                               {9, 10},
+                               {11, 12},
+                       },
+                       c: [][]float64{
+                               {1, 2, 3},
+                               {0, 5, 6},
+                               {0, 0, 9},
+                       },
+                       beta: 2,
+                       ans: [][]float64{
+                               {2, 4, 6},
+                               {0, 10, 12},
+                               {0, 0, 18},
+                       },
+               },
+               {
+                       ul:    blas.Lower,
+                       tA:    blas.NoTrans,
+                       n:     3,
+                       k:     2,
+                       alpha: 0,
+                       a: [][]float64{
+                               {1, 2},
+                               {3, 4},
+                               {5, 6},
+                       },
+                       b: [][]float64{
+                               {7, 8},
+                               {9, 10},
+                               {11, 12},
+                       },
+                       c: [][]float64{
+                               {1, 0, 0},
+                               {2, 3, 0},
+                               {4, 5, 6},
+                       },
+                       beta: 2,
+                       ans: [][]float64{
+                               {2, 0, 0},
+                               {4, 6, 0},
+                               {8, 10, 12},
+                       },
+               },
+               {
+                       ul:    blas.Upper,
+                       tA:    blas.NoTrans,
+                       n:     3,
+                       k:     2,
+                       alpha: 3,
+                       a: [][]float64{
+                               {1, 2},
+                               {3, 4},
+                               {5, 6},
+                       },
+                       b: [][]float64{
+                               {7, 8},
+                               {9, 10},
+                               {11, 12},
+                       },
+                       c: [][]float64{
+                               {1, 2, 3},
+                               {0, 4, 5},
+                               {0, 0, 6},
+                       },
+                       beta: 2,
+                       ans: [][]float64{
+                               {140, 250, 360},
+                               {0, 410, 568},
+                               {0, 0, 774},
+                       },
+               },
+               {
+                       ul:    blas.Lower,
+                       tA:    blas.NoTrans,
+                       n:     3,
+                       k:     2,
+                       alpha: 3,
+                       a: [][]float64{
+                               {1, 2},
+                               {3, 4},
+                               {5, 6},
+                       },
+                       b: [][]float64{
+                               {7, 8},
+                               {9, 10},
+                               {11, 12},
+                       },
+                       c: [][]float64{
+                               {1, 0, 0},
+                               {2, 4, 0},
+                               {3, 5, 6},
+                       },
+                       beta: 2,
+                       ans: [][]float64{
+                               {140, 0, 0},
+                               {250, 410, 0},
+                               {360, 568, 774},
+                       },
+               },
+               {
+                       ul:    blas.Upper,
+                       tA:    blas.Trans,
+                       n:     3,
+                       k:     2,
+                       alpha: 3,
+                       a: [][]float64{
+                               {1, 3, 5},
+                               {2, 4, 6},
+                       },
+                       b: [][]float64{
+                               {7, 9, 11},
+                               {8, 10, 12},
+                       },
+                       c: [][]float64{
+                               {1, 2, 3},
+                               {0, 4, 5},
+                               {0, 0, 6},
+                       },
+                       beta: 2,
+                       ans: [][]float64{
+                               {140, 250, 360},
+                               {0, 410, 568},
+                               {0, 0, 774},
+                       },
+               },
+               {
+                       ul:    blas.Lower,
+                       tA:    blas.Trans,
+                       n:     3,
+                       k:     2,
+                       alpha: 3,
+                       a: [][]float64{
+                               {1, 3, 5},
+                               {2, 4, 6},
+                       },
+                       b: [][]float64{
+                               {7, 9, 11},
+                               {8, 10, 12},
+                       },
+                       c: [][]float64{
+                               {1, 0, 0},
+                               {2, 4, 0},
+                               {3, 5, 6},
+                       },
+                       beta: 2,
+                       ans: [][]float64{
+                               {140, 0, 0},
+                               {250, 410, 0},
+                               {360, 568, 774},
+                       },
+               },
+       } {
+               aFlat := flatten(test.a)
+               bFlat := flatten(test.b)
+               cFlat := flatten(test.c)
+               ansFlat := flatten(test.ans)
+               blasser.Dsyr2k(test.ul, test.tA, test.n, test.k, test.alpha, aFlat, len(test.a[0]), bFlat, len(test.b[0]), test.beta, cFlat, len(test.c[0]))
+               if !floats.EqualApprox(ansFlat, cFlat, 1e-14) {
+                       t.Errorf("Case %v. Want %v, got %v.", i, ansFlat, cFlat)
+               }
+       }
+}