scout-gear 7.2.0 → 7.3.0

data/test/scout/tsv/test_index.rb

@@ -1,8 +1,25 @@
 require File.expand_path(__FILE__).sub(%r(/test/.*), '/test/test_helper.rb')
 require File.expand_path(__FILE__).sub(%r(.*/test/), '').sub(/test_(.*)\.rb/,'\1')
 
+require 'scout/tsv'
 class TestTSVIndex < Test::Unit::TestCase
-  def test_true
+  def load_segment_data(data)
+    tsv = TSV.open(data, type: :list, :sep=>":", :cast => proc{|e| e =~ /(\s*)(_*)/; ($1.length..($1.length + $2.length - 1))})
+
+    tsv = tsv.add_field "Start" do |key, values|
+      values["Range"].first
+    end
+
+    tsv = tsv.add_field "End" do |key, values|
+      values["Range"].last
+    end
+
+    tsv = tsv.slice ["Start", "End"]
+ 
+    tsv
+  end
+
+  def test_index
     content =<<-'EOF'
 #: :sep=/\s+/#:type=:double#:merge=:concat
 #Id    ValueA    ValueB    OtherID
@@ -20,12 +37,34 @@ row2    a    b    id3
     end
 
     TmpFile.with_file(content) do |filename|
-      index = TSV.index(filename, :target => "ValueB", :fields => "OtherID")
+      index = TSV.index(filename, :target => "ValueB", :fields => ["OtherID"])
       assert_equal 'B', index["a"]
       assert_nil index["B"]
     end
   end
 
+  def test_from_tsv
+    content =<<-'EOF'
+#: :sep=/\s+/#:type=:double#:merge=:concat
+#Id    ValueA    ValueB    OtherID
+row1    a|aa|aaa    b    Id1|Id2
+row2    A    B    Id3|a
+row2    a    b    id3
+    EOF
+
+    TmpFile.with_file(content) do |filename|
+      tsv = TSV.open(filename)
+      index = TSV.index(tsv, :target => "ValueB", :fields => "OtherID")
+      assert_equal 'B', index["a"]
+      assert_nil index["B"]
+
+      index = tsv.index(:target => "ValueB", :fields => "OtherID")
+      assert_equal 'B', index["a"]
+      assert_nil index["B"]
+    end
+  end
+
+
   def test_persist
     content =<<-'EOF'
 #: :sep=/\s+/#:type=:double#:merge=:concat
@@ -43,6 +82,59 @@ row2    a    b    id3
     end
   end
 
+  def test_range_index
+    data =<<-EOF
+# 012345678901234567890
+#ID:Range
+a:   ______
+b: ______
+c:    _______
+d:  ____
+e:    ______
+f:             ___
+g:         ____
+    EOF
+    TmpFile.with_file(data) do |datafile|
+      tsv = load_segment_data(datafile)
+      f   = tsv.range_index("Start", "End", :persist => true)
+
+      assert_equal %w(), f[0].sort
+      assert_equal %w(b), f[1].sort
+      assert_equal %w(), f[20].sort
+      assert_equal %w(), f[(20..100)].sort
+      assert_equal %w(a b d), f[3].sort
+      assert_equal %w(a b c d e), f[(3..4)].sort
+    end
+  end
+
+  def test_range_index_persistent
+    data =<<-EOF
+# 012345678901234567890
+#ID:Range
+a:   ______
+b: ______
+c:    _______
+d:  ____
+e:    ______
+f:             ___
+g:         ____
+    EOF
+    TmpFile.with_file(data) do |datafile|
+      load_segment_data(datafile)
+      TmpFile.with_file(load_segment_data(datafile).to_s) do |tsvfile|
+        f = TSV.range_index(tsvfile, "Start", "End", :persist => true)
+
+        assert_equal %w(), f[0].sort
+        assert_equal %w(b), f[1].sort
+        assert_equal %w(), f[20].sort
+        assert_equal %w(), f[(20..100)].sort
+        assert_equal %w(a b d), f[3].sort
+        assert_equal %w(a b c d e), f[(3..4)].sort
+      end
+    end
+  end
+
+
   def __test_speed
     content =<<-'EOF'
 #: :sep=/\s+/#:type=:double#:merge=:concat

data/test/scout/tsv/test_open.rb

