wishdev-rio 0.4.3.1

data/test/tc/files_select.rb

@@ -0,0 +1,92 @@
+#!/usr/local/bin/ruby
+if $0 == __FILE__
+  Dir.chdir File.dirname(__FILE__)+'/../'
+  $:.unshift File.expand_path('../lib/')
+end
+
+require 'rio'
+require 'tc/testcase'
+
+class TC_files_select < Test::RIO::TestCase
+  @@once = false
+  def self.once
+    @@once = true
+    rio('d0').delete!.mkdir
+    rio('f0').delete!.puts!("f0:L0")
+    rio('f1').delete!.puts!("f1:L0")
+    rio('d0/f2').puts!("d0/f2:L0")
+    rio('d0/f3').puts!("d0/f3:L0")
+  end
+  def setup
+    super
+    self.class.once unless @@once
+  end
+
+  def test_files_method_with_dir_rio_each
+    #$trace_states = true
+    ans = []
+    rio('d0').files.each { |f|
+      ans << f
+    }
+    assert_equal(['d0/f2','d0/f3'],smap(ans))
+  end
+  def test_files_method_with_dir_rio_each_skip
+    #$trace_states = true
+    ans = []
+    rio('d0').skip('f*').files.each { |f|
+      ans << f
+    }
+    assert(ans.empty?)
+  end
+  def test_files_method_with_file_rio_each_skip
+    #$trace_states = true
+    ans = []
+    rio('f0').skip('f*').files.each { |f|
+      ans << f
+    }
+    assert(ans.empty?)
+  end
+  def test_files_method_with_file_rio_each
+    #$trace_states = true
+    ans = []
+    rio('f0').files.each { |f|
+      ans << f
+    }
+    assert_equal(['f0'],smap(ans))
+  end
+  def test_files_and_lines_with_file_rio_each
+    #$trace_states = true
+    ans = []
+    rio('f0').files.lines.each { |f|
+      ans << f
+    }
+    assert_equal(["f0:L0\n"],smap(ans))
+  end
+  def test_files_method_with_file_rio_to_a
+    #$trace_states = true
+    ans = rio('f0').files.to_a
+    assert_equal(['f0'],smap(ans))
+  end
+  def test_files_method_with_file_rio_ss
+    #$trace_states = true
+    ans = rio('f0').files[]
+    assert_equal(['f0'],smap(ans))
+  end
+  def test_files_method_with_file_rio_ss_skip
+    #$trace_states = true
+    ans = rio('f0').skip('f*').files[]
+    assert_equal([],smap(ans))
+  end
+  def test_files_method_with_file_rio_ss_args_select
+    #$trace_states = true
+    ans = rio('f0').files['f*']
+    assert_equal(['f0'],smap(ans))
+  end
+  def test_files_method_with_file_rio_ss_args_dontselect
+    #$trace_states = true
+    ans = rio('f0').files['g*']
+    assert_equal([],smap(ans))
+  end
+
+end
+

data/test/tc/get.rb

