rambling-trie 1.0.2 → 1.0.3

data/spec/lib/rambling/trie/enumerable_spec.rb

@@ -3,34 +3,37 @@ require 'spec_helper'
 module Rambling
   module Trie
     describe Enumerable do
-      let(:root) { Rambling::Trie::RawNode.new }
+      let(:node) { Rambling::Trie::Nodes::Raw.new }
       let(:words) { %w(add some words and another word) }
 
       before do
-        words.each { |word| root.add word.clone }
+        add_words node, words
       end
 
       describe '#each' do
         it 'returns an enumerator' do
-          expect(root.each).to be_a Enumerator
+          expect(node.each).to be_a Enumerator
         end
 
         it 'includes every word contained in the trie' do
-          root.each { |word| expect(words).to include word }
-          expect(root.count).to eq words.count
+          node.each do |word|
+            expect(words).to include word
+          end
+
+          expect(node.count).to eq words.count
         end
       end
 
       describe '#size' do
         it 'delegates to #count' do
-          expect(root.size).to eq words.size
+          expect(node.size).to eq words.size
         end
       end
 
       it 'includes the core Enumerable module' do
-        expect(root.all? { |word| words.include? word }).to be true
-        expect(root.any? { |word| word.start_with? 's' }).to be true
-        expect(root.to_a).to match_array words
+        expect(node.all? { |word| words.include? word }).to be true
+        expect(node.any? { |word| word.start_with? 's' }).to be true
+        expect(node.to_a).to match_array words
       end
     end
   end

data/spec/lib/rambling/trie/inspectable_spec.rb

@@ -1,30 +1,30 @@
 require 'spec_helper'
 
 describe Rambling::Trie::Inspectable do
-  let(:root) { Rambling::Trie::RawNode.new }
+  let(:node) { Rambling::Trie::Nodes::Raw.new }
 
   before do
-    %w(only three words).each { |word| root.add word }
+    add_words node, %w(only three words)
   end
 
   describe '#inspect' do
-    let(:node) { root[:o] }
-    let(:terminal_node) { root[:o][:n][:l][:y] }
+    let(:child) { node[:o] }
+    let(:terminal_node) { node[:o][:n][:l][:y] }
 
     it 'returns a pretty printed version of the node' do
-      expect(root.inspect).to eq "#<Rambling::Trie::RawNode letter: nil, terminal: nil, children: [:o, :t, :w]>"
-      expect(node.inspect).to eq "#<Rambling::Trie::RawNode letter: :o, terminal: nil, children: [:n]>"
-      expect(terminal_node.inspect).to eq "#<Rambling::Trie::RawNode letter: :y, terminal: true, children: []>"
+      expect(node.inspect).to eq "#<Rambling::Trie::Nodes::Raw letter: nil, terminal: nil, children: [:o, :t, :w]>"
+      expect(child.inspect).to eq "#<Rambling::Trie::Nodes::Raw letter: :o, terminal: nil, children: [:n]>"
+      expect(terminal_node.inspect).to eq "#<Rambling::Trie::Nodes::Raw letter: :y, terminal: true, children: []>"
     end
 
     context 'for a compressed node' do
       let(:compressor) { Rambling::Trie::Compressor.new }
-      let(:compressed_root) { compressor.compress root }
-      let(:compressed_node) { compressed_root[:only] }
+      let(:compressed_node) { compressor.compress node }
+      let(:compressed_child) { compressed_node[:only] }
 
       it 'returns a pretty printed version of the compressed node' do
-        expect(compressed_root.inspect).to eq "#<Rambling::Trie::CompressedNode letter: nil, terminal: nil, children: [:only, :three, :words]>"
-        expect(compressed_node.inspect).to eq "#<Rambling::Trie::CompressedNode letter: :only, terminal: true, children: []>"
+        expect(compressed_node.inspect).to eq "#<Rambling::Trie::Nodes::Compressed letter: nil, terminal: nil, children: [:only, :three, :words]>"
+        expect(compressed_child.inspect).to eq "#<Rambling::Trie::Nodes::Compressed letter: :only, terminal: true, children: []>"
       end
     end
   end

