lib/visualize_ruby/edge.rb in visualize_ruby-0.1.0 vs lib/visualize_ruby/edge.rb in visualize_ruby-0.2.1

@@ -3,11 +3,11 @@
     attr_reader :name,
                 :node_a,
                 :node_b,
                 :dir
     def initialize(name: nil, nodes:, dir: :forward)
-      @name   = name
+      @name   = name.to_s if name
       @node_a = nodes[0]
       @node_b = nodes[1]
       @dir    = dir
     end
 
@@ -22,15 +22,17 @@
 
     def direction_symbol
       case dir
       when :forward
         "->"
+      when :none
+        "-"
       end
     end
 
     def inspect
       "#<VisualizeRuby::Edge #{to_a.join(" ")}>"
     end
 
     alias_method :to_s, :inspect
   end
-end
\ No newline at end of file
+end