mirror of
https://github.com/OpenDiablo2/OpenDiablo2
synced 2025-01-09 10:57:06 -05:00
89 lines
1.9 KiB
Go
89 lines
1.9 KiB
Go
package d2astar
|
|
|
|
import (
|
|
"math"
|
|
"testing"
|
|
)
|
|
|
|
func addTruck(x, y int, label string) *Truck {
|
|
t1 := new(Truck)
|
|
t1.X = x
|
|
t1.Y = y
|
|
t1.label = label
|
|
|
|
return t1
|
|
}
|
|
|
|
func addTube(t1, t2 *Truck, cost float64) *Tube {
|
|
tube1 := new(Tube)
|
|
tube1.Cost = cost
|
|
tube1.from = t1
|
|
tube1.to = t2
|
|
|
|
t1.outTo = append(t1.outTo, *tube1)
|
|
t2.outTo = append(t2.outTo, *tube1)
|
|
|
|
return tube1
|
|
}
|
|
|
|
// Consider a world with Nodes (Trucks) and Edges (Tubes), Edges each having a cost
|
|
//
|
|
// E
|
|
// /|
|
|
// / |
|
|
// S--M
|
|
//
|
|
// S=Start at (0,0)
|
|
// E=End at (1,1)
|
|
// M=Middle at (0,1)
|
|
//
|
|
// S-M and M-E are clean clear tubes. cost: 1
|
|
//
|
|
// S-E is either:
|
|
//
|
|
// 1) TestGraphPath_ShortDiagonal : diagonal is a nice clean clear Tube , cost: 1.9
|
|
// Solver should traverse the bridge.
|
|
// Expect solution: Start, End Total cost: 1.9
|
|
//
|
|
// 1) TestGraphPath_LongDiagonal : diagonal is a Tube plugged full of
|
|
// "enormous amounts of material"!, cost: 10000.
|
|
// Solver should avoid the plugged tube.
|
|
// Expect solution Start,Middle,End Total cost: 2.0
|
|
|
|
func createGorelandGraphPathDiagonal(t *testing.T, diagonalCost, expectedDist float64) {
|
|
world := new(Goreland)
|
|
|
|
trStart := addTruck(0, 0, "Start")
|
|
trMid := addTruck(0, 1, "Middle")
|
|
trEnd := addTruck(1, 1, "End")
|
|
|
|
addTube(trStart, trEnd, diagonalCost)
|
|
addTube(trStart, trMid, 1)
|
|
addTube(trMid, trEnd, 1)
|
|
|
|
t.Logf("Goreland. Diagonal cost: %v\n\n", diagonalCost)
|
|
|
|
p, dist, found := Path(trStart, trEnd, math.MaxFloat64)
|
|
|
|
if !found {
|
|
t.Log("Could not find a path")
|
|
} else {
|
|
t.Logf("Resulting path\n%s", world.RenderPath(p))
|
|
}
|
|
|
|
if !found && expectedDist >= 0 {
|
|
t.Fatal("Could not find a path")
|
|
}
|
|
|
|
if found && dist != expectedDist {
|
|
t.Fatalf("Expected dist to be %v but got %v", expectedDist, dist)
|
|
}
|
|
}
|
|
|
|
func TestGraphPaths_ShortDiagonal(t *testing.T) {
|
|
createGorelandGraphPathDiagonal(t, 1.9, 1.9)
|
|
}
|
|
func TestGraphPaths_LongDiagonal(t *testing.T) {
|
|
createGorelandGraphPathDiagonal(t, 10000, 2.0)
|
|
}
|