data/spec/lib/rambling/trie/nodes/compressed_spec.rb

@@ -0,0 +1,35 @@
+require 'spec_helper'
+
+describe Rambling::Trie::Nodes::Compressed do
+  let(:raw_node) { Rambling::Trie::Nodes::Raw.new }
+  let(:compressor) { Rambling::Trie::Compressor.new }
+  let(:node) { compressor.compress raw_node }
+
+  it_behaves_like 'a trie node implementation' do
+    def add_word_to_tree word
+      add_word raw_node, word
+    end
+
+    def add_words_to_tree words
+      add_words raw_node, words
+    end
+
+    def assign_letter letter
+      raw_node.letter = letter
+    end
+  end
+
+  describe '#compressed?' do
+    it 'returns true' do
+      expect(node).to be_compressed
+    end
+  end
+
+  describe '#add' do
+    it 'raises an error' do
+      expect do
+        add_word node, 'restaurant'
+      end.to raise_error Rambling::Trie::InvalidOperation
+    end
+  end
+end

data/spec/lib/rambling/trie/nodes/node_spec.rb

@@ -0,0 +1,7 @@
+require 'spec_helper'
+
+describe Rambling::Trie::Nodes::Node do
+  it_behaves_like 'a trie node' do
+    let(:node) { Rambling::Trie::Nodes::Node.new }
+  end
+end

data/spec/lib/rambling/trie/nodes/raw_spec.rb

@@ -0,0 +1,177 @@
+require 'spec_helper'
+
+describe Rambling::Trie::Nodes::Raw do
+  let(:node) { Rambling::Trie::Nodes::Raw.new }
+
+  it_behaves_like 'a trie node implementation' do
+    def add_word_to_tree word
+      add_word node, word
+    end
+
+    def add_words_to_tree words
+      add_words node, words
+    end
+
+    def assign_letter letter
+      node.letter = letter
+    end
+
+    it 'is not a word' do
+      expect(node).not_to be_word
+    end
+  end
+
+  describe '#compressed?' do
+    it 'returns false' do
+      expect(node).not_to be_compressed
+    end
+  end
+
+  describe '#add' do
+    context 'when the node has no branches' do
+      before do
+        add_word node, 'abc'
+      end
+
+      it 'adds only one child' do
+        expect(node.children.size).to eq 1
+      end
+
+      it 'adds the full subtree' do
+        expect(node[:a]).not_to be_nil
+        expect(node[:a][:b]).not_to be_nil
+        expect(node[:a][:b][:c]).not_to be_nil
+      end
+
+      it 'marks only the last child as terminal' do
+        expect(node).not_to be_terminal
+        expect(node[:a]).not_to be_terminal
+        expect(node[:a][:b]).not_to be_terminal
+        expect(node[:a][:b][:c]).to be_terminal
+      end
+    end
+
+    context 'when a word is added more than once' do
+      before do
+        add_word node, 'ack'
+        add_word node, 'ack'
+      end
+
+      it 'only counts it once' do
+        expect(node.children.size).to eq 1
+        expect(node[:a].children.size).to eq 1
+        expect(node[:a][:c].children.size).to eq 1
+        expect(node[:a][:c][:k].children.size).to eq 0
+      end
+
+      it 'does not change the terminal nodes in the tree' do
+        expect(node).not_to be_terminal
+        expect(node[:a]).not_to be_terminal
+        expect(node[:a][:c]).not_to be_terminal
+        expect(node[:a][:c][:k]).to be_terminal
+      end
+
+      it 'still returns the "added" node' do
+        child = add_word node, 'ack'
+        expect(child.letter).to eq :a
+      end
+    end
+
+    context 'when the word does not exist in the tree but the letters do' do
+      before do
+        add_words node, %w(ack a)
+      end
+
+      it 'does not add another branch' do
+        expect(node.children.size).to eq 1
+      end
+
+      it 'marks the corresponding node as terminal' do
+        expect(node[:a]).to be_terminal
+
+        expect(node).not_to be_terminal
+        expect(node[:a][:c]).not_to be_terminal
+        expect(node[:a][:c][:k]).to be_terminal
+      end
+
+      it 'returns the added node' do
+        child = add_word node, 'a'
+        expect(child.letter).to eq :a
+      end
+    end
+
+    context 'when the node has a letter and a parent' do
+      let(:parent) { Rambling::Trie::Nodes::Raw.new }
+      let(:node) { Rambling::Trie::Nodes::Raw.new :a, parent }
+
+      context 'adding an empty string' do
+        before do
+          add_word node, ''
+        end
+
+        it 'does not alter the node letter' do
+          expect(node.letter).to eq :a
+        end
+
+        it 'does not change the node children' do
+          expect(node.children.size).to eq 0
+        end
+
+        it 'changes the node to terminal' do
+          expect(node).to be_terminal
+        end
+      end
+
+      context 'adding a one letter word' do
+        before do
+          add_word node, 'b'
+        end
+
+        it 'does not alter the node letter' do
+          expect(node.letter).to eq :a
+        end
+
+        it 'adds a child with the expected letter' do
+          expect(node.children.size).to eq 1
+          expect(node.children.first.letter).to eq :b
+        end
+
+        it 'reports it has the expected letter a key' do
+          expect(node).to have_key(:b)
+        end
+
+        it 'returns the child corresponding to the key' do
+          expect(node[:b]).to eq node.children_tree[:b]
+        end
+
+        it 'does not mark itself as terminal' do
+          expect(node).not_to be_terminal
+        end
+
+        it 'marks the first child as terminal' do
+          expect(node[:b]).to be_terminal
+        end
+      end
+
+      context 'adding a large word' do
+        before do
+          add_word node, 'mplified'
+        end
+
+        it 'marks the last letter as terminal' do
+          expect(node[:m][:p][:l][:i][:f][:i][:e][:d]).to be_terminal
+        end
+
+        it 'does not mark any other letter as terminal' do
+          expect(node[:m][:p][:l][:i][:f][:i][:e]).not_to be_terminal
+          expect(node[:m][:p][:l][:i][:f][:i]).not_to be_terminal
+          expect(node[:m][:p][:l][:i][:f]).not_to be_terminal
+          expect(node[:m][:p][:l][:i]).not_to be_terminal
+          expect(node[:m][:p][:l]).not_to be_terminal
+          expect(node[:m][:p]).not_to be_terminal
+          expect(node[:m]).not_to be_terminal
+        end
+      end
+    end
+  end
+end

