1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
|
// 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"
"testing"
"golang.org/x/exp/rand"
"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=%c,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)
}
}
}
}
}
}
}
}
|
