Got #to_dot working with blocks

[graph.njae.git] / lib / graph.njae / graph.rb

diff --git a/lib/graph.njae/graph.rb b/lib/graph.njae/graph.rb
index f3d00328d4a4956087ea0a276604417a74c19b7d..eb2ddc6e3e55b296720e400cba6aea87296f38a5 100644 (file)
--- a/lib/graph.njae/graph.rb
+++ b/lib/graph.njae/graph.rb
@@ -46,7 +46,7 @@ module GraphNjae
         if vertex_block.nil?
           dot << v.to_dot(vertex_args)
         else
-          dot << v.do_dot(&vertex_block)
+          dot << v.to_dot(&vertex_block)
         end
         dot << "\n"
       end
@@ -54,7 +54,7 @@ module GraphNjae
         if edge_block.nil?
           dot << e.to_dot(edge_args)
         else
-          dot << e.do_dot(&lambda {edge_block})
+          dot << e.to_dot(&edge_block)
         end
         dot << "\n"
       end