@@ -0,0 +1,9 @@
+require File.expand_path(__FILE__).sub(%r(/test/.*), '/test/test_helper.rb')
+require File.expand_path(__FILE__).sub(%r(.*/test/), '').sub(/test_(.*)\.rb/,'\1')
+
+class TestOpenTraverse < Test::Unit::TestCase
+  def test_true
+    assert true
+  end
+end
+

data/test/scout/tsv/test_parser.rb

@@ -29,15 +29,16 @@ class TestTSVParser < Test::Unit::TestCase
     assert_equal (1..10).collect{|v| [v,v] }, values
   end
 
-  def __test_benchmark
+  def ___test_benchmark
     num = 10_000
     txt = num.times.inject(nil) do |acc,i|
       (acc.nil? ? "" : acc << "\n") << (0..10).collect{|v| v == 0 ? i : [v,v] * "|" } * "\t"
     end 
 
     txt = StringIO.new(([txt] * (10))*"\n")
-    Misc.benchmark 1 do
-    #Misc.profile do
+    Misc.benchmark 5 do
+      txt.rewind
+      #Misc.profile do
       data = TSV.parse_stream(txt, fix: true, type: :double, bar: true, merge: :concat)
       assert_equal num, data.size
       assert_equal 20, data['1'][0].length
@@ -170,4 +171,28 @@ k a|A b|B
 
     assert_equal [["k", [%w(b B)]]], values
   end
+
+
+  def test_parse_persist_serializer
+    content =<<-EOF
+Key ValueA ValueB
+k 1 2
+    EOF
+    content = StringIO.new content
+
+    TmpFile.with_file do |db|
+      data = ScoutCabinet.open db, true, "HDB"
+      TSV.parse content, sep: " ", header_hash: '', data: data, cast: :to_i, type: :list
+      assert_equal [1, 2], data["k"]
+    end
+
+    TmpFile.with_file do |db|
+      content.rewind
+      data = ScoutCabinet.open db, true, "HDB"
+      TSV.parse content, sep: " ", header_hash: '', data: data, cast: :to_i, type: :list, serializer: :float_array
+      assert_equal [1.0, 2.0], data["k"]
+    end
+
+  end
+
 end

data/test/scout/tsv/test_persist.rb

@@ -23,6 +23,13 @@ row2    a    a    id3
     assert_include tsv.keys, 'row1'
     assert_include tsv.keys, 'row2'
 
+    tsv = Persist.persist("TEST Persist TSV", :tsv) do 
+      TmpFile.with_file(content) do |filename|
+        TSV.open(filename)
+      end
+    end
+
+
     assert_nothing_raised do
       tsv = Persist.persist("TEST Persist TSV", :tsv) do 
         raise

data/test/scout/tsv/test_traverse.rb

@@ -1,9 +1,116 @@
 require File.expand_path(__FILE__).sub(%r(/test/.*), '/test/test_helper.rb')
 require File.expand_path(__FILE__).sub(%r(.*/test/), '').sub(/test_(.*)\.rb/,'\1')
 
