github.com/prysmaticlabs/prysm@v1.4.4/beacon-chain/p2p/peers/scorers/scorers_test.go (about)

     1  package scorers_test
     2  
     3  import (
     4  	"io/ioutil"
     5  	"math"
     6  	"testing"
     7  
     8  	"github.com/prysmaticlabs/prysm/beacon-chain/p2p/peers/scorers"
     9  	"github.com/prysmaticlabs/prysm/cmd/beacon-chain/flags"
    10  	"github.com/prysmaticlabs/prysm/shared/featureconfig"
    11  	"github.com/sirupsen/logrus"
    12  )
    13  
    14  func TestMain(m *testing.M) {
    15  	logrus.SetLevel(logrus.DebugLevel)
    16  	logrus.SetOutput(ioutil.Discard)
    17  
    18  	resetCfg := featureconfig.InitWithReset(&featureconfig.Flags{
    19  		EnablePeerScorer: true,
    20  	})
    21  	defer resetCfg()
    22  
    23  	resetFlags := flags.Get()
    24  	flags.Init(&flags.GlobalFlags{
    25  		BlockBatchLimit:            64,
    26  		BlockBatchLimitBurstFactor: 10,
    27  	})
    28  	defer func() {
    29  		flags.Init(resetFlags)
    30  	}()
    31  	m.Run()
    32  }
    33  
    34  // roundScore returns score rounded in accordance with the score manager's rounding factor.
    35  func roundScore(score float64) float64 {
    36  	return math.Round(score*scorers.ScoreRoundingFactor) / scorers.ScoreRoundingFactor
    37  }