Store connections in Graph struct

vposloncec · Jan 26, 2024 · c9b3a9a · c9b3a9a
1 parent c93880b
commit c9b3a9a
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/base/graph.go b/base/graph.go
@@ -10,12 +10,14 @@ type Graph struct {
 	log            *zap.SugaredLogger
 	nodeLog        *zap.SugaredLogger
 	Nodes          []*Node
+	Connections    []ConnectionPair
 	printAdjacency bool
 }
 
 func NewGraph(log *zap.SugaredLogger, nodeNum int, connections []ConnectionPair) *Graph {
 	g := Graph{
-		Nodes: make([]*Node, nodeNum),
+		Nodes:       make([]*Node, nodeNum),
+		Connections: connections,
 	}
 	g.log = log.Named("Graph")
 	g.nodeLog = log.Named("Node")

diff --git a/integration_test.go b/integration_test.go
@@ -25,7 +25,8 @@ func BenchmarkStart(b *testing.B) {
 
 func TestStartRandom(t *testing.T) {
 	nodes := 155000
-	connMultiplier := 0.8
+	// multiplier pronaden na predavanju CS22W, Lecture 1.3,timestamp 12:14
+	connMultiplier := 6.3
 	connNum := int(float64(nodes) * float64(connMultiplier))
 
 	orchestration.StartRandom(nodes, connNum)