routing: introduce a heap to keep track of closest nodes during pathfinding

This commit introduces a new heap struct that will be used to keep
track of the next closest node to the source during path finding within
our modified Dijkstra's algorithm.
This commit is contained in:
Olaoluwa Osuntokun
2017-03-19 14:06:08 -07:00
parent 5d434839a4
commit 606b23df43
3 changed files with 111 additions and 7 deletions

52
routing/heap_test.go Normal file
View File

@@ -0,0 +1,52 @@
package routing
import (
"container/heap"
prand "math/rand"
"reflect"
"sort"
"testing"
"time"
)
// TestHeapOrdering ensures that the items inserted into the heap are properly
// retrieved in minimum order of distance.
func TestHeapOrdering(t *testing.T) {
// First, create a blank heap, we'll use this to push on randomly
// generated items.
var nodeHeap distanceHeap
prand.Seed(time.Now().Unix())
// Create 100 random entries adding them to the heap created above, but
// also a list that we'll sort with the entries.
const numEntries = 100
sortedEntries := make([]nodeWithDist, 0, numEntries)
for i := 0; i < numEntries; i++ {
entry := nodeWithDist{
dist: prand.Float64(),
}
heap.Push(&nodeHeap, entry)
sortedEntries = append(sortedEntries, entry)
}
// Sort the regular slice, we'll compare this against all the entries
// popped from the heap.
sort.Sort(&distanceHeap{sortedEntries})
// One by one, pop of all the entries from the heap, they should come
// out in sorted order.
var poppedEntries []nodeWithDist
for nodeHeap.Len() != 0 {
e := heap.Pop(&nodeHeap).(nodeWithDist)
poppedEntries = append(poppedEntries, e)
}
// Finally, ensure that the items popped from the heap and the items we
// sorted are identical at this rate.
if !reflect.DeepEqual(poppedEntries, sortedEntries) {
t.Fatalf("items don't match: expected %v, got %v", sortedEntries,
poppedEntries)
}
}