graph_test.go

package goraph

import (
	"fmt"
	"os"
	"testing"

	"github.com/gyuho/goraph/testgraph"
)

func TestNewGraph(t *testing.T) {
	g1 := NewGraph()
	fmt.Println("g1:", g1.String())

	f, err := os.Open("testdata/graph.json")
	if err != nil {
		t.Fatal(err)
	}
	defer f.Close()
	g2, err := NewGraphFromJSON(f, "graph_00")
	if err != nil {
		t.Fatal(err)
	}
	fmt.Println("g2:", g2.String())
}

func TestNewGraphFromJSON_graph(t *testing.T) {
	f, err := os.Open("testdata/graph.json")
	if err != nil {
		t.Fatal(err)
	}
	defer f.Close()
	jg, err := NewGraphFromJSON(f, "graph_00")
	g, ok := jg.(*graph)
	if err != nil || !ok {
		t.Fatalf("nil graph %v", err)
	}
	if g.nodeToTargets[StringID("C")][StringID("S")] != 9.0 {
		t.Fatalf("weight from C to S must be 9.0 but %f", g.nodeToTargets[StringID("C")][StringID("S")])
	}
	for _, tg := range testgraph.GraphSlice {
		f, err := os.Open("testdata/graph.json")
		if err != nil {
			t.Fatal(err)
		}
		defer f.Close()
		jg, err := NewGraphFromJSON(f, tg.Name)
		g, ok := jg.(*graph)
		if err != nil || !ok {
			t.Fatalf("nil graph %v", err)
		}
		if g.GetNodeCount() != tg.TotalNodeCount {
			t.Fatalf("%s | Expected %d but %d", tg.Name, tg.TotalNodeCount, g.GetNodeCount())
		}
		for _, elem := range tg.EdgeToWeight {
			weight1, err := g.GetWeight(StringID(elem.Nodes[0]), StringID(elem.Nodes[1]))
			if err != nil {
				t.Fatal(err)
			}
			weight2 := elem.Weight
			if weight1 != weight2 {
				t.Fatalf("Expected %f but %f", weight2, weight1)
			}
		}
	}
}

func TestNewGraphFromYAML_graph(t *testing.T) {
	f, err := os.Open("testdata/graph.yml")
	if err != nil {
		t.Fatal(err)
	}
	defer f.Close()
	jg, err := NewGraphFromYAML(f, "graph_00")
	g, ok := jg.(*graph)
	if err != nil || !ok {
		t.Fatalf("nil graph %v", err)
	}
	if g.nodeToTargets[StringID("C")][StringID("S")] != 9.0 {
		t.Fatalf("weight from C to S must be 9.0 but %f", g.nodeToTargets[StringID("C")][StringID("S")])
	}
	for _, tg := range testgraph.GraphSlice {
		f, err := os.Open("testdata/graph.yml")
		if err != nil {
			t.Fatal(err)
		}
		defer f.Close()
		jg, err := NewGraphFromYAML(f, tg.Name)
		g, ok := jg.(*graph)
		if err != nil || !ok {
			t.Fatalf("nil graph %v", err)
		}
		if g.GetNodeCount() != tg.TotalNodeCount {
			t.Fatalf("%s | Expected %d but %d", tg.Name, tg.TotalNodeCount, g.GetNodeCount())
		}
		for _, elem := range tg.EdgeToWeight {
			weight1, err := g.GetWeight(StringID(elem.Nodes[0]), StringID(elem.Nodes[1]))
			if err != nil {
				t.Fatal(err)
			}
			weight2 := elem.Weight
			if weight1 != weight2 {
				t.Fatalf("Expected %f but %f", weight2, weight1)
			}
		}
	}
}

func TestGraph_GetVertices(t *testing.T) {
	for _, tg := range testgraph.GraphSlice {
		f, err := os.Open("testdata/graph.json")
		if err != nil {
			t.Fatal(err)
		}
		defer f.Close()
		g, err := NewGraphFromJSON(f, tg.Name)
		if err != nil {
			t.Fatal(err)
		}
		if g.GetNodeCount() != tg.TotalNodeCount {
			t.Fatalf("wrong number of vertices: %s", g)
		}
	}
}

func TestGraph_Init(t *testing.T) {
	for _, tg := range testgraph.GraphSlice {
		f, err := os.Open("testdata/graph.json")
		if err != nil {
			t.Fatal(err)
		}
		defer f.Close()
		g, err := NewGraphFromJSON(f, tg.Name)
		if err != nil {
			t.Fatal(err)
		}
		g.Init()
		if g.GetNodeCount() != 0 {
			t.Fatalf("not initialized: %s", g)
		}
	}
}

