1 // Copyright ©2015 The Gonum Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style
3 // license that can be found in the LICENSE file.
11 "golang.org/x/exp/rand"
13 "gonum.org/v1/gonum/floats"
16 type Dorgqrer interface {
18 Dorgqr(m, n, k int, a []float64, lda int, tau, work []float64, lwork int)
21 func DorgqrTest(t *testing.T, impl Dorgqrer) {
22 rnd := rand.New(rand.NewSource(1))
23 // TODO(btracey): Base tests off of nb and nx.
24 for _, test := range []struct{ m, n, k, lda int }{
55 a := make([]float64, m*lda)
59 work := make([]float64, 1)
60 tau := make([]float64, n)
64 // Compute QR factorization.
65 impl.Dgeqrf(m, n, a, lda, tau, work, -1)
66 work = make([]float64, int(work[0]))
67 impl.Dgeqrf(m, n, a, lda, tau, work, len(work))
69 aUnblocked := make([]float64, len(a))
74 impl.Dorg2r(m, n, k, aUnblocked, lda, tau, work)
75 // make sure work isn't used before initialized
79 impl.Dorgqr(m, n, k, a, lda, tau, work, len(work))
80 if !floats.EqualApprox(a, aUnblocked, 1e-10) {
81 t.Errorf("Q Mismatch. m = %d, n = %d, k = %d, lda = %d", m, n, k, lda)