rio 0.3.7 → 0.3.8

data/test/tc/entsel.rb

@@ -0,0 +1,110 @@
+#!/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_entsel < Test::RIO::TestCase
+  @@once = false
+  @@dirs = nil
+  DN = 'd'
+  OD = 'o'
+  def self.once
+    @@once = true
+    rio(DN).rmtree.mkpath.chdir {
+      mknd(3,3)
+    }
+  end
+  def setup
+    super
+    self.class.once unless @@once
+    @exp = rio(DN).all[]
+    @lev = []
+    @lev[1] = @exp.select{ |el| el.length == 3 }
+    @lev[2] = @exp.select{ |el| el.length == 5 }
+    @lev[3] = @exp.select{ |el| el.length == 7 }
+    @olev = []
+    @olev[1] = @lev[1].map{ |el| el.sub(/^d/,'o') }
+    @olev[2] = @lev[2].map{ |el| el.sub(/^d/,'o') }
+    @olev[3] = @lev[3].map{ |el| el.sub(/^d/,'o') }
+  end
+  include RIO_TestCase
+
+  def self.mknd(n,level)
+    return unless level > 0
+    dirs = (0...n).map{|i| rio(i.to_s).mkdir}
+    dirs.each{ |d|
+      d.chdir {
+        mknd(n,level-1)
+      }
+    }
+  end
+
+  def test_skipdirs_copy1
+    odir = rio(OD).delete!
+    rio(DN).skipdirs(1) > odir
+    assert_array_equal([],odir.all[])
+  end
+  def test_skipdirs_copy2
+    odir = rio(OD).delete!
+    rio(DN).skipdirs(2) > odir
+    assert_array_equal(@olev[1],odir.all[])
+  end
+  def test_skipdirs_copy3
+    odir = rio(OD).delete!
+    rio(DN).skipdirs(3) > odir
+    assert_array_equal(@olev[1]+@olev[2],odir.all[])
+  end
+  def test_dirs_copy1
+    odir = rio(OD).delete!
+    rio(DN).dirs(1) > odir
+    assert_array_equal(@olev[1],odir.all[])
+  end
+  def test_dirs_copy2
+    odir = rio(OD).delete!
+    rio(DN).dirs(0..2) > odir
+    assert_array_equal(@olev[1]+@olev[2],odir.all[])
+  end
+  def test_dirs_copy3
+    odir = rio(OD).delete!
+    rio(DN).dirs(2) > odir
+    assert_array_equal([],odir.all[])
+  end
+  def test_dirs_array
+    rios1 = rios2str(rio(DN).dirs(1).to_a)
+    assert_equal(@lev[1],rios1) 
+    rios2 = rios2str(rio(DN).dirs(2).to_a)
+    assert_equal([],rios2)
+    rios3 = rios2str(rio(DN).dirs(3).to_a)
+    assert_equal([],rios3)
+  end
+  def test_skipdirs_array
+    rios1 = rios2str(rio(DN).skipdirs(1).to_a)
+    assert_array_equal([],rios1) 
+    rios2 = rios2str(rio(DN).skipdirs(2).to_a)
+    assert_array_equal(@lev[1],rios2)
+    rios3 = rios2str(rio(DN).skipdirs(3).to_a)
+    assert_array_equal(@lev[1],rios3)
+  end
+  def test_all_skipdirs_array
+    rios1 = rios2str(rio(DN).all.skipdirs(1).to_a)
+    assert_array_equal(@lev[2]+@lev[3],rios1) 
+    rios2 = rios2str(rio(DN).all.skipdirs(2).to_a)
+    assert_array_equal(@lev[1]+@lev[3],rios2) 
+    rios3 = rios2str(rio(DN).all.skipdirs(3).to_a)
+    assert_array_equal(@lev[1]+@lev[2],rios3) 
+  end
+  def test_all_dirs_array
+    rios1 = rios2str(rio(DN).all.dirs(1).to_a)
+    assert_equal(@lev[1],rios1) 
+    rios2 = rios2str(rio(DN).all.dirs(2).to_a)
+    assert_equal(@lev[2],rios2)
+    rios3 = rios2str(rio(DN).all.dirs(3).to_a)
+    assert_equal(@lev[3],rios3)
+  end
+  def test_sel
+  end
+end

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

@@ -86,7 +86,7 @@ class TC_get < Test::RIO::TestCase
       assert_equal(expa+expa,ans)
     end
     
-    # assure get returns nil once after iteration using #get (with auto close)
+    # assure get does not return nil once after iteration using #each (with auto close)
     begin
       f = rio('f.txt')
       ans = []
@@ -96,8 +96,8 @@ class TC_get < Test::RIO::TestCase
       end
       assert_equal(expa,ans)
 
-      rec = f.get
-      assert_nil(rec)
+      #rec = f.get
+      #assert_nil(rec)
 
       rec = f.get
       assert_equal(expa[0],rec)
@@ -114,7 +114,7 @@ class TC_get < Test::RIO::TestCase
         ans << rec
       end
       rec = f.get
-      assert_nil(rec)
+      assert_equal(expa[0],rec)
     end
 
     begin
@@ -137,15 +137,16 @@ class TC_get < Test::RIO::TestCase
     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(expa,ans)
