diff options
Diffstat (limited to 'prog/minimization_test.go')
| -rw-r--r-- | prog/minimization_test.go | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/prog/minimization_test.go b/prog/minimization_test.go index e0736206a..7862bcc01 100644 --- a/prog/minimization_test.go +++ b/prog/minimization_test.go @@ -251,7 +251,7 @@ func TestMinimize(t *testing.T) { if err != nil { t.Fatalf("failed to deserialize original program #%v: %v", ti, err) } - p1, ci := Minimize(p, test.callIndex, MinimizeParams{}, test.pred) + p1, ci := Minimize(p, test.callIndex, MinimizeCorpus, test.pred) res := p1.Serialize() if string(res) != test.result { t.Fatalf("minimization produced wrong result #%v\norig:\n%v\nexpect:\n%v\ngot:\n%v", @@ -270,10 +270,10 @@ func TestMinimizeRandom(t *testing.T) { ct := target.DefaultChoiceTable() r := rand.New(rs) for i := 0; i < iters; i++ { - for _, crash := range []bool{false, true} { + for _, mode := range []MinimizeMode{MinimizeCorpus, MinimizeCrash} { p := target.Generate(rs, 5, ct) copyP := p.Clone() - minP, _ := Minimize(p, len(p.Calls)-1, MinimizeParams{Light: crash}, func(p1 *Prog, callIndex int) bool { + minP, _ := Minimize(p, len(p.Calls)-1, mode, func(p1 *Prog, callIndex int) bool { if r.Intn(2) == 0 { return false } @@ -296,7 +296,11 @@ func TestMinimizeCallIndex(t *testing.T) { for i := 0; i < iters; i++ { p := target.Generate(rs, 5, ct) ci := r.Intn(len(p.Calls)) - p1, ci1 := Minimize(p, ci, MinimizeParams{Light: r.Intn(2) == 0}, func(p1 *Prog, callIndex int) bool { + mode := MinimizeCorpus + if r.Intn(2) == 0 { + mode = MinimizeCrash + } + p1, ci1 := Minimize(p, ci, mode, func(p1 *Prog, callIndex int) bool { return r.Intn(2) == 0 }) if ci1 < 0 || ci1 >= len(p1.Calls) || p.Calls[ci].Meta.Name != p1.Calls[ci1].Meta.Name { |