dfs_test.go

package dfs

import (
"github.com/stretchr/testify/assert"
"github.com/xiaomeng79/go-algorithm/data-structures/graph"
"testing"
)

funcTestUndirectedDfs(t*testing.T) {
//建立一个无向图
h:=graph.NewUndirected()

//增加顶点，
fori:=0; i<10; i++ {
h.AddVertex(graph.VertexId(i))
 }
//增加边
fori:=0; i<9; i++ {
h.AddEdge(graph.VertexId(i), graph.VertexId(i+1), 1)
 }
counter:=0
UndirectedDfs(h, graph.VertexId(4), func(id graph.VertexId) {
counter+=int(id)
 })
assert.Equal(t, 45, counter)
}

funcTestDirectedDfs(t*testing.T) {
//建立一个有向图
h:=graph.NewDirected()

fori:=0; i<10; i++ {
v:=graph.VertexId(i)
h.AddVertex(v)
 }

fori:=0; i<9; i++ {
h.AddEdge(graph.VertexId(i), graph.VertexId(i+1), 1)
 }

counter:=0
DirectedDfs(h, graph.VertexId(3), func(v graph.VertexId) {
counter+=int(v)
 })

assert.Equal(t, 42, counter)
}