diff options
Diffstat (limited to 'prog/size_test.go')
| -rw-r--r-- | prog/size_test.go | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/prog/size_test.go b/prog/size_test.go index 49945273b..6e8a28c4d 100644 --- a/prog/size_test.go +++ b/prog/size_test.go @@ -10,8 +10,9 @@ import ( func TestAssignSizeRandom(t *testing.T) { target, rs, iters := initTest(t) + ct := target.DefaultChoiceTable() for i := 0; i < iters; i++ { - p := target.Generate(rs, 10, nil) + p := target.Generate(rs, 10, ct) data0 := p.Serialize() for _, call := range p.Calls { target.assignSizesCall(call) @@ -19,7 +20,7 @@ func TestAssignSizeRandom(t *testing.T) { if data1 := p.Serialize(); !bytes.Equal(data0, data1) { t.Fatalf("different lens assigned, initial:\n%s\nnew:\n%s\n", data0, data1) } - p.Mutate(rs, 10, nil, nil) + p.Mutate(rs, 10, ct, nil) p.Serialize() for _, call := range p.Calls { target.assignSizesCall(call) |