lib/knife_draw/chef_graph.rb in knife-draw-0.9.0 vs lib/knife_draw/chef_graph.rb in knife-draw-1.0.0

@@ -39,12 +39,12 @@
 
     def connect(source, target)
       graph.add_edge(source, target, color: EDGE_COLOR)
     end
 
-    def draw!(outputfile="output.png")
+    def draw!(outputfile)
       filename=outputfile
-      format = File.extname(filename)[1..-1]
+      format = File.extname(filename)[1..-1] || :png
       graph.output format => filename
     end
   end
 end