@@ -0,0 +1,152 @@
+#!/usr/local/bin/ruby
+if $0 == __FILE__
+  Dir.chdir File.dirname(__FILE__)+'/../'
+  $:.unshift File.expand_path('../lib/')
+end
+
+require 'rio'
+require 'tc/testcase'
+
+class TC_get < Test::RIO::TestCase
+  @@once = false
+  def self.once
+    @@once = true
+  end
+  def setup
+    super
+    self.class.once unless @@once
+  end
+  def test_file
+    (0..4).each do |n_lines|
+      get_each(n_lines)
+    end
+  end
+  def get_each(n_lines)
+    f = rio('f.txt')
+    expa = (0..n_lines-1).map { |a| "Line#{a}\n" }
+    f < expa
+    
+    f = rio('f.txt')
+    assert_equal(expa,f[])
+    #  $trace_states = true
+
+    # iterate twice using #get (no auto close)
+    begin
+      f = rio('f.txt').nocloseoneof
+      ans = []
+      while rec = f.get
+        ans << rec
+      end
+      assert_equal(expa,ans)
+      while rec = f.get
+        ans << rec
+      end
+      assert_equal(expa,ans)
+    end
+
+    # iterate using #get then using #each (no auto close)
+    begin
+      f = rio('f.txt').nocloseoneof
+      ans = []
+      while rec = f.get
+        ans << rec
+      end
+      assert_equal(expa,ans)
+      f.each do |rec|
+        ans << rec
+      end
+      assert_equal(expa,ans)
+    end
+    
+    # iterate using #each then using #get (no auto close)
+    begin
+      f = rio('f.txt').nocloseoneof
+      ans = []
+      f.each do |rec|
+        ans << rec
+      end
+      assert_equal(expa,ans)
+      while rec = f.get
+        ans << rec
+      end
+      assert_equal(expa,ans)
+    end
+    
+    # iterate twice using #get (with auto close)
+    begin
+      f = rio('f.txt')
+      ans = []
+      while rec = f.get
+        ans << rec
+      end
+      assert_equal(expa,ans)
+      while rec = f.get
+        ans << rec
+      end
+      assert_equal(expa+expa,ans)
+    end
+    
+    # assure get does not return nil once after iteration using #each (with auto close)
+    begin
+      f = rio('f.txt')
+      ans = []
+
+      f.each do |rec|
+        ans << rec
+      end
+      assert_equal(expa,ans)
+
+      #rec = f.get
+      #assert_nil(rec)
+
+      rec = f.get
+      assert_equal(expa[0],rec)
+    end
+
+    begin
+      f = rio('f.txt')
+      ans = []
+
+      rec = f.get
+      assert_equal(expa[0],rec)
+      ans << rec unless rec.nil?
+      f.each do |rec|
+        ans << rec
+      end
+      rec = f.get
+      assert_equal(expa[0],rec)
+    end
+
+    begin
+      f = rio('f.txt')
+      ans = []
+
+      (0...n_lines).each do |n|
+        rec = f.get
+        assert_equal(expa[n],rec)
+      end
+      rec = f.get
+      assert_nil(rec)
+
+      f.each do |rec|
+        ans << rec
+      end
+      assert_equal(expa,ans)
+    end
+
+    begin
+      f = rio('f.txt')
+      ans = []
+      # read all the lines
+      (0...n_lines).each do |n|
+        rec = f.get
+        assert_equal(expa[n],rec)
+      end
+      assert(f.eof?)
+      f.each do |rec|
+        ans << rec
+      end
+      assert_equal([],ans)
+    end
+  end
+end

data/test/tc/getrec.rb

@@ -0,0 +1,137 @@
+#!/usr/local/bin/ruby
+if $0 == __FILE__
+  Dir.chdir File.dirname(__FILE__)+'/../'
+  $:.unshift File.expand_path('../lib/')
+end
+require 'rio'
+require 'test/unit'
+require 'test/unit/testsuite'
+
+class TC_RIO_getrec < Test::Unit::TestCase
+  def test_basic
+    qp = rio('qp')
+    rio(qp,'getrec').rmtree.mkpath.chdir do
+      (0..4).each do |n_lines|
+        getrec_each(n_lines)
+      end
+    end
+  end
+  def getrec_each(n_lines)
+    f = rio('f.txt')
+    expa = (0..n_lines-1).map { |a| "Line#{a}\n" }
+    f < expa
+
+    f = rio('f.txt')
+    assert_equal(expa,f[])
+    #  $trace_states = true
+    begin
+      f = rio('f.txt').nocloseoneof
+      ans = []
+      while rec = f.getrec
+        ans << rec
+      end
+      assert_equal(expa,ans)
+      while rec = f.getrec
+        ans << rec
+      end
+      assert_equal(expa,ans)
+    end
+    
+    begin
+      f = rio('f.txt').nocloseoneof
+      ans = []
+      while rec = f.getrec
+        ans << rec
+      end
+      assert_equal(expa,ans)
+      f.each do |rec|
+        ans << rec
+      end
+      assert_equal(expa,ans)
+    end
+    
+    begin
+      f = rio('f.txt').nocloseoneof
+      ans = []
+      f.each do |rec|
+        ans << rec
+      end
+      assert_equal(expa,ans)
+      while rec = f.getrec
+        ans << rec
+      end
+      assert_equal(expa,ans)
+    end
+    
+    begin
+      f = rio('f.txt')
+      ans = []
+      while rec = f.getrec
+        ans << rec
+      end
+      assert_equal(expa,ans)
+      while rec = f.getrec
+        ans << rec
+      end
+      assert_equal(expa+expa,ans)
+    end
+    
+    begin
+      f = rio('f.txt')
+      ans = []
+
+      f.each do |rec|
+        ans << rec
+      end
+      assert_equal(expa,ans)
+
+      rec = f.getrec
+      assert_equal(expa[0],rec)
+    end
+
+    begin
+      f = rio('f.txt')
+      ans = []
+
+      rec = f.getrec
+      assert_equal(expa[0],rec)
+      ans << rec unless rec.nil?
+      f.each do |rec|
+        ans << rec
+      end
+      rec = f.getrec
+      assert_equal(expa[0],rec)
+    end
+
+    begin
+      f = rio('f.txt')
+      ans = []
+
+      (0...n_lines).each do |n|
+        rec = f.getrec
+        assert_equal(expa[n],rec)
+      end
+      rec = f.getrec
+      assert_nil(rec)
+
+      f.each do |rec|
+        ans << rec
+      end
+      assert_equal(expa,ans)
+    end
+
+    begin
+      f = rio('f.txt')
+      ans = []
+
+      (0...n_lines).each do |n|
+        rec = f.getrec
+        assert_equal(expa[n],rec)
+      end
+      f.each do |rec|
+        ans << rec
+      end
+      assert_equal([],ans)
+    end
+  end
+end

