triez 0.2

data/test/triez_test.rb

@@ -0,0 +1,116 @@
+require "test/unit"
+require_relative "../lib/triez"
+
+GC.stress
+
+class TriezTest < Test::Unit::TestCase
+  def test_init_options
+    t = Triez.new obj_value: true
+    assert_equal true, t.obj_value?
+    assert_equal false, t.suffix?
+    t = Triez.new suffix: true
+    assert_equal true, t.suffix?
+    assert_equal false, t.obj_value?
+  end
+
+  def test_hat_trie
+    t = Triez.new obj_value: true
+
+    v1 = (1 << 40)
+    v2 = (1 << 141)
+    t['万塘路一锅鸡'] = v1
+    t['万塘路'] = v2
+    assert_equal v1, t['万塘路一锅鸡']
+    assert_equal v2, t['万塘路']
+    assert_equal nil, t['万']
+    assert_equal false, t.has_key?('万')
+    assert_equal true, t.has_key?('万塘路')
+
+    assert_equal v1, t.delete('万塘路一锅鸡')
+    assert_equal nil, t['万塘路一锅鸡']
+    assert_equal v2, t['万塘路']
+
+    a = t.search_with_prefix ''
+    assert_equal [['万塘路', v2]], a
+
+    t['马当路'] = 3
+    a = t.search_with_prefix '万塘'
+    assert_equal [['路', v2]], a
+  end
+
+  def test_insertion_and_search_on_many_keys
+    t = Triez.new
+    as = ('A'..'z').to_a
+    bs = ('一'..'百').to_a
+    as.each do |a|
+      # 10k chars to ensure burst
+      bs.each do |b|
+        t[a + b] = 0
+      end
+    end
+    assert_equal as.size * bs.size, t.size
+
+    a = t.search_with_prefix 'a'
+    assert_equal bs.to_a, a.map(&:first).sort
+
+    a = []
+    t.search_with_prefix 'b', sort: true, limit: 3 do |k, v|
+      a << k
+    end
+    assert_equal 3, a.size
+    assert_equal a, a.sort
+  end
+
+  def test_each_and_raise
+    t = Triez.new
+    t['abcd'] = 0
+    t['abc'] = 1
+
+    assert_raise NameError do
+      t.each do |k, v|
+        raise NameError, k
+      end
+    end
+
+    assert_raise ArgumentError do
+      t.each
+    end
+  end
+
+  def test_append
+    t = Triez.new
+    ('a'..'z').each do |c|
+      t << c
+    end
+    assert_equal 26, t.size
+    assert_equal 0, t['c']
+    assert_equal true, t.has_key?('c')
+  end
+
+  def test_suffix_insert
+    t = Triez.new suffix: true
+    t << '12345'
+    assert_equal 5, t.size
+  end
+
+  def test_full_text_search
+    sequences = {
+      'ACTGAAAAAAACTG' => 1,
+      'ATACGGTCCA' => 2,
+      'GCTTGTACGT' => 3
+    }
+    t = Triez.new suffix: true
+    sequences.each do |seq, id|
+      t[seq] = id
+    end
+    assert_equal 2, t.search_with_prefix('CGGT').map(&:last).flatten.first
+  end
+
+  def test_nul_char_in_keys
+    t = Triez.new
+    t["a\0b"] = 1
+    assert_equal 1, t["a\0b"]
+    assert_equal 1, t.size
+    assert_equal nil, t["a"]
+  end
+end

metadata

@@ -0,0 +1,63 @@
+--- !ruby/object:Gem::Specification
+name: triez
+version: !ruby/object:Gem::Version
+  version: '0.2'
+  prerelease: 
+platform: ruby
+authors:
+- Zete Lui
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2013-02-04 00:00:00.000000000 Z
+dependencies: []
+description: fast, efficient, unicode aware HAT trie with prefix / suffix support.
+email: 
+executables: []
+extensions:
+- ext/extconf.rb
+extra_rdoc_files: []
+files:
+- copying
+- readme.md
+- lib/triez.rb
+- test/triez_test.rb
+- ext/hat-stub.c
+- ext/triez.cc
+- ext/common.h
+- ext/extconf.rb
+- ext/hat-trie/ahtable.c
+- ext/hat-trie/ahtable.h
+- ext/hat-trie/COPYING
+- ext/hat-trie/hat-trie.c
+- ext/hat-trie/hat-trie.h
+- ext/hat-trie/misc.h
+- ext/hat-trie/murmurhash3.c
+- ext/hat-trie/murmurhash3.h
+- ext/hat-trie/pstdint.h
+homepage: https://github.com/luikore/triez
+licenses: []
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: 1.9.2
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: fast, efficient, unicode aware HAT trie with prefix / suffix support
+test_files: []
+has_rdoc: false