summaryrefslogtreecommitdiffstats
path: root/lib/puppet/gratr/digraph.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/puppet/gratr/digraph.rb')
-rw-r--r--lib/puppet/gratr/digraph.rb5
1 files changed, 4 insertions, 1 deletions
diff --git a/lib/puppet/gratr/digraph.rb b/lib/puppet/gratr/digraph.rb
index 2c4850d43..ae875376d 100644
--- a/lib/puppet/gratr/digraph.rb
+++ b/lib/puppet/gratr/digraph.rb
@@ -68,7 +68,10 @@ module GRATR
# Reverse all edges in a graph
def reversal
return new(self) unless directed?
- edges.inject(self.class.new) {|a,e| a << e.reverse}
+ result = self.class.new
+ edges.inject(result) {|a,e| a << e.reverse}
+ vertices.each { |v| result.add_vertex!(v) unless result.vertex?(v) }
+ result
end
# Return true if the Graph is oriented.