tyler-trie 0.2.2 → 0.2.3

data/VERSION.yml

@@ -1,4 +1,4 @@
 --- 
-patch: 2
+patch: 3
 major: 0
 minor: 2

data/ext/trie/trie.c

@@ -7,6 +7,7 @@
 VALUE cTrie, cTrieNode;
 
 static TrieChar* stringToTrieChar(VALUE string) {
+    StringValue(string);
     return (TrieChar*) RSTRING(string)->ptr;
 }
 
@@ -148,6 +149,9 @@ static VALUE walk_all_paths(VALUE children, SBTrieState *state, char *prefix) {
 }
 
 static VALUE trie_children(VALUE self, VALUE prefix) {
+    if(NIL_P(prefix))
+	return rb_ary_new();
+
     SBTrie *sb_trie;
     Data_Get_Struct(self, SBTrie, sb_trie);
 
@@ -204,6 +208,9 @@ static VALUE walk_all_paths_with_values(VALUE children, SBTrieState *state, char
 }
 
 static VALUE trie_children_with_values(VALUE self, VALUE prefix) {
+    if(NIL_P(prefix))
+	return rb_ary_new();
+
     SBTrie *sb_trie;
     Data_Get_Struct(self, SBTrie, sb_trie);
 

data/spec/trie_spec.rb

@@ -75,7 +75,7 @@ describe Trie do
       children.should include('rocket')
     end
 
-    it 'returns nil if prefix does not exist' do
+    it 'returns blank array if prefix does not exist' do
       @trie.children('ajsodij').should == []
     end
 
@@ -85,6 +85,10 @@ describe Trie do
       children.should include('rock')
       children.should include('rocket')
     end
+
+    it 'returns blank array if prefix is nil' do
+      @trie.children(nil).should == []
+    end
   end
 
   describe :children_with_values do
@@ -110,6 +114,10 @@ describe Trie do
       children.should include(['abc',2])
       children.should include(['abcd',4])
     end
+
+    it 'returns blank array if prefix is nil' do
+      @trie.children_with_values(nil).should == []
+    end
   end
 
   describe :walk_to_terminal do

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: tyler-trie
 version: !ruby/object:Gem::Version 
-  version: 0.2.2
+  version: 0.2.3
 platform: ruby
 authors: 
 - Tyler McMullen