From 0e8f14dcc4db15a509dc2ba4b291e2c28fc51ca0 Mon Sep 17 00:00:00 2001 From: Dmitry Vyukov Date: Thu, 24 Nov 2022 12:33:42 +0100 Subject: prog: refactor HeatMap Provide NumMutations method instead of Size. It allows HeatMap to choose number of mutations better (e.g. for completely empty/flat images w/o interesting data). --- prog/heatmap_test.go | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'prog/heatmap_test.go') diff --git a/prog/heatmap_test.go b/prog/heatmap_test.go index 4103a4f97..9c891f258 100644 --- a/prog/heatmap_test.go +++ b/prog/heatmap_test.go @@ -57,10 +57,10 @@ func TestGenericHeatmap(t *testing.T) { t.Fatalf("bad decode: %v", err) } for i := 0; i < iters; i++ { - hm := MakeGenericHeatmap(data).(*GenericHeatmap) + hm := MakeGenericHeatmap(data, r).(*GenericHeatmap) for j := 0; j < tries; j++ { - index := hm.ChooseLocation(r) + index := hm.ChooseLocation() if !checkIndex(index, len(data), test.regions) { hm.debugPrint(t, data, test.regions) t.Fatalf("selected index %d does not fall in a region\n", index) -- cgit mrf-deployment