data/test/tc/gzip.rb

@@ -0,0 +1,109 @@
+#!/usr/local/bin/ruby
+if $0 == __FILE__
+  Dir.chdir File.dirname(__FILE__)+'/../'
+  $:.unshift File.expand_path('../lib/')
+end
+
+require 'rio'
+require 'tc/testcase'
+
+class TC_gzip < Test::RIO::TestCase
+  @@once = false
+  def self.once
+    @@once = true
+  end
+  def setup()
+    super
+  end
+
+  def test_string
+    plain_string = "Hello World\n"
+
+    gzip_string = ""
+    ans_string = ""
+    rio(?",plain_string) > rio(?",gzip_string).gzip
+    rio(?",gzip_string).gzip > rio(?",ans_string)
+    assert_equal(plain_string,ans_string)
+    
+    gzip_string = ""
+    ans_string = ""
+    rio(?",gzip_string).gzip < rio(?",plain_string)
+    rio(?",ans_string) < rio(?",gzip_string).gzip
+    assert_equal(plain_string,ans_string)
+    
+  end
+
+  def test_string2
+    str = "Hello World\n"
+    src = rio(?").print!(str)
+    ans = rio(?")
+
+    ans.string = ""
+    rio(?").gzip < src > ans
+    assert_equal(str,ans.string)
+    
+    ans.string = ""
+    src | rio(?").gzip | ans
+    assert_equal(str,ans.string)
+    
+  end
+
+  def test_file
+    str = "Hello World"
+    rio('src').puts!(str)
+
+    rio('src.gz').delete!
+    rio('ans').delete!
+    rio('src') > rio('src.gz').gzip
+    rio('src.gz').gzip > rio('ans')
+    assert_equal(str,rio('ans').chomp.line[0])
+    
+    rio('src.gz').delete!
+    rio('ans').delete!
+    rio('src.gz').gzip < rio('src') 
+    rio('ans') < rio('src.gz').gzip
+    assert_equal(str,rio('ans').chomp.line[0])
+    
+  end
+
+  def test_file_each
+    str = "Hello World\n"
+    rio('src').print!(str)
+
+    rio('src.gz').delete!
+    rio('ans').delete!
+    dst = rio('src.gz').gzip
+    rio('src').each do |l|
+      dst.print(l)
+    end
+    dst.close
+    ans = rio('ans')
+    rio('src.gz').gzip do |l|
+      ans.print(l)
+    end
+    ans.close
+
+    assert_equal(str,rio('ans').contents)
+    
+#    rio('src.gz').delete!
+#    rio('ans').delete!
+#    rio('src.gz').gzip < rio('src') 
+#    rio('ans') < rio('src.gz').gzip
+#    assert_equal(str,rio('ans').contents)
+    
+  end
+
+  def test_tempfile
+    str = "Hello World\n"
+    src = rio(?").print!(str)
+    ans = rio(?")
+
+    tmp = rio(??).gzip
+    tmp < src
+    assert_equal(str,tmp.rewind.contents)
+    tmp.rewind > ans
+    assert_equal(str,ans.contents)
+  end
+
+
+end