remarkovable 0.1

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: a17b52251e917de43adfcb8a3265137ebea1f089
+  data.tar.gz: 04a8d08821c42e82e3cef0caf71cc841e1161cc4
+SHA512:
+  metadata.gz: 802a5d3df97044b11e293caa463313d14c5eb9c87b63269de627b385f27281026704da3e95c865aba6ac8d091cf0f236a72d44f9a8093dbda53b28b87a00cbef
+  data.tar.gz: 97f87565cd104939392b8fcc9899b419e2852818ae392888a5a0a0df2d4e31b7292d2eddf87793a8d1991b61a53333870e718c8bd132db45e2de7fec6820626f

data/lib/remarkovable.rb

@@ -0,0 +1,52 @@
+class Remarkovable
+  attr_accessor :markov_model
+
+  def initialize(string: string, prefix_length: 2)
+    return if string.nil?
+    @markov_model = Hash.new { |hash, key| hash[key] = [] }
+
+    words = string.split(/([.!?])|\s+/)
+    words.each_with_index do |word, i|
+      key = [word]
+
+      (prefix_length - 1).times do |n|
+        next_word = i + n + 1
+        key << words[next_word]
+      end
+
+      key = key.join(' ')
+      match = words[i + prefix_length]
+      add_triad(key: key, match: match) if i < words.size - prefix_length
+    end
+  end
+
+  def speak(custom_key: nil)
+    return if @markov_model.nil?
+    key = if !custom_key.nil? && @markov_model[custom_key].any?
+      custom_key
+    else
+      @markov_model.keys.sample
+    end
+    output = Array(key.capitalize.split(' '))
+
+    until (output & %w(. ! ?)).any?
+      match = @markov_model[key].sample
+      if match.nil?
+        key = @markov_model.keys.sample
+        next
+      end
+      output << match
+      output.flatten!
+      key = [output[-2], output[-1]].join(' ')
+    end
+
+    output.join(' ').gsub(/\s+([,.!?])/, '\1')
+  end
+
+  private
+
+  def add_triad(key: key, match: match)
+    @markov_model[key] = {} unless @markov_model[key]
+    @markov_model[key] += [match]
+  end
+end

data/test/test_remarkovable.rb

@@ -0,0 +1,139 @@
+require 'minitest/autorun'
+require './lib/remarkovable.rb'
+
+class RemarkovableTest < Minitest::Test
+  def test_add_pair_simple_case
+    string = 'we are walking'
+    mc = Remarkovable.new(string: string)
+    expected_output = {
+      'we are' => ['walking']
+    }
+
+    assert_equal expected_output, mc.markov_model
+  end
+
+  def test_add_pair_nil
+    string = nil
+    mc = Remarkovable.new(string: string)
+    expected_output = nil
+
+    assert_equal nil, mc.speak
+  end
+
+  def test_add_pair_three_words_and_punctuation
+    string = 'we (are) walking.'
+    mc = Remarkovable.new(string: string)
+    expected_output = {
+      'we (are)' => ['walking'],
+      '(are) walking' => ['.']
+    }
+
+    assert_equal expected_output, mc.markov_model
+  end
+
+  def test_add_pair_three_words_and_spaces
+    string = 'we are  walking.'
+    mc = Remarkovable.new(string: string)
+    expected_output = {
+      'we are' => ['walking'],
+      'are walking' => ['.']
+    }
+
+    assert_equal expected_output, mc.markov_model
+  end
+
+  def test_add_pair_three_words_and_quotes
+    string = 'we are "walking".'
+    mc = Remarkovable.new(string: string)
+    expected_output = {
+      'we are' => ['"walking"'],
+      'are "walking"' => ['.']
+    }
+
+    assert_equal expected_output, mc.markov_model
+  end
+
+  def test_add_pair_key_duplication
+    string = 'we are walking we are talking we are walking.'
+    mc = Remarkovable.new(string: string)
+    expected_output = {
+      'we are' => %w(walking talking walking),
+      'are walking' => ['we', '.'],
+      'walking we' => ['are'],
+      'are talking' => ['we'],
+      'talking we' => ['are']
+    }
+
+    assert_equal expected_output, mc.markov_model
+  end
+
+  def test_add_pair_with_newlines
+    string = "we are walking\n\rwe are talking we are walking."
+    mc = Remarkovable.new(string: string)
+    expected_output = {
+      'we are' => %w(walking talking walking),
+      'are walking' => ['we', '.'],
+      'walking we' => ['are'],
+      'are talking' => ['we'],
+      'talking we' => ['are']
+    }
+
+    assert_equal expected_output, mc.markov_model
+  end
+
+  def test_custom_prefix_length_1
+    string = 'we are.'
+    mc = Remarkovable.new(string: string, prefix_length: 1)
+    expected_output = {
+      'we' => ['are'],
+      'are' => ['.']
+    }
+
+    assert_equal expected_output, mc.markov_model
+  end
+
+  def test_custom_prefix_length_3
+    string = 'we are walking.'
+    mc = Remarkovable.new(string: string, prefix_length: 3)
+    expected_output = {
+      'we are walking' => ['.']
+    }
+
+    assert_equal expected_output, mc.markov_model
+  end
+
+  def test_custom_prefix_length_4
+    string = 'we are walking talking.'
+    mc = Remarkovable.new(string: string, prefix_length: 4)
+    expected_output = {
+      'we are walking talking' => ['.']
+    }
+
+    assert_equal expected_output, mc.markov_model
+  end
+
+  def test_custom_prefix_length_5
+    string = 'we are walking talking singing.'
+    mc = Remarkovable.new(string: string, prefix_length: 5)
+    expected_output = {
+      'we are walking talking singing' => ['.']
+    }
+
+    assert_equal expected_output, mc.markov_model
+  end
+
+  def test_custom_key
+    string = "we are walking\n\rwe are talking we are walking."
+    mc = Remarkovable.new(string: string)
+    expected_output = /^We are./
+
+    assert_match expected_output, mc.speak(custom_key: 'we are')
+  end
+
+  def test_custom_key_no_match
+    string = "we are walking\n\rwe are talking we are walking."
+    mc = Remarkovable.new(string: string)
+
+    assert_nil mc.speak(custom_key: 'foo bar') =~ /Foo bar/i
+  end
+end

metadata

@@ -0,0 +1,47 @@
+--- !ruby/object:Gem::Specification
+name: remarkovable
+version: !ruby/object:Gem::Version
+  version: '0.1'
+platform: ruby
+authors:
+- Jake Worth
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2015-10-11 00:00:00.000000000 Z
+dependencies: []
+description: A gem that produces Markov chain output from any text.
+email:
+- jake@jakeworth.com
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- lib/remarkovable.rb
+- test/test_remarkovable.rb
+homepage: http://github.com/jwworth/remarkovable
+licenses:
+- MIT
+metadata: {}
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 2.4.5
+signing_key: 
+specification_version: 4
+summary: Markov chains for all.
+test_files:
+- test/test_remarkovable.rb