mirror of
https://github.com/gonum/gonum.git
synced 2025-10-09 00:50:16 +08:00
99 lines
3.1 KiB
Go
99 lines
3.1 KiB
Go
package discrete_test
|
|
|
|
import (
|
|
"github.com/gonum/discrete"
|
|
"testing"
|
|
)
|
|
|
|
func TestTileGraph(t *testing.T) {
|
|
tg := discrete.NewTileGraph(4, 4, false)
|
|
if tg == nil || tg.String() != "▀▀▀▀\n▀▀▀▀\n▀▀▀▀\n▀▀▀▀" {
|
|
t.Fatal("Tile graph not generated correctly")
|
|
}
|
|
|
|
tg.SetPassability(0, 1, true)
|
|
if tg == nil || tg.String() != "▀ ▀▀\n▀▀▀▀\n▀▀▀▀\n▀▀▀▀" {
|
|
t.Fatal("Passability set incorrectly")
|
|
}
|
|
|
|
tg.SetPassability(0, 1, false)
|
|
if tg == nil || tg.String() != "▀▀▀▀\n▀▀▀▀\n▀▀▀▀\n▀▀▀▀" {
|
|
t.Fatal("Passability set incorrectly")
|
|
}
|
|
|
|
tg.SetPassability(0, 1, true)
|
|
if tg == nil || tg.String() != "▀ ▀▀\n▀▀▀▀\n▀▀▀▀\n▀▀▀▀" {
|
|
t.Fatal("Passability set incorrectly")
|
|
}
|
|
|
|
tg.SetPassability(0, 2, true)
|
|
if tg == nil || tg.String() != "▀ ▀\n▀▀▀▀\n▀▀▀▀\n▀▀▀▀" {
|
|
t.Fatal("Passability set incorrectly")
|
|
}
|
|
|
|
tg.SetPassability(1, 2, true)
|
|
if tg == nil || tg.String() != "▀ ▀\n▀▀ ▀\n▀▀▀▀\n▀▀▀▀" {
|
|
t.Fatal("Passability set incorrectly")
|
|
}
|
|
|
|
tg.SetPassability(2, 2, true)
|
|
if tg == nil || tg.String() != "▀ ▀\n▀▀ ▀\n▀▀ ▀\n▀▀▀▀" {
|
|
t.Fatal("Passability set incorrectly")
|
|
}
|
|
|
|
tg.SetPassability(3, 2, true)
|
|
if tg == nil || tg.String() != "▀ ▀\n▀▀ ▀\n▀▀ ▀\n▀▀ ▀" {
|
|
t.Fatal("Passability set incorrectly")
|
|
}
|
|
|
|
if tg2, err := discrete.GenerateTileGraph("▀ ▀\n▀▀ ▀\n▀▀ ▀\n▀▀ ▀"); err != nil {
|
|
t.Error("Tile graph errored on interpreting valid template string\n▀ ▀\n▀▀ ▀\n▀▀ ▀\n▀▀ ▀")
|
|
} else if tg2.String() != "▀ ▀\n▀▀ ▀\n▀▀ ▀\n▀▀ ▀" {
|
|
t.Error("Tile graph failed to generate properly with input string\n▀ ▀\n▀▀ ▀\n▀▀ ▀\n▀▀ ▀")
|
|
}
|
|
|
|
if tg.CoordsToID(0, 0) != 0 {
|
|
t.Error("Coords to ID fails on 0,0")
|
|
} else if tg.CoordsToID(3, 3) != 15 {
|
|
t.Error("Coords to ID fails on 3,3")
|
|
} else if tg.CoordsToID(0, 3) != 3 {
|
|
t.Error("Coords to ID fails on 0,3")
|
|
} else if tg.CoordsToID(3, 0) != 12 {
|
|
t.Error("Coords to ID fails on 3,0")
|
|
}
|
|
|
|
if r, c := tg.IDToCoords(0); r != 0 || c != 0 {
|
|
t.Error("ID to Coords fails on 0,0")
|
|
} else if r, c := tg.IDToCoords(15); r != 3 || c != 3 {
|
|
t.Error("ID to Coords fails on 3,3")
|
|
} else if r, c := tg.IDToCoords(3); r != 0 || c != 3 {
|
|
t.Error("ID to Coords fails on 0,3")
|
|
} else if r, c := tg.IDToCoords(12); r != 3 || c != 0 {
|
|
t.Error("ID to Coords fails on 3,0")
|
|
}
|
|
|
|
if succ := tg.Successors(0); succ != nil || len(succ) != 0 {
|
|
t.Error("Successors for impassable tile not 0")
|
|
}
|
|
|
|
if succ := tg.Successors(2); succ == nil || len(succ) != 2 {
|
|
t.Error("Incorrect number of successors for (0,2)")
|
|
} else {
|
|
for _, s := range succ {
|
|
if s != 1 && s != 6 {
|
|
t.Error("Successor for (0,2) neither (0,1) nor (1,2)")
|
|
}
|
|
}
|
|
}
|
|
|
|
if tg.Degree(2) != 4 {
|
|
t.Error("Degree returns incorrect number for (0,2)")
|
|
}
|
|
if tg.Degree(1) != 2 {
|
|
t.Error("Degree returns incorrect number for (0,2)")
|
|
}
|
|
if tg.Degree(0) != 0 {
|
|
t.Error("Degree returns incorrect number for impassable tile (0,0)")
|
|
}
|
|
}
|