From db5631c279bd4e72b76fa17f26bace0f38dee156 Mon Sep 17 00:00:00 2001
From: Daniel Cadenas <dcadenas@gmail.com>
Date: Sat, 23 Dec 2023 15:51:51 -0300
Subject: [PATCH] Refactor

---
 pagerank.go      | 4 ++--
 pagerank_test.go | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/pagerank.go b/pagerank.go
index ca9ef6b..49704d6 100644
--- a/pagerank.go
+++ b/pagerank.go
@@ -25,8 +25,8 @@ func (pr *pageRank) keyAsArrayIndex(key int) int {
 	index, ok := pr.keyToIndex[key]
 
 	if !ok {
-		pr.currentAvailableIndex++
 		index = pr.currentAvailableIndex
+		pr.currentAvailableIndex++
 		pr.keyToIndex[key] = index
 		pr.indexToKey[index] = key
 	}
@@ -146,7 +146,7 @@ func (pr *pageRank) Rank(followingProb, tolerance float64, resultFunc func(label
 func (pr *pageRank) Clear() {
 	pr.inLinks = [][]int{}
 	pr.numberOutLinks = []int{}
-	pr.currentAvailableIndex = -1
+	pr.currentAvailableIndex = 0
 	pr.keyToIndex = make(map[int]int)
 	pr.indexToKey = make(map[int]int)
 }
diff --git a/pagerank_test.go b/pagerank_test.go
index f68b483..b50f34c 100644
--- a/pagerank_test.go
+++ b/pagerank_test.go
@@ -25,7 +25,7 @@ func assertRank(t *testing.T, pageRank Interface, expected map[int]float64) {
 	const tolerance = 0.0001
 	pageRank.Rank(0.85, tolerance, func(label int, rank float64) {
 		rankAsPercentage := toPercentage(rank)
-		if math.Abs(rankAsPercentage - expected[label]) > tolerance {
+		if math.Abs(rankAsPercentage-expected[label]) > tolerance {
 			t.Error("Rank for", label, "should be", expected[label], "but was", rankAsPercentage)
 		}
 	})
@@ -235,6 +235,6 @@ func BenchmarkOneMillion(b *testing.B) {
 		result[key] = val
 	})
 
-	fmt.Println("5 first values are", result[0], ",", result[1], ",", result[2], ",", result[3], ",", result[4])
+	fmt.Println("5 first values are", result[0], ",", result[1], ",", result[2], ",", result[3], ",", result[4], " Size:", len(result))
 	pageRank.Clear()
 }