Explorar o código

update benchmark to work with new private api

Dan Frank %!s(int64=12) %!d(string=hai) anos
pai
achega
3b501aa84a
Modificáronse 1 ficheiros con 3 adicións e 1 borrados
  1. 3 1
      hostpool_test.go

+ 3 - 1
hostpool_test.go

@@ -141,7 +141,9 @@ func BenchmarkEpsilonGreedy(b *testing.B) {
 	b.StartTimer()
 	for i := 0; i < b.N; i++ {
 		if i != 0 && i%100 == 0 {
-			p.performEpsilonGreedyDecay()
+			for _, h := range p.hosts {
+				h.performDecay()
+			}
 		}
 		hostR := p.Get()
 		p.timer = &mockTimer{t: int(timings[i])}