data/spec/lib/rambling/trie/serializers/file_spec.rb

@@ -1,11 +1,11 @@
 require 'spec_helper'
 
 describe Rambling::Trie::Serializers::File do
-  let(:serializer) { Rambling::Trie::Serializers::File.new }
-
   it_behaves_like 'a serializer' do
-    let(:filepath) { File.join ::SPEC_ROOT, 'tmp', 'trie-root.file' }
-    let(:content) { 'a few words to validate that load and dump are working' }
+    let(:serializer) { Rambling::Trie::Serializers::File.new }
+    let(:format) { :file }
+
+    let(:content) { trie.to_a.join ' ' }
     let(:formatted_content) { content }
   end
 end

data/spec/lib/rambling/trie/serializers/marshal_spec.rb

@@ -1,14 +1,10 @@
 require 'spec_helper'
 
 describe Rambling::Trie::Serializers::Marshal do
-  let(:serializer) { Rambling::Trie::Serializers::Marshal.new }
-
-  let(:words) { %w(a few words to validate that load and dump are working) }
-  let(:trie) { Rambling::Trie.create { |t| words.each { |w| t << w } } }
-
   it_behaves_like 'a serializer' do
-    let(:filepath) { File.join ::SPEC_ROOT, 'tmp', 'trie-root.marshal' }
-    let(:content) { trie.root }
+    let(:serializer) { Rambling::Trie::Serializers::Marshal.new }
+    let(:format) { :marshal }
+
     let(:formatted_content) { Marshal.dump content }
   end
 end

data/spec/lib/rambling/trie/serializers/yaml_spec.rb

@@ -1,14 +1,10 @@
 require 'spec_helper'
 
 describe Rambling::Trie::Serializers::Yaml do