-class TestClass < Test::Unit::TestCase
-  def test_true
-    assert true
+require 'scout/tsv'
+
+class TestTSVTraverse < Test::Unit::TestCase
+  def test_tsv_traverse_double
+    content =<<-'EOF'
+#: :sep=/\s+/#:type=:double
+#Id    ValueA    ValueB    OtherID
+row1    a|aa|aaa    b    Id1|Id2
+row2    A    B    Id3
+row2    AA    BB    Id33
+    EOF
+
+    sss 0
+    tsv = TmpFile.with_file(content) do |filename|
+      TSV.open(filename, :persist => true)
+    end
+
+    res = {}
+    tsv.traverse :key, %w(OtherID ValueB) do |k,v|
+      res[k] = v
+    end
+    assert_equal [%w(Id3 Id33), %w(B BB)], res["row2"]
+
+    res = {}
+    tsv.traverse :key, %w(OtherID ValueB), type: :list do |k,v|
+      res[k] = v
+    end
+    assert_equal ["Id3", "B"], res["row2"]
+
+    res = {}
+    tsv.traverse "OtherID", %w(Id ValueB), one2one: true do |k,v|
+      res[k] = v
+    end
+    assert_equal [[nil], %w(BB)], res["Id33"]
+
+    res = {}
+    tsv.traverse "OtherID", %w(Id ValueB), one2one: true, type: :list do |k,v|
+      res[k] = v
+    end
+    assert_equal ["row2", "B"], res["Id3"]
+    assert_equal [nil, "BB"], res["Id33"]
+
+    tsv.traverse "OtherID", %w(Id ValueB), one2one: false, type: :list do |k,v|
+      res[k] = v
+    end
+    assert_equal ["row2", "B"], res["Id3"]
+    assert_equal [nil, "BB"], res["Id33"]
+
+    res = {}
+    key_name, field_names = tsv.traverse "OtherID" do |k,v|
+      res[k] = v
+    end
+    assert_equal "OtherID", key_name
+    assert_equal %w(Id ValueA ValueB), field_names
+  end
+
+  def test_tsv_traverse_list
+    content =<<-'EOF'
+#: :sep=/\s+/#:type=:list
+#Id    ValueA    ValueB    OtherID
+row1    a    b    Id1
+row2    A    B    Id3
+    EOF
+
+    tsv = TmpFile.with_file(content) do |filename|
+      TSV.open(filename, :persist => true)
+    end
+
+    res = {}
+    tsv.traverse :key, %w(OtherID ValueB) do |k,v|
+      res[k] = v
+    end
+    assert_equal ["Id3", "B"], res["row2"]
+
+    res = {}
+    tsv.traverse :key, %w(OtherID ValueB), type: :double do |k,v|
+      res[k] = v
+    end
+    assert_equal [%w(Id3), %w(B)], res["row2"]
+
+    res = {}
+    tsv.traverse :key, %w(OtherID ValueB), type: :flat do |k,v|
+      res[k] = v
+    end
+    assert_equal %w(Id3 B), res["row2"]
+  end
+
+  def test_tsv_traverse_single
+    content =<<-'EOF'
+#: :sep=/\s+/#:type=:single
+#Id    ValueA
+row1    a
+row2    A
+    EOF
+
+    tsv = TmpFile.with_file(content) do |filename|
+      TSV.open(filename, :persist => true)
+    end
+
+    res = {}
+    tsv.traverse "ValueA", %w(Id) do |k,v|
+      res[k] = v
+    end
+    assert_equal "row1", res["a"]
+
+    res = {}
+    tsv.traverse "ValueA", %w(Id), type: :double do |k,v|
+      res[k] = v
+    end
+    assert_equal [["row1"]], res["a"]
   end
 end
 

data/test/scout/tsv/test_util.rb

@@ -0,0 +1,23 @@
+require File.expand_path(__FILE__).sub(%r(/test/.*), '/test/test_helper.rb')
+require File.expand_path(__FILE__).sub(%r(.*/test/), '').sub(/test_(.*)\.rb/,'\1')
+
+require 'scout/tsv'
+class TestTSVUtil < Test::Unit::TestCase
+  def test_open_persist
+    content =<<-'EOF'
+#: :sep=/\s+/#:type=:double#:merge=:concat
+#Id    ValueA    ValueB    OtherID
+row1    a|aa|aaa    b    Id1|Id2
+row2    A    B    Id3
+row2    a    a    id3
+    EOF
+
+    tsv = TmpFile.with_file(content) do |filename|
+      TSV.open(filename, :sep => " " )
+    end
+    assert_equal %w(row1 row2), tsv.collect{|k,v| k }
+    assert NamedArray === tsv.collect{|k,v| v }.first
+  end
+
+end
+

data/test/scout/tsv/util/test_filter.rb

