graph: fix undirected to directed copy behaviour

gonum · Feb 14, 2018 · 326fd9d · 326fd9d
1 parent 996b88e
commit 326fd9d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/graph/graph.go b/graph/graph.go
@@ -218,7 +218,7 @@ func Copy(dst Builder, src Graph) {
 	}
 	for _, u := range nodes {
 		for _, v := range src.From(u) {
-			dst.SetEdge(src.Edge(u, v))
+			dst.SetEdge(dst.NewEdge(u, v))
 		}
 	}
 }
@@ -241,7 +241,7 @@ func CopyWeighted(dst WeightedBuilder, src Weighted) {
 	}
 	for _, u := range nodes {
 		for _, v := range src.From(u) {
-			dst.SetWeightedEdge(src.WeightedEdge(u, v))
+			dst.SetWeightedEdge(dst.NewWeightedEdge(u, v, src.WeightedEdge(u, v).Weight()))
 		}
 	}
 }