From b65356b565286bb1d161cf88d1170168fc9b024c Mon Sep 17 00:00:00 2001 From: Dmitry Vyukov Date: Tue, 13 Aug 2019 17:02:48 +0200 Subject: prog: measure memory allocation in benchmarks Enable ReportAllocs. Also factor out common code into a helper function (duplicated in 3 places now). --- prog/mutation_test.go | 18 ++++-------------- 1 file changed, 4 insertions(+), 14 deletions(-) (limited to 'prog/mutation_test.go') diff --git a/prog/mutation_test.go b/prog/mutation_test.go index 6b50b975e..b041f02d2 100644 --- a/prog/mutation_test.go +++ b/prog/mutation_test.go @@ -220,13 +220,8 @@ func runMutationTests(t *testing.T, tests [][2]string) { } func BenchmarkMutate(b *testing.B) { - olddebug := debug - debug = false - defer func() { debug = olddebug }() - target, err := GetTarget("linux", "amd64") - if err != nil { - b.Fatal(err) - } + target, cleanup := initBench(b) + defer cleanup() ct := linuxAmd64ChoiceTable(target) const progLen = 30 p := target.Generate(rand.NewSource(0), progLen, nil) @@ -240,13 +235,8 @@ func BenchmarkMutate(b *testing.B) { } func BenchmarkGenerate(b *testing.B) { - olddebug := debug - debug = false - defer func() { debug = olddebug }() - target, err := GetTarget("linux", "amd64") - if err != nil { - b.Fatal(err) - } + target, cleanup := initBench(b) + defer cleanup() ct := linuxAmd64ChoiceTable(target) const progLen = 30 b.ResetTimer() -- cgit mrf-deployment