@@ -0,0 +1,188 @@
+require File.expand_path(__FILE__).sub(%r(/test/.*), '/test/test_helper.rb')
+require File.expand_path(__FILE__).sub(%r(.*/test/), '').sub(/test_(.*)\.rb/,'\1')
+
+require 'scout/tsv'
+class TestTSVFilters < Test::Unit::TestCase
+  def test_collect
+    content1 =<<-EOF
+#: :sep=/\\s+/#:case_insensitive=false
+#Id    ValueA    ValueB
+row1    a|aa|aaa    b
+row2    A    B
+    EOF
+
+    TmpFile.with_file(content1) do |filename|
+      tsv = TSV.open filename
+      assert_equal 2, tsv.collect.size
+      tsv.filter
+      tsv.add_filter "field:ValueA", ["A"]
+      assert_equal 1, tsv.collect.size
+    end
+  end
+
+  def test_through
+    content1 =<<-EOF
+#: :sep=/\\s+/#:case_insensitive=false
+#Id    ValueA    ValueB
+row1    a|aa|aaa    b
+row2    A    B
+    EOF
+
+    TmpFile.with_file(content1) do |filename|
+      tsv = TSV.open filename
+      tsv.filter
+      tsv.add_filter "field:ValueA", ["A"]
+      elem = []
+      tsv.through do |k,v| elem << k end
+      assert_equal 1, elem.size
+    end
+  end
+
+  def test_2_filters
+    content1 =<<-EOF
+#: :sep=/\\s+/#:case_insensitive=false
+#Id    ValueA    ValueB
+row1    a|aa|aaa    b
+row2    A    B
+row3    A    C
+    EOF
+
+    TmpFile.with_file(content1) do |filename|
+      tsv = TSV.open filename
+      tsv.filter
+      tsv.add_filter "field:ValueA", ["A"]
+
+      elem = []
+      tsv.through do |k,v| elem << k end
+      assert_equal 2, elem.size
+      assert_equal ["row2", "row3"], elem.sort
+
+      tsv.add_filter "field:ValueB", ["C"]
+
+      elem = []
+      tsv.through do |k,v| elem << k end
+      assert_equal 1, elem.size
+    end
+  end
+
+  def test_filter_persistence
+    content1 =<<-EOF
+#: :sep=/\\s+/#:case_insensitive=false
+#Id    ValueA    ValueB
+row1    a|aa|aaa    b
+row2    A    B
+row3    A    C
+    EOF
+
+    TmpFile.with_file(content1) do |filename|
+      tsv = TSV.open filename
+      tsv.filter
+      tsv.add_filter "field:ValueA", ["A"], tmpdir.fieldValueA.find(:user)
+
+      elem = []
+      tsv.through do |k,v| elem << k end
+      assert_equal 2, elem.size
+
+      tsv.add_filter "field:ValueB", ["C"]
+
+      elem = []
+      tsv.through do |k,v| elem << k end
+      assert_equal 1, elem.size
+
+      tsv = TSV.open filename
+      tsv.filter
+      tsv.add_filter "field:ValueA", ["A"], tmpdir.fieldValueA.find(:user)
+
+      elem = []
+      tsv.through do |k,v| elem << k end
+      assert_equal 2, elem.size
+    end
+  end
+
+  def __test_filter_persistence_update
+    content1 =<<-EOF
+#: :sep=/\\s+/#:case_insensitive=false
+#Id    ValueA    ValueB
+row1    a|aa|aaa    b
+row2    A    B
+    EOF
+
+    TmpFile.with_file(content1) do |filename|
+      tsv = TSV.open filename
+      tsv.filter
+      tsv.add_filter "field:ValueA", ["A"], Scout.tmp.test.Filter.fieldValueA.find(:user)
+
+      elem = []
+      tsv.through do |k,v| elem << k end
+      assert_equal 1, elem.size
+
+      tsv["row3"] = [["A"], ["C"]]
+
+      elem = []
+      tsv.through do |k,v| elem << k end
+      assert_equal 2, elem.size
+
+      tsv.add_filter "field:ValueB", ["C"]
+
+      elem = []
+      tsv.through do |k,v| elem << k end
+      assert_equal 1, elem.size
+
+      tsv = TSV.open filename
+      tsv.filter
+      tsv.add_filter "field:ValueA", ["A"], Scout.tmp.test.Filter.fieldValueA.find(:user)
+
+      elem = []
+      tsv.through do |k,v| elem << k end
+      assert_equal 1, elem.size
+    end
+  end
+
+  def test_delete
+    content =<<-EOF
+#ID ValueA ValueB Comment
+row1 a b c
+row2 A B C
+    EOF
+
+    TmpFile.with_file(content) do |filename|
+      tsv = TSV.open(File.open(filename), type: :double, :sep => /\s/)
+      assert_equal 2, tsv.keys.length
+      tsv.delete "row2"
+      assert_equal 1, tsv.keys.length
+
+      tsv = TSV.open(File.open(filename), type: :double, :sep => /\s/)
+      tsv.filter
+      tsv.add_filter "field:ValueA", ["A"]
+
+      assert_equal 1, tsv.keys.length
+      assert_equal ["row2"], tsv.keys
+
+      tsv.delete "row2"
+      assert_equal 0, tsv.keys.length
+
+      tsv.pop_filter
+      assert_equal ["row1"], tsv.keys
+    end
+  end
+
+  def test_filename
+    content1 =<<-EOF
+#: :sep=/\\s+/#:case_insensitive=false
+#Id    ValueA    ValueB
+row1    a|aa|aaa    b
+row2    A    B
+    EOF
+
+    TmpFile.with_file(content1) do |filename|
+      tsv = TSV.open filename
+      tsv.filter
+      tsv.add_filter "field:ValueA", ["A"]
+      assert tsv.filename =~ /ValueA/
+      tsv.pop_filter
+      assert tsv.filename !~ /ValueA/
+    end
+ 
+  end
+end
+