+      assert_equal([],ans)
     end
   end
 end

data/test/tc/getrec.rb

@@ -85,9 +85,6 @@ class TC_RIO_getrec < Test::Unit::TestCase
       end
       assert_equal(expa,ans)
 
-      rec = f.getrec
-      assert_nil(rec)
-
       rec = f.getrec
       assert_equal(expa[0],rec)
     end
@@ -103,7 +100,7 @@ class TC_RIO_getrec < Test::Unit::TestCase
         ans << rec
       end
       rec = f.getrec
-      assert_nil(rec)
+      assert_equal(expa[0],rec)
     end
 
     begin
@@ -134,7 +131,7 @@ class TC_RIO_getrec < Test::Unit::TestCase
       f.each do |rec|
         ans << rec
       end
-      assert_equal(expa,ans)
+      assert_equal([],ans)
     end
   end
 end

data/test/tc/gzip.rb

@@ -0,0 +1,82 @@
+#!/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\n"
+    rio('src').print!(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').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

data/test/tc/io_each_byte.rb

@@ -0,0 +1,60 @@
+#!/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_io_each_byte < Test::RIO::TestCase
+  @@once = false
+  DATA = "012"
+  FN_RUBY = 'fruby.txt'
+  FN_RIO = 'frio.txt'
+  def self.once
+    @@once = true
+    rio(FN_RUBY) < DATA
+    rio(FN_RIO) < DATA
+  end
+  def setup
+    super
+    self.class.once unless @@once
+  end
+  def open_io(&block)
+    begin
+      fruby = ::File.open(FN_RUBY)
+      frio = rio(FN_RIO)
+      yield(fruby,frio)
+    ensure
+      fruby.close
+      frio.close
+    end
+  end
+  
+  def check_method(fruby,frio,n,sym,*args)
+    n.times {
+      assert_equal(fruby.__send__(sym,*args),frio.__send__(sym,*args))
+      assert_equal(fruby.eof?,frio.eof?)
+    }
+  end
+  def check_calls(fruby,frio,n,*args)
+    check_method(fruby,frio,n,:read,*args)
+  end
+  def check_iter(fruby,frio,sym,*args)
+    ans_ruby = []
+    ans_rio = []
+    fruby.__send__(sym,*args) { |el|
+      ans_ruby << el
+    }
+    frio.__send__(sym,*args) { |el|
+      ans_rio << el
+    }
+    assert_equal(ans_ruby,ans_rio)
+  end
+  def test_basic
+    open_io { |fruby,frio|
+      check_iter(fruby,frio,:each_byte)
+    }
+  end
+end

data/test/tc/io_read.rb

@@ -0,0 +1,80 @@
+#!/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_io_read < Test::RIO::TestCase
+  @@once = false
+  DATA = "012345"
+  FN_RUBY = 'fruby.txt'
+  FN_RIO = 'frio.txt'
+  def self.once
+    @@once = true
+    rio(FN_RUBY) < DATA
+    rio(FN_RIO) < DATA
+  end
+  def setup
+    super
+    self.class.once unless @@once
+  end
+  def open_io(&block)
+    begin
+      fruby = ::File.open(FN_RUBY)
+      frio = rio(FN_RIO)
+      yield(fruby,frio)
+    ensure
+      fruby.close
+      frio.close
+    end
+  end
+  
+  def check_method(fruby,frio,n,sym,*args)
+    n.times {
+      assert_equal(fruby.__send__(sym,*args),frio.__send__(sym,*args))
+      assert_equal(fruby.eof?,frio.eof?)
+    }
+  end
+  def check_calls(fruby,frio,n,*args)
+    check_method(fruby,frio,n,:read,*args)
+  end
+  def test_positive_integer
+    open_io { |fruby,frio|
+      check_calls(fruby,frio,3,3)
+    }
+    open_io { |fruby,frio|
+      check_calls(fruby,frio,2,4)
+    }
+    open_io { |fruby,frio|
+      check_calls(fruby,frio,2,7)
+    }
+  end
+  def test_positive_integer_buffer
+    open_io { |fruby,frio|
+      buffer = ""
+      check_calls(fruby,frio,1,3,buffer)
+      assert_equal(DATA[0...3],buffer)
+      check_calls(fruby,frio,1,3,buffer)
+      assert_equal(DATA[3...6],buffer)
+    }
+    open_io { |fruby,frio|
+      check_calls(fruby,frio,2,4)
+    }
+    open_io { |fruby,frio|
+      check_calls(fruby,frio,2,7)
+    }
+  end
+  def test_nil
+    open_io { |fruby,frio|
+      check_calls(fruby,frio,2,nil)
+    }
+  end
+  def test_noarg
+    open_io { |fruby,frio|
+      check_calls(fruby,frio,2)
+    }
+  end
+end

data/test/tc/iometh.rb

@@ -0,0 +1,149 @@
+#!/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_iometh < 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 returns nil once after iteration using #get (with auto close)
+    begin
+      f = rio('f.txt')
+      ans = []
+
+      f.each do |rec|
+        ans << rec
+      end
+      assert_equal(expa,ans)
+
+      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 = []
+
+      (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