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
|
// Copyright ©2016 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 optimize
import (
"testing"
"gonum.org/v1/gonum/mat"
"gonum.org/v1/gonum/optimize/functions"
"gonum.org/v1/gonum/stat/distmv"
)
func TestGuessAndCheck(t *testing.T) {
t.Parallel()
dim := 30
problem := Problem{
Func: functions.ExtendedRosenbrock{}.Func,
}
mu := make([]float64, dim)
sigma := mat.NewSymDense(dim, nil)
for i := 0; i < dim; i++ {
sigma.SetSym(i, i, 1)
}
d, ok := distmv.NewNormal(mu, sigma, nil)
if !ok {
panic("bad test")
}
initX := make([]float64, dim)
_, err := Minimize(problem, initX, nil, &GuessAndCheck{Rander: d})
if err != nil {
t.Errorf("unexpected error running Minimize with nil settings: %v", err)
}
settings := &Settings{}
settings.Concurrent = 5
settings.MajorIterations = 15
_, err = Minimize(problem, initX, settings, &GuessAndCheck{Rander: d})
if err != nil {
t.Errorf("unexpected error running Minimize with settings %+v: %v", settings, err)
}
}
|
