pgraph: Filter graph should preserve vertices without an edge

There was a bug in filter graph that caused it to not preserve any
vertices that weren't bound to another with an edge. This fixes the bug
and adds a test too.
This commit is contained in:
James Shubin
2024-01-07 17:53:31 -05:00
parent d61936c06d
commit cb021e3a25
2 changed files with 27 additions and 1 deletions

View File

@@ -191,6 +191,28 @@ func TestFilterGraph1(t *testing.T) {
}
}
func TestFilterGraph2(t *testing.T) {
G, _ := NewGraph("g5")
v1 := NV("v1")
v2 := NV("v2")
v3 := NV("v3")
v4 := NV("v4")
e1 := NE("e1")
G.AddEdge(v1, v2, e1)
G.AddVertex(v3)
G.AddVertex(v4)
save := []Vertex{v1, v2, v3}
out, err := G.FilterGraph("new g5", save)
if err != nil {
t.Errorf("failed with: %v", err)
}
if c, i := len(save), out.NumVertices(); c != i {
t.Errorf("should have %d vertices instead of: %d", c, i)
}
}
func TestDisconnectedGraphs1(t *testing.T) {
G, _ := NewGraph("g6")
v1 := NV("v1")