data/test/scout/tsv/util/test_process.rb

@@ -0,0 +1,47 @@
+require File.expand_path(__FILE__).sub(%r(/test/.*), '/test/test_helper.rb')
+require File.expand_path(__FILE__).sub(%r(.*/test/), '').sub(/test_(.*)\.rb/,'\1')
+
+require 'scout/tsv'
+class TestTSVProcess < Test::Unit::TestCase
+  def test_process
+    content =<<-'EOF'
+#: :sep=/\s+/#:type=:double
+#Id    ValueA    ValueB    OtherID
+row1    a|aa|aaa    b    Id1|Id2
+row2    A    B    Id3
+row2    AA    BB    Id33
+    EOF
+
+    tsv = TmpFile.with_file(content) do |filename|
+      TSV.open(filename)
+    end
+
+    tsv.process "ValueA" do |v|
+      v.collect{|e| e.upcase }
+    end
+
+    assert_equal %w(A AA AAA), tsv["row1"][0]
+  end
+
+  def test_add_field
+    content =<<-'EOF'
+#: :sep=/\s+/#:type=:double
+#Id    ValueA    ValueB    OtherID
+row1    a|aa|aaa    b    Id1|Id2
+row2    A    B    Id3
+row2    AA    BB    Id33
+    EOF
+
+    tsv = TmpFile.with_file(content) do |filename|
+      TSV.open(filename)
+    end
+
+    tsv.add_field "ValueC" do |k,v|
+      v[0].collect{|e| e.gsub("a", "c").gsub("A", "C") }
+    end
+
+    assert_equal %w(c cc ccc), tsv["row1"]["ValueC"]
+    assert_equal %w(C CC), tsv["row2"]["ValueC"]
+  end
+end
+

data/test/scout/tsv/util/test_select.rb

@@ -0,0 +1,44 @@
+require File.expand_path(__FILE__).sub(%r(/test/.*), '/test/test_helper.rb')
+require File.expand_path(__FILE__).sub(%r(.*/test/), '').sub(/test_(.*)\.rb/,'\1')
+
+require 'scout/tsv'
+class TestTSVSelect < Test::Unit::TestCase
+  def test_select
+    content =<<-'EOF'
+#: :sep=/\s+/#:type=:double
+#Id    ValueA    ValueB    OtherID
+row1    a|aa|aaa    b    Id1|Id2
+row2    A    B    Id3
+row2    AA    BB    Id33
+    EOF
+
+    tsv = TmpFile.with_file(content) do |filename|
+      TSV.open(filename, :persist => true)
+    end
+
+    s = tsv.select do |k,v|
+      k.include? "2"
+    end
+
+    assert_equal ['row2'], s.keys
+  end
+
+  def test_reorder
+    content =<<-'EOF'
+#: :sep=/\s+/#:type=:double
+#Id    ValueA    ValueB    OtherID
+row1    a1|a2    b1|b2    Id1|Id2
+row2    A1|A3    B1|B3    Id1|Id3
+    EOF
+
+    tsv = TmpFile.with_file(content) do |filename|
+      TSV.open(filename)
+    end
+
+    r = tsv.reorder "OtherID", %w(ValueB Id)
+
+    assert_equal %w(row1 row2), r["Id1"]["Id"]
+    assert_equal %w(row2), r["Id3"]["Id"]
+  end
+end
+