Fix for numberic labels, tested. Fixes #59.

glejeune · Dec 5, 2012 · 7fc6632 · 7fc6632
1 parent 3c8c8f5
commit 7fc6632
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/lib/graphviz/types/lbl_string.rb b/lib/graphviz/types/lbl_string.rb
@@ -6,7 +6,7 @@ def check(data)
       end
 
       def output
-        html = /^<([<|(^<)*<].*)>$/m.match(@data)
+        html = /^<([<|(^<)*<].*)>$/m.match(@data.to_s)
         if html != nil
           "<#{html[1]}>"
         else

diff --git a/test/test_graph.rb b/test/test_graph.rb
@@ -105,4 +105,13 @@ def test_search
      assert_equal c1.find_node("a0"), a0
      assert_nil   c1.search_node("a0")
   end
+
+  def test_to_s
+    assert_nothing_raised 'to_s with edge with numeric label failed.' do
+      a = @graph.add_nodes('a')
+      b = @graph.add_nodes('b')
+      @graph.add_edges(a, b, label: 5)
+      @graph.to_s
+    end
+  end
 end