-  let(:serializer) { Rambling::Trie::Serializers::Yaml.new }
-
-  let(:words) { %w(a few words to validate that load and dump are working) }
-  let(:trie) { Rambling::Trie.create { |t| words.each { |w| t << w } } }
-
   it_behaves_like 'a serializer' do
-    let(:filepath) { File.join ::SPEC_ROOT, 'tmp', 'trie-root.yml' }
-    let(:content) { trie.root }
+    let(:serializer) { Rambling::Trie::Serializers::Yaml.new }
+    let(:format) { :yml }
+
     let(:formatted_content) { YAML.dump content }
   end
 end

data/spec/lib/rambling/trie/serializers/zip_spec.rb

@@ -1,30 +1,26 @@
 require 'spec_helper'
 
 describe Rambling::Trie::Serializers::Zip do
-  let(:properties) { Rambling::Trie::Configuration::Properties.new }
-  let(:serializer) { Rambling::Trie::Serializers::Zip.new properties }
-
-  let(:words) { %w(a few words to validate that load and dump are working) }
-  let(:trie) { Rambling::Trie.create { |t| words.each { |w| t << w } } }
-  let(:tmp_path) { File.join ::SPEC_ROOT, 'tmp' }
+  it_behaves_like 'a serializer' do
+    let(:properties) { Rambling::Trie::Configuration::Properties.new }
+    let(:serializer) { Rambling::Trie::Serializers::Zip.new properties }
+    let(:format) { 'marshal.zip' }
 
-  before do
-    properties.tmp_path = tmp_path
-  end
+    before do
+      properties.tmp_path = tmp_path
+    end
 
-  it_behaves_like 'a serializer' do
-    let(:filename) { 'trie-root.marshal' }
-    let(:filepath) { File.join tmp_path, "#{filename}.zip" }
-    let(:content) { trie.root }
+    let(:filename) { File.basename(filepath).gsub /\.zip/, ''}
     let(:formatted_content) { zip Marshal.dump content }
-  end
 
-  def zip content
-    io = Zip::OutputStream.write_buffer do |io|
-      io.put_next_entry filename
-      io.write content
+    def zip content
+      io = Zip::OutputStream.write_buffer do |io|
+        io.put_next_entry filename
+        io.write content
+      end
+
+      io.rewind
+      io.read
     end
-    io.rewind
-    io.read
   end
 end

data/spec/lib/rambling/trie/stringifyable_spec.rb

@@ -2,11 +2,11 @@ require 'spec_helper'
 
 describe Rambling::Trie::Stringifyable do
   describe '#as_word' do
-    let(:node) { Rambling::Trie::RawNode.new }
+    let(:node) { Rambling::Trie::Nodes::Raw.new }
 
     context 'for an empty node' do
       before do
-        node.add ''
+        add_word node, ''
       end
 
       it 'returns nil' do
@@ -17,7 +17,7 @@ describe Rambling::Trie::Stringifyable do
     context 'for one letter' do
       before do
         node.letter = :a
-        node.add ''
+        add_word node, ''
       end
 
       it 'returns the expected one letter word' do
@@ -28,7 +28,7 @@ describe Rambling::Trie::Stringifyable do
     context 'for a small word' do
       before do
         node.letter = :a
-        node.add 'll'
+        add_word node, 'll'
       end
 
       it 'returns the expected small word' do
@@ -43,7 +43,7 @@ describe Rambling::Trie::Stringifyable do
     context 'for a long word' do
       before do
         node.letter = :b
-        node.add 'eautiful'
+        add_word node, 'eautiful'
       end
 
       it 'returns the expected long word' do
@@ -52,7 +52,7 @@ describe Rambling::Trie::Stringifyable do
     end
 
     context 'for a node with nil letter' do
-      let(:node) { Rambling::Trie::RawNode.new nil }
+      let(:node) { Rambling::Trie::Nodes::Raw.new nil }
 
       it 'returns nil' do
         expect(node.as_word).to be_empty
@@ -65,8 +65,7 @@ describe Rambling::Trie::Stringifyable do
 
       before do
         node.letter = :a
-        node.add 'm'
-        node.add 'dd'
+        add_words node, %w(m dd)
       end
 
       it 'returns the words for the terminal nodes' do