+++ /dev/null
-// 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 (
- "math"
- "testing"
-
- "golang.org/x/exp/rand"
-
- "gonum.org/v1/gonum/floats"
-)
-
-type Dorgqrer interface {
- Dorg2rer
- Dorgqr(m, n, k int, a []float64, lda int, tau, work []float64, lwork int)
-}
-
-func DorgqrTest(t *testing.T, impl Dorgqrer) {
- rnd := rand.New(rand.NewSource(1))
- // TODO(btracey): Base tests off of nb and nx.
- for _, test := range []struct{ m, n, k, lda int }{
- {10, 10, 10, 0},
- {10, 10, 10, 20},
- {30, 10, 10, 0},
- {30, 20, 10, 20},
-
- {100, 100, 100, 0},
- {100, 100, 50, 0},
- {130, 100, 100, 0},
- {130, 100, 50, 0},
- {100, 100, 100, 150},
- {100, 100, 50, 150},
- {130, 100, 100, 150},
- {130, 100, 50, 150},
-
- {200, 200, 200, 0},
- {200, 200, 150, 0},
- {230, 200, 200, 0},
- {230, 200, 150, 0},
- {200, 200, 200, 250},
- {200, 200, 150, 250},
- {230, 200, 200, 250},
- {230, 200, 150, 250},
- } {
- m := test.m
- n := test.n
- k := test.k
- lda := test.lda
- if lda == 0 {
- lda = n
- }
- a := make([]float64, m*lda)
- for i := range a {
- a[i] = rnd.Float64()
- }
- work := make([]float64, 1)
- tau := make([]float64, n)
- for i := range tau {
- tau[i] = math.NaN()
- }
- // Compute QR factorization.
- impl.Dgeqrf(m, n, a, lda, tau, work, -1)
- work = make([]float64, int(work[0]))
- impl.Dgeqrf(m, n, a, lda, tau, work, len(work))
-
- aUnblocked := make([]float64, len(a))
- copy(aUnblocked, a)
- for i := range work {
- work[i] = math.NaN()
- }
- impl.Dorg2r(m, n, k, aUnblocked, lda, tau, work)
- // make sure work isn't used before initialized
- for i := range work {
- work[i] = math.NaN()
- }
- impl.Dorgqr(m, n, k, a, lda, tau, work, len(work))
- if !floats.EqualApprox(a, aUnblocked, 1e-10) {
- t.Errorf("Q Mismatch. m = %d, n = %d, k = %d, lda = %d", m, n, k, lda)
- }
- }
-}