merges tag levels

Author: gregors

lib/boilerpipe/document/text_block.rb

@@ -68,9 +68,9 @@ def merge_next(other)
         @offset_blocks_end = [@offset_blocks_end, other.offset_blocks_end].max
         @content |= other.is_content?
         @labels.merge(other.labels.clone)
-        init_densities
-
         @tag_level = [@tag_level, other.tag_level].min
+
+        init_densities
       end
 
       def to_s

spec/document/text_block_spec.rb

@@ -133,6 +133,18 @@ module Boilerpipe
 
         expect(block.labels).to eq Set.new(['boom', 'pow'])
       end
+
+      it 'sets the tag level to the minimum of the two blocks' do
+        block = Document::TextBlock.new('one')
+        block.set_tag_level(2)
+
+        another_block = Document::TextBlock.new('two')
+        another_block.set_tag_level(1)
+
+        block.merge_next(another_block)
+
+        expect(block.tag_level).to eq 1
+      end
     end
 
     describe '#add_label' do