func TestGraph_DeleteNode(t *testing.T) {
	f, err := os.Open("testdata/graph.json")
	if err != nil {
		t.Fatal(err)
	}
	defer f.Close()
	g, err := NewGraphFromJSON(f, "graph_01")
	if err != nil {
		t.Fatal(err)
	}
	if !g.DeleteNode(StringID("D")) {
		t.Fatal("D does not exist in the graph")
	}
	if nd, err := g.GetNode(StringID("D")); err == nil {
		t.Fatalf("No Node Expected but got %s", nd)
	}
	if v, err := g.GetSources(StringID("C")); err != nil || len(v) != 1 {
		t.Fatalf("Expected 1 edge incoming to C but %v\n\n%s", err, g)
	}
	if v, err := g.GetTargets(StringID("C")); err != nil || len(v) != 2 {
		t.Fatalf("Expected 2 edges outgoing from C but %v\n\n%s", err, g)
	}
	if v, err := g.GetTargets(StringID("F")); err != nil || len(v) != 2 {
		t.Fatalf("Expected 2 edges outgoing from F but %v\n\n%s", err, g)
	}
	if v, err := g.GetSources(StringID("F")); err != nil || len(v) != 2 {
		t.Fatalf("Expected 2 edges incoming to F but %v\n\n%s", err, g)
	}
	if v, err := g.GetTargets(StringID("B")); err != nil || len(v) != 3 {
		t.Fatalf("Expected 3 edges outgoing from B but %v\n\n%s", err, g)
	}
	if v, err := g.GetSources(StringID("E")); err != nil || len(v) != 4 {
		t.Fatalf("Expected 4 edges incoming to E but %v\n\n%s", err, g)
	}
	if v, err := g.GetTargets(StringID("E")); err != nil || len(v) != 3 {
		t.Fatalf("Expected 3 edges outgoing from E but %v\n\n%s", err, g)
	}
	if v, err := g.GetTargets(StringID("T")); err != nil || len(v) != 3 {
		t.Fatalf("Expected 3 edges outgoing from T but %v\n\n%s", err, g)
	}
}

func TestGraph_DeleteEdge(t *testing.T) {
	f, err := os.Open("testdata/graph.json")
	if err != nil {
		t.Fatal(err)
	}
	defer f.Close()

	g, err := NewGraphFromJSON(f, "graph_01")
	if err != nil {
		t.Fatal(err)
	}

	if err := g.DeleteEdge(StringID("B"), StringID("D")); err != nil {
		t.Fatal(err)
	}
	if v, err := g.GetSources(StringID("D")); err != nil || len(v) != 4 {
		t.Fatalf("Expected 4 edges incoming to D but %v\n\n%s", err, g)
	}

	if err := g.DeleteEdge(StringID("B"), StringID("C")); err != nil {
		t.Fatal(err)
	}
	if err := g.DeleteEdge(StringID("S"), StringID("C")); err != nil {
		t.Fatal(err)
	}
	if v, err := g.GetTargets(StringID("S")); err != nil || len(v) != 2 {
		t.Fatalf("Expected 2 edges outgoing from S but %v\n\n%s", err, g)
	}

	if err := g.DeleteEdge(StringID("C"), StringID("E")); err != nil {
		t.Fatal(err)
	}
	if err := g.DeleteEdge(StringID("E"), StringID("D")); err != nil {
		t.Fatal(err)
	}
	if v, err := g.GetTargets(StringID("E")); err != nil || len(v) != 3 {
		t.Fatalf("Expected 3 edges outgoing from E but %v\n\n%s", err, g)
	}
	if v, err := g.GetSources(StringID("E")); err != nil || len(v) != 3 {
		t.Fatalf("Expected 3 edges incoming to E but %v\n\n%s", err, g)
	}

	if err := g.DeleteEdge(StringID("F"), StringID("E")); err != nil {
		t.Fatal(err)
	}
	if v, err := g.GetSources(StringID("E")); err != nil || len(v) != 2 {
		t.Fatalf("Expected 2 edges incoming to E but %v\n\n%s", err, g)
	}
}

func TestGraph_ReplaceEdge(t *testing.T) {
	f, err := os.Open("testdata/graph.json")
	if err != nil {
		t.Fatal(err)
	}
	defer f.Close()
	g, err := NewGraphFromJSON(f, "graph_00")
	if err != nil {
		t.Fatal(err)
	}
	if err := g.ReplaceEdge(StringID("C"), StringID("S"), 1.0); err != nil {
		t.Fatal(err)
	}
	if v, err := g.GetWeight(StringID("C"), StringID("S")); err != nil || v != 1.0 {
		t.Fatalf("weight from C to S must be 1.0 but %v\n\n%v", err, g)
	}
}