tgf 1.0.0 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 778b65f8f590c0ee1d45862fe2087033e450227c
-  data.tar.gz: b114f6d6fdf8dfbf56bec05d18927c3a1c154756
+  metadata.gz: a96992f1bba65c52fa98bb5bd3807745adbf861c
+  data.tar.gz: c2ca6531648d0657b8a62d6ff49136114a7d72a1
 SHA512:
-  metadata.gz: ccafbf74c3dcb2af553044a5d5e43007100b062fedeb0f7effba37c262ca85c274ff0fc3ec609fbf59e279a168ffff1dd8e0f0f9ba496fe568b1b8fb85cc5a91
-  data.tar.gz: e8e5a7082ceff72e218d1621b3bfa807b35763c7f7594070c6376212dcb0a319f4be330fd448cb4b8a6da2137e715fe3dc20741698544dd395bbff60956ccafa
+  metadata.gz: 100aa7a1f6f20e75bdf0c50cca73255bc4091f9755321ef0e7a6b94dfbca315f6a4cba5be01840dce497eaf47acecbfe84a07dd2c495aebeb525995ed10f62a5
+  data.tar.gz: d2ebfa4fc348dab9595ce5cf8893110aa33088628100cb6429345980459ca243c530f04d72337ac024d0ddec52047899b85a22b34a0ac5813ed3c1b624bd1456

data/lib/tgf.rb

@@ -40,4 +40,27 @@ module TGF
 
     return nodes, edges
   end
+
+  ##
+  # Prints a set of +TGF::Node+s and +TGF::Edge+s to +file+.
+  # +file+ can be any object that responds to +<<+; the graph
+  # is passed line by line to +<<+, and the return value is
+  # used as the object for the next call. The value +<<+ returns
+  # in the final invocation is returned by +write+.
+
+  def self.write file, nodes = [], edges = [], sep = $/
+    nodes.each do |node|
+      label = node.label.nil? ? '' : " #{node.label}"
+      file <<= "#{node.id}#{label}#{sep}"
+    end
+    return file if edges.empty?
+
+    file <<= "##{sep}"
+    edges.each do |edge|
+      label = edge.label.nil? ? '' : " #{edge.label}"
+      file <<= "#{edge.from} #{edge.to}#{label}#{sep}"
+    end
+
+    file    
+  end
 end

data/test/test_tgf.rb

@@ -1,7 +1,8 @@
 require 'minitest/autorun'
 require 'tgf'
+require 'tempfile'
 
-class TGFTest < Minitest::Test
+class TGFParseTest < Minitest::Test
   def test_simple
     nodes, edges = TGF.parse "1\n2\n#\n1 2\n"
     assert_equal 2, nodes.size
@@ -50,4 +51,63 @@ class TGFTest < Minitest::Test
     assert_equal '2', edges[0].to
     assert_equal 'hat', edges[0].label
   end
+
+  def test_file_simple
+    file = Tempfile.new('tgf')
+    file.write "1\n2\n#\n1 2\n"
+    file.rewind
+    nodes, edges = TGF.parse file
+    assert_equal 2, nodes.size
+    assert_equal 1, edges.size
+
+    assert_equal '1', nodes[0].id
+    assert_nil nodes[0].label
+
+    assert_equal '2', nodes[1].id
+    assert_nil nodes[1].label
+
+    assert_equal '1', edges[0].from
+    assert_equal '2', edges[0].to
+    assert_nil edges[0].label
+    file.close!
+  end
+end
+
+class TGFWriteTest < Minitest::Test
+  def node id, label = nil
+    TGF::Node.new (label.nil? ? id.to_s : "#{id} #{label}")
+  end
+
+  def edge from, to, label = nil
+    TGF::Edge.new (label.nil? ? "#{from} #{to}" : "#{from} #{to} #{label}")
+  end
+
+  def test_simple
+    text = TGF.write '', [node(1), node(2)], [edge(1,2)]
+    assert_equal "1\n2\n#\n1 2\n", text
+  end
+
+  def test_custom_separator
+    text = TGF.write '', [node(1), node(2)], [edge(1,2)], ';'
+    assert_equal "1;2;#;1 2;", text
+  end
+
+  def test_node_label
+    text = TGF.write '', [node(1, 'hat')]
+    assert_equal "1 hat\n", text
+  end
+
+  def test_edge_label
+    text = TGF.write '', [], [edge(1, 2, 'hat')]
+    assert_equal "#\n1 2 hat\n", text
+  end
+
+  def test_file_simple
+    file = Tempfile.new('tgf')
+    TGF.write file, [node(1), node(2)], [edge(1,2)]
+    file.rewind
+    assert_equal "1\n2\n#\n1 2\n", file.read
+    file.close!
+  end
 end
+

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: tgf
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.1.0
 platform: ruby
 authors:
 - Remis
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2014-04-29 00:00:00.000000000 Z
+date: 2014-05-06 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake
@@ -24,7 +24,7 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '0.9'
-description: A gem for parsing Trivial Graph Format files or strings
+description: A gem for parsing & writing Trivial Graph Format files or strings
 email: remis.thoughts@gmail.com
 executables: []
 extensions: []