diff --git a/lib/twitter_ebooks/model.rb b/lib/twitter_ebooks/model.rb index a40b046..092098b 100644 --- a/lib/twitter_ebooks/model.rb +++ b/lib/twitter_ebooks/model.rb @@ -96,7 +96,7 @@ module Ebooks end self end - + def initialize @tokens = [] @@ -113,7 +113,7 @@ module Ebooks return @tikis[token] else (@tokens.length+1)%1000 == 0 and puts "#{@tokens.length+1} tokens" - @tokens << token + @tokens << token return @tikis[token] = @tokens.length-1 end end diff --git a/spec/model_spec.rb b/spec/model_spec.rb index a8a85f7..2d2d039 100644 --- a/spec/model_spec.rb +++ b/spec/model_spec.rb @@ -36,7 +36,7 @@ describe Ebooks::Model do report2 = MemoryUsage.report do model = Ebooks::Model.load(file.path) end - expect(report2.total_memsize).to be < 3000000 + expect(report2.total_memsize).to be < 4000000 expect(model.tokens[0]).to be_a String expect(model.sentences[0][0]).to be_a Fixnum