package/gonum
1
0
Fork 0
mirror of https://github.com/gonum/gonum.git synced 2025-10-08 08:30:14 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

graph/encoding/dot: fix line visit marking

Browse Source
This commit is contained in:
Dan Kortschak
2020-11-28 21:57:14 +10:30
parent d24ce68544
commit e2a07cfc01
2 changed files with 85 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

71
graph/encoding/dot/decode_test.go
View File

@@ -393,6 +393,77 @@ func TestMultigraphDecoding(t *testing.T) {
}
}
func TestMultigraphLineIDsharing(t *testing.T) {
for i, test := range []struct {
directed bool
lines []multi.Line
expected string
}{
{
directed: true,
lines: []multi.Line{
{F: multi.Node(0), T: multi.Node(1), UID: 0},
{F: multi.Node(0), T: multi.Node(1), UID: 1},
{F: multi.Node(0), T: multi.Node(2), UID: 0},
{F: multi.Node(2), T: multi.Node(0), UID: 0},
},
expected: directedMultigraph,
},
{
directed: false,
lines: []multi.Line{
{F: multi.Node(0), T: multi.Node(1), UID: 0},
{F: multi.Node(0), T: multi.Node(1), UID: 1},
{F: multi.Node(0), T: multi.Node(2), UID: 0},
{F: multi.Node(0), T: multi.Node(2), UID: 1},
},
expected: undirectedMultigraph,
},
{
directed: true,
lines: []multi.Line{
{F: multi.Node(0), T: multi.Node(0), UID: 0},
{F: multi.Node(0), T: multi.Node(0), UID: 1},
{F: multi.Node(1), T: multi.Node(1), UID: 0},
{F: multi.Node(1), T: multi.Node(1), UID: 1},
},
expected: directedSelfLoopMultigraph,
},
{
directed: false,
lines: []multi.Line{
{F: multi.Node(0), T: multi.Node(0), UID: 0},
{F: multi.Node(0), T: multi.Node(0), UID: 1},
{F: multi.Node(1), T: multi.Node(1), UID: 0},
{F: multi.Node(1), T: multi.Node(1), UID: 1},
},
expected: undirectedSelfLoopMultigraph,
},
} {
var dst encoding.MultiBuilder
if test.directed {
dst = multi.NewDirectedGraph()
} else {
dst = multi.NewUndirectedGraph()
}
for _, l := range test.lines {
dst.SetLine(l)
}
buf, err := MarshalMulti(dst, "", "", "\t")
if err != nil {
t.Errorf("i=%d: unable to marshal graph; %v", i, dst)
continue
}
actual := string(buf)
if actual != test.expected {
t.Errorf("i=%d: graph content mismatch; want:\n%s\n\nactual:\n%s", i, test.expected, actual)
continue
}
}
}
const directedMultigraph = `digraph {
// Node definitions.
0;