| // Copyright ©2015 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 testlapack |
| |
| import ( |
| "fmt" |
| "math/rand" |
| "testing" |
| |
| "gonum.org/v1/gonum/blas" |
| ) |
| |
| type Dlaseter interface { |
| Dlaset(uplo blas.Uplo, m, n int, alpha, beta float64, a []float64, lda int) |
| } |
| |
| func DlasetTest(t *testing.T, impl Dlaseter) { |
| rnd := rand.New(rand.NewSource(1)) |
| for ti, test := range []struct { |
| m, n int |
| }{ |
| {0, 0}, |
| {1, 1}, |
| {1, 10}, |
| {10, 1}, |
| {2, 2}, |
| {2, 10}, |
| {10, 2}, |
| {11, 11}, |
| {11, 100}, |
| {100, 11}, |
| } { |
| m := test.m |
| n := test.n |
| for _, uplo := range []blas.Uplo{blas.Upper, blas.Lower, blas.All} { |
| for _, extra := range []int{0, 10} { |
| a := randomGeneral(m, n, n+extra, rnd) |
| alpha := 1.0 |
| beta := 2.0 |
| |
| impl.Dlaset(uplo, m, n, alpha, beta, a.Data, a.Stride) |
| |
| prefix := fmt.Sprintf("Case #%v: m=%v,n=%v,uplo=%v,extra=%v", |
| ti, m, n, uplo, extra) |
| if !generalOutsideAllNaN(a) { |
| t.Errorf("%v: out-of-range write to A", prefix) |
| } |
| for i := 0; i < min(m, n); i++ { |
| if a.Data[i*a.Stride+i] != beta { |
| t.Errorf("%v: unexpected diagonal of A", prefix) |
| } |
| } |
| if uplo == blas.Upper || uplo == blas.All { |
| for i := 0; i < m; i++ { |
| for j := i + 1; j < n; j++ { |
| if a.Data[i*a.Stride+j] != alpha { |
| t.Errorf("%v: unexpected upper triangle of A", prefix) |
| } |
| } |
| } |
| } |
| if uplo == blas.Lower || uplo == blas.All { |
| for i := 1; i < m; i++ { |
| for j := 0; j < min(i, n); j++ { |
| if a.Data[i*a.Stride+j] != alpha { |
| t.Errorf("%v: unexpected lower triangle of A", prefix) |
| } |
| } |
| } |
| } |
| } |
| } |
| } |
| } |