love_is 0.1.1 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: ff5f2bef155fe366a7f657a5a7b1cb89ebf0e438
-  data.tar.gz: ea4c2d7221a39b3f53fea61fda5526cafbddb9ab
+  metadata.gz: a7a46999f42dfa66c85e97e254a3a7d408abe6e6
+  data.tar.gz: 3c2d201291075b8c268c5241986a7886fee71cc3
 SHA512:
-  metadata.gz: 3d7db8904d9a22b8674608416e00529b79607ae094cb495522b56cfc2794bebd0b9e21ef5eafd2c3757d1e29798dba22f57db308d9f9a2c262a395c44764eab2
-  data.tar.gz: 9b2289f3c420fccf86564e42a333a86ac6c310f457cf27f5808a6f9a52804dfca33b2ae4d57a7f45dc5e04344c33c3baa21b12c5e88fb89bf62d1a667a9048ba
+  metadata.gz: e5d4d94674c3664d497fe471bc057c79d1b81848c5ba0d2d3d31f0aeee701afebca6e9642eae1b6760785a3249ccd337b9b8d822315cd6bd50c426a83f48677f
+  data.tar.gz: fd30f6cba37662faa7e0e7375dc43c29760062cdc20d59070fa5db07a6a0f597026914bb831839a2a4aea8dab871bde747f6fc25e5b140f952783e63f117b6ab

data/lib/love_is.rb

@@ -1,5 +1,7 @@
 #!/usr/bin/env ruby
 
+require 'yaml'
+
 module LoveIs
 end
 

data/lib/love_is/generator.rb

@@ -19,12 +19,16 @@ class LoveIs::Generator
   end
 
   def initial_values
-    ngram.times.inject(tree.child('*')) do |node|
+    ngram.times.inject(first_word) do |node|
       @chain << node.value
       node.next
     end
   end
 
+  def first_word
+    tree.child(SENTENCE_BEGIN)
+  end
+
   def next_value
     node = tree[*chain[(-ngram + 1)..-1]]
     node ? node.next.value : nil

data/lib/love_is/parser.rb

@@ -17,4 +17,12 @@ module LoveIs::Parser
   def sentence2array(sentence)
     sentence.split(/\s/).unshift(SENTENCE_BEGIN)
   end
+
+  def data_to_yaml(filename)
+    File.open(File.expand_path(filename), 'w') { |f| f.puts tree.to_yaml }
+  end
+
+  def data_from_yaml(filename)
+    @tree = YAML.load(File.read(File.expand_path(filename)))
+  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: love_is
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.3
 platform: ruby
 authors:
 - Sergey Smagin