testutils

var LineNetworkConfig = newNetworkConfigFn(func(adjacentNodes AdjacentNodes) {
	n := len(adjacentNodes)
	for i := 0; i+1 < n; i++ {
		adjacentNodes.Connect(i, i+1)
	}
})

var SmallWorldNetworkConfig = newNetworkConfigFn(func(adjacentNodes AdjacentNodes) {
	const (
		rateOfNeighborConnections = 0.2
		rateOfRandomConnections   = 0.2
	)

	n := len(adjacentNodes)
	for i := range n {

		adjacentNodes.ConnectRandomDirection(i, (i+1)%n)

		for distance := 2; distance < 4; distance++ {
			if rand.Float64() < rateOfNeighborConnections {
				adjacentNodes.ConnectRandomDirection(i, (i+distance)%n)
			}
		}

		if rand.Float64() < rateOfRandomConnections {
			adjacentNodes.ConnectRandomDirection(i, rand.Intn(n))
		}
	}
})