poefy-pg 1.1.0 → 2.0.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +5 -5
- data/.rspec +1 -1
- data/LICENSE +13 -13
- data/README.md +7 -7
- data/Rakefile +8 -8
- data/lib/poefy/pg/version.rb +24 -24
- data/lib/poefy/pg.rb +294 -294
- data/poefy-pg.gemspec +3 -1
- data/spec/poefy_pg_spec.rb +697 -697
- data/spec/spec_helper.rb +11 -11
- metadata +5 -9
data/spec/poefy_pg_spec.rb
CHANGED
|
@@ -1,697 +1,697 @@
|
|
|
1
|
-
#!/usr/bin/env ruby
|
|
2
|
-
# Encoding: UTF-8
|
|
3
|
-
|
|
4
|
-
################################################################################
|
|
5
|
-
|
|
6
|
-
describe Poefy::Poem, "-- Postgres" do
|
|
7
|
-
|
|
8
|
-
before(:all) do
|
|
9
|
-
require 'poefy/pg'
|
|
10
|
-
@root = Poefy.root
|
|
11
|
-
dbs = %w{spec_test_tiny spec_shakespeare spec_whitman}
|
|
12
|
-
dbs.each do |db_name|
|
|
13
|
-
Poefy::Database.single_exec! <<-SQL
|
|
14
|
-
SET client_min_messages TO WARNING;
|
|
15
|
-
DROP TABLE IF EXISTS #{db_name};
|
|
16
|
-
SQL
|
|
17
|
-
end
|
|
18
|
-
end
|
|
19
|
-
|
|
20
|
-
after(:all) do
|
|
21
|
-
dbs = %w{spec_test_tiny spec_shakespeare spec_whitman}
|
|
22
|
-
dbs.each do |db_name|
|
|
23
|
-
Poefy::Database.single_exec! <<-SQL
|
|
24
|
-
SET client_min_messages TO WARNING;
|
|
25
|
-
DROP TABLE IF EXISTS #{db_name};
|
|
26
|
-
SQL
|
|
27
|
-
end
|
|
28
|
-
end
|
|
29
|
-
|
|
30
|
-
##############################################################################
|
|
31
|
-
|
|
32
|
-
describe "using tiny dataset 'spec_test_tiny'" do
|
|
33
|
-
corpus = "spec_test_tiny"
|
|
34
|
-
|
|
35
|
-
# Create a small corpus of a few rhymes.
|
|
36
|
-
text_array = %w{man plan flan can dan fish dish wish bee sea tree flea}
|
|
37
|
-
text_array.map!{ |i| 'a ' + i }
|
|
38
|
-
text_string = text_array.join("\n")
|
|
39
|
-
row_count = text_array.count
|
|
40
|
-
|
|
41
|
-
before(:each) do
|
|
42
|
-
@poefy = Poefy::Poem.new(corpus, { proper: false })
|
|
43
|
-
end
|
|
44
|
-
after(:each) do
|
|
45
|
-
@poefy.close
|
|
46
|
-
end
|
|
47
|
-
it "initialised object not nil" do
|
|
48
|
-
expect(@poefy).to_not be_nil
|
|
49
|
-
end
|
|
50
|
-
|
|
51
|
-
# Create corpora in the three different ways.
|
|
52
|
-
describe "@poefy#make_database!" do
|
|
53
|
-
|
|
54
|
-
it "Use array of strings" do
|
|
55
|
-
@poefy.make_database! text_array
|
|
56
|
-
expect(@poefy.corpus.
|
|
57
|
-
expect(@poefy.corpus.count).to be row_count
|
|
58
|
-
poem = @poefy.poem({ rhyme: 'aabb' })
|
|
59
|
-
expect(poem.count).to be 4
|
|
60
|
-
end
|
|
61
|
-
|
|
62
|
-
it "Use one long newline delimited string" do
|
|
63
|
-
@poefy.make_database! text_string
|
|
64
|
-
expect(@poefy.corpus.
|
|
65
|
-
expect(@poefy.corpus.count).to be row_count
|
|
66
|
-
poem = @poefy.poem({ rhyme: 'aabb' })
|
|
67
|
-
expect(poem.count).to be 4
|
|
68
|
-
end
|
|
69
|
-
|
|
70
|
-
it "Use text lines from a file" do
|
|
71
|
-
|
|
72
|
-
# Create a temp file.
|
|
73
|
-
tmp = Tempfile.new('spec-', Poefy.root + '/spec')
|
|
74
|
-
text_path = tmp.path
|
|
75
|
-
tmp.write text_string
|
|
76
|
-
tmp.close
|
|
77
|
-
|
|
78
|
-
@poefy.make_database! text_path
|
|
79
|
-
expect(@poefy.corpus.
|
|
80
|
-
expect(@poefy.corpus.count).to be row_count
|
|
81
|
-
poem = @poefy.poem({ rhyme: 'aabb' })
|
|
82
|
-
expect(poem.count).to be 4
|
|
83
|
-
|
|
84
|
-
# Delete the temp file.
|
|
85
|
-
tmp.delete
|
|
86
|
-
end
|
|
87
|
-
end
|
|
88
|
-
|
|
89
|
-
# Make sure that the description can be updated as specified
|
|
90
|
-
# and that it doesn't cause SQL injection.
|
|
91
|
-
describe "corpus description using #desc=" do
|
|
92
|
-
it "@poefy.corpus.desc is initially empty" do
|
|
93
|
-
expect(@poefy.corpus.desc).to eq ''
|
|
94
|
-
end
|
|
95
|
-
|
|
96
|
-
values = [
|
|
97
|
-
"test",
|
|
98
|
-
" -- test",
|
|
99
|
-
"; -- test",
|
|
100
|
-
"test' -- ",
|
|
101
|
-
"test'' -- ",
|
|
102
|
-
"'test' -- ",
|
|
103
|
-
"'test'' -- ",
|
|
104
|
-
"Shakespeare's sonnets",
|
|
105
|
-
"Shakespeare's -- sonnets",
|
|
106
|
-
"Shakespeare's; -- sonnets",
|
|
107
|
-
"test' ; INSERT INTO spec_test_tiny VALUES('foo') -- ",
|
|
108
|
-
"105 OR 1=1",
|
|
109
|
-
"' or ''='"
|
|
110
|
-
]
|
|
111
|
-
values.each do |value|
|
|
112
|
-
it "@poefy.corpus.desc = #{value}" do
|
|
113
|
-
@poefy.corpus.desc = value
|
|
114
|
-
expect(@poefy.corpus.desc).to eq value
|
|
115
|
-
expect(@poefy.corpus.count).to be row_count
|
|
116
|
-
end
|
|
117
|
-
end
|
|
118
|
-
end
|
|
119
|
-
|
|
120
|
-
describe ":rhyme option" do
|
|
121
|
-
|
|
122
|
-
describe "should raise Poefy::MissingFormOrRhyme" do
|
|
123
|
-
it "blank, no argument" do
|
|
124
|
-
expect {
|
|
125
|
-
@poefy.poem
|
|
126
|
-
}.to raise_error(Poefy::MissingFormOrRhyme)
|
|
127
|
-
end
|
|
128
|
-
it "({ })" do
|
|
129
|
-
expect {
|
|
130
|
-
@poefy.poem ({ })
|
|
131
|
-
}.to raise_error(Poefy::MissingFormOrRhyme)
|
|
132
|
-
end
|
|
133
|
-
it "({ rhyme: nil })" do
|
|
134
|
-
expect {
|
|
135
|
-
@poefy.poem ({ rhyme: nil })
|
|
136
|
-
}.to raise_error(Poefy::MissingFormOrRhyme)
|
|
137
|
-
end
|
|
138
|
-
it "({ rhyme: ' ' })" do
|
|
139
|
-
expect {
|
|
140
|
-
@poefy.poem ({ rhyme: ' ' })
|
|
141
|
-
}.to raise_error(Poefy::MissingFormOrRhyme)
|
|
142
|
-
end
|
|
143
|
-
it "({ rhyme: '' })" do
|
|
144
|
-
expect {
|
|
145
|
-
@poefy.poem ({ rhyme: '' })
|
|
146
|
-
}.to raise_error(Poefy::MissingFormOrRhyme)
|
|
147
|
-
end
|
|
148
|
-
end
|
|
149
|
-
|
|
150
|
-
describe "should return correct number of lines" do
|
|
151
|
-
rhymes = %w{a b z A aa ab zz AA AB AA1 A1 B1 Z1 AB1 A1A1A1A1B1B1B1B1B1}
|
|
152
|
-
rhymes += ['A1A1A1 A1A1A1 B1B1B1B1B1B1','a b c a b c']
|
|
153
|
-
rhymes += [' abc','abc ',' abc ']
|
|
154
|
-
rhymes += ['n aaa n','n aXXXa N1']
|
|
155
|
-
rhymes.each do |i|
|
|
156
|
-
it "({ rhyme: '#{i}' })" do
|
|
157
|
-
poem = @poefy.poem ({ rhyme: i })
|
|
158
|
-
expect(poem.count).to be i.gsub(/[0-9]/,'').length
|
|
159
|
-
end
|
|
160
|
-
end
|
|
161
|
-
end
|
|
162
|
-
|
|
163
|
-
describe "should accept characters other than number" do
|
|
164
|
-
rhymes = %w{. , : .. ., ,, :: (()) @ ~ <<>< A1A1A1...a;}
|
|
165
|
-
rhymes.each do |i|
|
|
166
|
-
it "({ rhyme: '#{i}' })" do
|
|
167
|
-
poem = @poefy.poem ({ rhyme: i })
|
|
168
|
-
expect(poem.count).to be i.gsub(/[0-9]/,'').length
|
|
169
|
-
end
|
|
170
|
-
end
|
|
171
|
-
end
|
|
172
|
-
|
|
173
|
-
describe "should raise error if can't parse rhyme string" do
|
|
174
|
-
rhymes = %w{a1 b1 ab1 Ab1 AAAAABb1 1 1111 1122 11221 ;;::1. }
|
|
175
|
-
rhymes += ['AA Bb1','11 11','11 1 1','..1.']
|
|
176
|
-
rhymes.each do |i|
|
|
177
|
-
it "({ rhyme: '#{i}' })" do
|
|
178
|
-
expect {
|
|
179
|
-
@poefy.poem ({ rhyme: i })
|
|
180
|
-
}.to raise_error(Poefy::RhymeError)
|
|
181
|
-
end
|
|
182
|
-
end
|
|
183
|
-
end
|
|
184
|
-
|
|
185
|
-
describe "should raise error if can't complete rhyme string" do
|
|
186
|
-
rhymes = %w{aaaaaa abcd aaaaabbbbb}
|
|
187
|
-
rhymes.each do |i|
|
|
188
|
-
it "({ rhyme: '#{i}' })" do
|
|
189
|
-
expect {
|
|
190
|
-
@poefy.poem ({ rhyme: i })
|
|
191
|
-
}.to raise_error(Poefy::NotEnoughData)
|
|
192
|
-
end
|
|
193
|
-
end
|
|
194
|
-
end
|
|
195
|
-
|
|
196
|
-
describe "should correctly repeat uppercase lines" do
|
|
197
|
-
lines = 200
|
|
198
|
-
it "({ rhyme: 'A' * #{lines} })" do
|
|
199
|
-
poem = @poefy.poem ({ rhyme: 'A' * lines })
|
|
200
|
-
expect(poem.count).to be lines
|
|
201
|
-
expect(poem.uniq.count).to be 1
|
|
202
|
-
end
|
|
203
|
-
it "({ rhyme: ('A'..'C').to_a.map { |i| i * #{lines} }.join })" do
|
|
204
|
-
rhyme = ('A'..'C').to_a.map { |i| i * lines }.join
|
|
205
|
-
poem = @poefy.poem ({ rhyme: rhyme })
|
|
206
|
-
expect(poem.count).to be lines * 3
|
|
207
|
-
expect(poem.uniq.count).to be 3
|
|
208
|
-
end
|
|
209
|
-
end
|
|
210
|
-
|
|
211
|
-
describe "should raise error if can't complete repeating rhyme string" do
|
|
212
|
-
lines = 200
|
|
213
|
-
it "({ rhyme: ('A'..'D').to_a.map { |i| i * #{lines} }.join })" do
|
|
214
|
-
rhyme = ('A'..'D').to_a.map { |i| i * lines }.join
|
|
215
|
-
expect {
|
|
216
|
-
@poefy.poem ({ rhyme: rhyme })
|
|
217
|
-
}.to raise_error(Poefy::NotEnoughData)
|
|
218
|
-
end
|
|
219
|
-
end
|
|
220
|
-
|
|
221
|
-
end
|
|
222
|
-
|
|
223
|
-
describe ":form option" do
|
|
224
|
-
|
|
225
|
-
describe "should return correct number of lines" do
|
|
226
|
-
it "({ form: :default })" do
|
|
227
|
-
poem = @poefy.poem ({ form: :default })
|
|
228
|
-
expect(poem.count).to be 1
|
|
229
|
-
end
|
|
230
|
-
end
|
|
231
|
-
|
|
232
|
-
describe "should raise error if given a named form it can't fulfil" do
|
|
233
|
-
it "({ form: 'sonnet' })" do
|
|
234
|
-
expect {
|
|
235
|
-
@poefy.poem ({ form: 'sonnet' })
|
|
236
|
-
}.to raise_error(Poefy::NotEnoughData)
|
|
237
|
-
end
|
|
238
|
-
it "({ form: :villanelle })" do
|
|
239
|
-
expect {
|
|
240
|
-
@poefy.poem ({ form: 'villanelle' })
|
|
241
|
-
}.to raise_error(Poefy::NotEnoughData)
|
|
242
|
-
end
|
|
243
|
-
end
|
|
244
|
-
|
|
245
|
-
describe "should raise error if given a junk named form" do
|
|
246
|
-
it "({ form: 'sonnet_junk' })" do
|
|
247
|
-
expect {
|
|
248
|
-
@poefy.poem ({ form: 'sonnet_junk' })
|
|
249
|
-
}.to raise_error(Poefy::MissingFormOrRhyme)
|
|
250
|
-
end
|
|
251
|
-
it "({ form: :not_a_form })" do
|
|
252
|
-
expect {
|
|
253
|
-
@poefy.poem ({ form: :not_a_form })
|
|
254
|
-
}.to raise_error(Poefy::MissingFormOrRhyme)
|
|
255
|
-
end
|
|
256
|
-
it "({ form: :not_a_form, indent: '0010' })" do
|
|
257
|
-
expect {
|
|
258
|
-
@poefy.poem ({ form: :not_a_form, indent: '0010' })
|
|
259
|
-
}.to raise_error(Poefy::MissingFormOrRhyme)
|
|
260
|
-
end
|
|
261
|
-
end
|
|
262
|
-
|
|
263
|
-
describe "should be valid if given a junk named form, and a rhyme" do
|
|
264
|
-
it "({ form: :not_a_form, rhyme: 'abcb' })" do
|
|
265
|
-
poem = @poefy.poem ({ form: :not_a_form, rhyme: 'abcb' })
|
|
266
|
-
expect(poem.count).to be 4
|
|
267
|
-
end
|
|
268
|
-
end
|
|
269
|
-
|
|
270
|
-
describe "should overwrite a named form if another option is specified" do
|
|
271
|
-
it "({ form: 'default', rhyme: 'ab' })" do
|
|
272
|
-
poem = @poefy.poem ({ form: 'default', rhyme: 'ab' })
|
|
273
|
-
expect(poem.count).to be 2
|
|
274
|
-
end
|
|
275
|
-
it "({ form: :villanelle, rhyme: 'abcb' })" do
|
|
276
|
-
poem = @poefy.poem ({ form: :villanelle, rhyme: 'abcb' })
|
|
277
|
-
expect(poem.count).to be 4
|
|
278
|
-
end
|
|
279
|
-
end
|
|
280
|
-
end
|
|
281
|
-
end
|
|
282
|
-
|
|
283
|
-
##############################################################################
|
|
284
|
-
|
|
285
|
-
describe "using dataset 'spec_shakespeare'" do
|
|
286
|
-
|
|
287
|
-
file_txt = "shakespeare_sonnets.txt"
|
|
288
|
-
file_db = "spec_shakespeare"
|
|
289
|
-
|
|
290
|
-
# All the Shakespeare lines are pentameter, so some forms should fail.
|
|
291
|
-
forms = Poefy::PoeticForms::POETIC_FORMS
|
|
292
|
-
forms_fail = [:limerick, :haiku, :common, :ballad, :double_dactyl]
|
|
293
|
-
forms_pass = forms.keys - forms_fail
|
|
294
|
-
|
|
295
|
-
before(:each) do
|
|
296
|
-
@poefy = Poefy::Poem.new(file_db, { proper: false })
|
|
297
|
-
end
|
|
298
|
-
after(:each) do
|
|
299
|
-
@poefy.close
|
|
300
|
-
end
|
|
301
|
-
|
|
302
|
-
it "initialised object not nil" do
|
|
303
|
-
expect(@poefy).to_not be_nil
|
|
304
|
-
end
|
|
305
|
-
|
|
306
|
-
describe "#make_database( '#{@root}/data/#{file_txt}', true )" do
|
|
307
|
-
it "should make the database '#{@root}/data/#{file_db}" do
|
|
308
|
-
input = `sed '/[a-z]/!d' #{@root}/data/#{file_txt}`
|
|
309
|
-
@poefy.make_database! input
|
|
310
|
-
expect(@poefy.corpus.
|
|
311
|
-
end
|
|
312
|
-
end
|
|
313
|
-
|
|
314
|
-
describe "using acrostic option" do
|
|
315
|
-
describe "should return correct number of lines" do
|
|
316
|
-
it "({ form: :sonnet, acrostic: 'pauldpthompson' })" do
|
|
317
|
-
poem = @poefy.poem ({ form: :sonnet,
|
|
318
|
-
acrostic: 'pauldpthompson' })
|
|
319
|
-
expect(poem.count).to be 14
|
|
320
|
-
end
|
|
321
|
-
end
|
|
322
|
-
describe "should raise Poefy::NotEnoughData" do
|
|
323
|
-
it "({ form: :sonnet, acrostic: 'qqqqqqqqqqqqqq' })" do
|
|
324
|
-
expect {
|
|
325
|
-
@poefy.poem ({ form: :sonnet,
|
|
326
|
-
acrostic: 'qqqqqqqqqqqqqq' })
|
|
327
|
-
}.to raise_error(Poefy::NotEnoughData)
|
|
328
|
-
end
|
|
329
|
-
end
|
|
330
|
-
end
|
|
331
|
-
|
|
332
|
-
describe "using form string" do
|
|
333
|
-
describe "should return correct number of lines" do
|
|
334
|
-
|
|
335
|
-
# Make sure each form's lines match the expected output.
|
|
336
|
-
# Generate a few to be sure.
|
|
337
|
-
forms_pass.each do |form|
|
|
338
|
-
it "({ form: #{form} })" do
|
|
339
|
-
10.times do
|
|
340
|
-
poem = @poefy.poem ({ form: form })
|
|
341
|
-
expect(poem.count).to satisfy do |c|
|
|
342
|
-
[*forms[form][:rhyme]].map do |r|
|
|
343
|
-
r.gsub(/[0-9]/,'').length
|
|
344
|
-
end.include?(c)
|
|
345
|
-
end
|
|
346
|
-
end
|
|
347
|
-
end
|
|
348
|
-
end
|
|
349
|
-
end
|
|
350
|
-
|
|
351
|
-
describe "should raise Poefy::NotEnoughData" do
|
|
352
|
-
forms_fail.each do |form|
|
|
353
|
-
it "({ form: #{form} })" do
|
|
354
|
-
4.times do
|
|
355
|
-
expect {
|
|
356
|
-
@poefy.poem ({ form: form })
|
|
357
|
-
}.to raise_error(Poefy::NotEnoughData)
|
|
358
|
-
end
|
|
359
|
-
end
|
|
360
|
-
end
|
|
361
|
-
end
|
|
362
|
-
end
|
|
363
|
-
|
|
364
|
-
describe "make sonnets" do
|
|
365
|
-
sonnet_options = [
|
|
366
|
-
{ rhyme: 'ababcdcdefefgg' },
|
|
367
|
-
{ rhyme: 'abab cdcd efef gg', indent: '0101 0101 0011 01' },
|
|
368
|
-
{ form: 'sonnet' },
|
|
369
|
-
{ form: :sonnet, syllable: 0 },
|
|
370
|
-
{ form: :sonnet, syllable: 10 },
|
|
371
|
-
{ form: :sonnet, regex: /^[A-Z].*$/ },
|
|
372
|
-
{ form: :sonnet, regex: '^[A-Z].*$' },
|
|
373
|
-
{ form: :sonnet, acrostic: 'pauldpthompson' },
|
|
374
|
-
{ form: 'sonnet', indent: '01010101001101' },
|
|
375
|
-
{ form: 'sonnet', proper: false }
|
|
376
|
-
]
|
|
377
|
-
sonnet_options.each do |option|
|
|
378
|
-
it "#{option}" do
|
|
379
|
-
4.times do
|
|
380
|
-
poem = @poefy.poem(option)
|
|
381
|
-
expect(poem).to_not be_nil
|
|
382
|
-
end
|
|
383
|
-
end
|
|
384
|
-
end
|
|
385
|
-
end
|
|
386
|
-
end
|
|
387
|
-
|
|
388
|
-
##############################################################################
|
|
389
|
-
|
|
390
|
-
describe "using dataset 'spec_whitman'" do
|
|
391
|
-
|
|
392
|
-
file_txt = "whitman_leaves.txt"
|
|
393
|
-
file_db = "spec_whitman"
|
|
394
|
-
|
|
395
|
-
# There's a good mix of syllable count, so all forms should pass.
|
|
396
|
-
forms = Poefy::PoeticForms::POETIC_FORMS
|
|
397
|
-
forms_pass = forms.keys
|
|
398
|
-
|
|
399
|
-
before(:each) do
|
|
400
|
-
@poefy = Poefy::Poem.new(file_db, { proper: false })
|
|
401
|
-
end
|
|
402
|
-
after(:each) do
|
|
403
|
-
@poefy.close
|
|
404
|
-
end
|
|
405
|
-
|
|
406
|
-
it "initialised object not nil" do
|
|
407
|
-
expect(@poefy).to_not be_nil
|
|
408
|
-
end
|
|
409
|
-
|
|
410
|
-
describe "#make_database( '#{@root}/data/#{file_txt}', true )" do
|
|
411
|
-
it "should make the database '#{@root}/data/#{file_db}" do
|
|
412
|
-
input = `sed '/[a-z]/!d' #{@root}/data/#{file_txt}`
|
|
413
|
-
@poefy.make_database! input
|
|
414
|
-
expect(@poefy.corpus.
|
|
415
|
-
end
|
|
416
|
-
end
|
|
417
|
-
|
|
418
|
-
describe "using form string" do
|
|
419
|
-
describe "should return correct number of lines" do
|
|
420
|
-
|
|
421
|
-
# Make sure each form's lines match the expected output.
|
|
422
|
-
# Generate a few to be sure.
|
|
423
|
-
forms_pass.each do |form|
|
|
424
|
-
it "({ form: #{form} })" do
|
|
425
|
-
10.times do
|
|
426
|
-
poem = @poefy.poem ({ form: form })
|
|
427
|
-
expect(poem.count).to satisfy do |c|
|
|
428
|
-
[*forms[form][:rhyme]].map do |r|
|
|
429
|
-
r.gsub(/[0-9]/,'').length
|
|
430
|
-
end.include?(c)
|
|
431
|
-
end
|
|
432
|
-
end
|
|
433
|
-
end
|
|
434
|
-
end
|
|
435
|
-
end
|
|
436
|
-
end
|
|
437
|
-
|
|
438
|
-
describe "make sonnets" do
|
|
439
|
-
sonnet_options = [
|
|
440
|
-
{ rhyme: 'ababcdcdefefgg' },
|
|
441
|
-
{ rhyme: 'abab cdcd efef gg', indent: '0101 0101 0011 01' },
|
|
442
|
-
{ form: 'sonnet' },
|
|
443
|
-
{ form: :sonnet, syllable: 0 },
|
|
444
|
-
{ form: :sonnet, syllable: 10 },
|
|
445
|
-
{ form: :sonnet, regex: /^[A-Z].*$/ },
|
|
446
|
-
{ form: :sonnet, regex: '^[A-Z].*$' },
|
|
447
|
-
{ form: :sonnet, acrostic: 'pauldpthompson' },
|
|
448
|
-
{ form: 'sonnet', indent: '01010101001101' },
|
|
449
|
-
{ form: 'sonnet', proper: false }
|
|
450
|
-
]
|
|
451
|
-
sonnet_options.each do |option|
|
|
452
|
-
it "#{option}" do
|
|
453
|
-
4.times do
|
|
454
|
-
poem = @poefy.poem(option)
|
|
455
|
-
expect(poem).to_not be_nil
|
|
456
|
-
end
|
|
457
|
-
end
|
|
458
|
-
end
|
|
459
|
-
end
|
|
460
|
-
|
|
461
|
-
describe "using syllable string" do
|
|
462
|
-
|
|
463
|
-
it "({ rhyme: 'abcb defe', syllable: '[8,6,8,6,0,8,6,8,6]' })" do
|
|
464
|
-
options = {
|
|
465
|
-
rhyme: 'abcb defe',
|
|
466
|
-
syllable: '[8,6,8,6,0,8,6,8,6]'
|
|
467
|
-
}
|
|
468
|
-
poem = @poefy.poem (options)
|
|
469
|
-
expect(poem.count).to be options[:rhyme].length
|
|
470
|
-
end
|
|
471
|
-
|
|
472
|
-
it "({ rhyme: 'abcb defe', syllable: '[8,6,8,6,8,6,8,6]' })" do
|
|
473
|
-
options = {
|
|
474
|
-
rhyme: 'abcb defe',
|
|
475
|
-
syllable: '[8,6,8,6,8,6,8,6]'
|
|
476
|
-
}
|
|
477
|
-
poem = @poefy.poem (options)
|
|
478
|
-
expect(poem.count).to be options[:rhyme].length
|
|
479
|
-
end
|
|
480
|
-
end
|
|
481
|
-
end
|
|
482
|
-
|
|
483
|
-
##############################################################################
|
|
484
|
-
|
|
485
|
-
describe "reusing the same Poem instance" do
|
|
486
|
-
it "should correctly merge the option hashes" do
|
|
487
|
-
|
|
488
|
-
# Default to use rondeau poetic form, and proper sentence validation
|
|
489
|
-
poefy = Poefy::Poem.new(
|
|
490
|
-
'spec_shakespeare',
|
|
491
|
-
{ form: 'rondeau', proper: true }
|
|
492
|
-
)
|
|
493
|
-
|
|
494
|
-
# Generate a properly sentenced rondeau
|
|
495
|
-
poem = poefy.poem
|
|
496
|
-
expect(poem.count).to be 17
|
|
497
|
-
|
|
498
|
-
# Generate a rondeau without proper validation
|
|
499
|
-
poem = poefy.poem ({ proper: false })
|
|
500
|
-
expect(poem.count).to be 17
|
|
501
|
-
|
|
502
|
-
# Generate a proper rondeau with a certain indentation
|
|
503
|
-
poem = poefy.poem ({ indent: '01012 0012 010112' })
|
|
504
|
-
expect(poem.count).to be 17
|
|
505
|
-
|
|
506
|
-
# Generate other forms
|
|
507
|
-
poem = poefy.poem ({ rhyme: 'abbaabbacdecde' })
|
|
508
|
-
expect(poem.count).to be 14
|
|
509
|
-
poem = poefy.poem ({ form: 'sonnet' })
|
|
510
|
-
expect(poem.count).to be 14
|
|
511
|
-
poem = poefy.poem ({ form: 'ballade' })
|
|
512
|
-
expect(poem.count).to be 31
|
|
513
|
-
|
|
514
|
-
# Generate a default rondeau again
|
|
515
|
-
poem = poefy.poem
|
|
516
|
-
expect(poem.count).to be 17
|
|
517
|
-
|
|
518
|
-
poefy.close
|
|
519
|
-
end
|
|
520
|
-
end
|
|
521
|
-
|
|
522
|
-
##############################################################################
|
|
523
|
-
|
|
524
|
-
describe "using the transform option" do
|
|
525
|
-
|
|
526
|
-
it "should correctly transform the output 1" do
|
|
527
|
-
poefy = Poefy::Poem.new :spec_shakespeare
|
|
528
|
-
transform_hash = {
|
|
529
|
-
4 => proc { |line, num, poem| line.upcase },
|
|
530
|
-
12 => proc { |line, num, poem| line.upcase }
|
|
531
|
-
}
|
|
532
|
-
poem = poefy.poem({ form: :sonnet, transform: transform_hash })
|
|
533
|
-
expect(poem.count).to be 14
|
|
534
|
-
expect(poem[3]).to eq poem[3].upcase
|
|
535
|
-
expect(poem[11]).to eq poem[11].upcase
|
|
536
|
-
poefy.close
|
|
537
|
-
end
|
|
538
|
-
|
|
539
|
-
it "should correctly transform the output 2" do
|
|
540
|
-
poefy = Poefy::Poem.new :spec_shakespeare
|
|
541
|
-
transform_hash = {
|
|
542
|
-
4 => proc { |line, num, poem| poem.count },
|
|
543
|
-
-3 => proc { |line, num, poem| poem.count },
|
|
544
|
-
7 => proc { |line, num, poem| 'test string' }
|
|
545
|
-
}
|
|
546
|
-
poem = poefy.poem({ form: :sonnet, transform: transform_hash })
|
|
547
|
-
expect(poem.count).to be 14
|
|
548
|
-
expect(poem[3]).to eq '14'
|
|
549
|
-
expect(poem[11]).to eq '14'
|
|
550
|
-
expect(poem[6]).to eq 'test string'
|
|
551
|
-
poefy.close
|
|
552
|
-
end
|
|
553
|
-
|
|
554
|
-
it "should correctly transform the output 3" do
|
|
555
|
-
poefy = Poefy::Poem.new :spec_shakespeare
|
|
556
|
-
transform_proc = proc { |line, num, poem| line.downcase }
|
|
557
|
-
poem = poefy.poem({ form: :sonnet, transform: transform_proc })
|
|
558
|
-
expect(poem.count).to be 14
|
|
559
|
-
poem.each do |i|
|
|
560
|
-
expect(i).to eq i.downcase
|
|
561
|
-
end
|
|
562
|
-
poefy.close
|
|
563
|
-
end
|
|
564
|
-
|
|
565
|
-
it "should correctly transform the output 4" do
|
|
566
|
-
poefy = Poefy::Poem.new :spec_shakespeare
|
|
567
|
-
transform_proc = proc { |line, num, poem| "#{num} #{line.downcase}" }
|
|
568
|
-
poem = poefy.poem({ form: :sonnet, transform: transform_proc })
|
|
569
|
-
expect(poem.count).to be 14
|
|
570
|
-
poem.each.with_index do |line, index|
|
|
571
|
-
expect(line).to eq line.downcase
|
|
572
|
-
first_word = line.split(' ').first
|
|
573
|
-
expect(first_word).to eq (index + 1).to_s
|
|
574
|
-
end
|
|
575
|
-
poefy.close
|
|
576
|
-
end
|
|
577
|
-
end
|
|
578
|
-
|
|
579
|
-
##############################################################################
|
|
580
|
-
|
|
581
|
-
describe "using the form_from_text option" do
|
|
582
|
-
before(:all) do
|
|
583
|
-
@text = <<-TEXT
|
|
584
|
-
[Chorus 1]
|
|
585
|
-
Oh yeah, I'll tell you something
|
|
586
|
-
I think you'll understand
|
|
587
|
-
When I'll say that something
|
|
588
|
-
I want to hold your hand
|
|
589
|
-
I want to hold your hand
|
|
590
|
-
I want to hold your hand
|
|
591
|
-
|
|
592
|
-
[Verse 1]
|
|
593
|
-
Oh please, say to me
|
|
594
|
-
You'll let me be your man
|
|
595
|
-
And please, say to me
|
|
596
|
-
You'll let me hold your hand
|
|
597
|
-
I'll let me hold your hand
|
|
598
|
-
I want to hold your hand
|
|
599
|
-
TEXT
|
|
600
|
-
@line_count = @text.split("\n").count
|
|
601
|
-
end
|
|
602
|
-
|
|
603
|
-
it "should use the exact poetic form 1" do
|
|
604
|
-
poefy = Poefy::Poem.new(:spec_whitman, {
|
|
605
|
-
form_from_text: @text
|
|
606
|
-
})
|
|
607
|
-
poem = poefy.poem
|
|
608
|
-
poem.map!(&:strip!)
|
|
609
|
-
expect(poem.count).to be @line_count
|
|
610
|
-
expect(poem[0]).to eq "[Chorus 1]"
|
|
611
|
-
expect(poem[8]).to eq "[Verse 1]"
|
|
612
|
-
expect(poem[5]).to eq poem[4]
|
|
613
|
-
expect(poem[6]).to eq poem[4]
|
|
614
|
-
poefy.close
|
|
615
|
-
end
|
|
616
|
-
|
|
617
|
-
it "should use the exact poetic form 2" do
|
|
618
|
-
poefy = Poefy::Poem.new :spec_whitman
|
|
619
|
-
poem = poefy.poem({
|
|
620
|
-
form_from_text: @text
|
|
621
|
-
})
|
|
622
|
-
poem.map!(&:strip!)
|
|
623
|
-
expect(poem.count).to be @line_count
|
|
624
|
-
expect(poem[0]).to eq "[Chorus 1]"
|
|
625
|
-
expect(poem[8]).to eq "[Verse 1]"
|
|
626
|
-
expect(poem[5]).to eq poem[4]
|
|
627
|
-
expect(poem[6]).to eq poem[4]
|
|
628
|
-
poefy.close
|
|
629
|
-
end
|
|
630
|
-
|
|
631
|
-
it "should correctly modify the poetic form 1" do
|
|
632
|
-
poefy = Poefy::Poem.new(:spec_whitman, {
|
|
633
|
-
form_from_text: @text,
|
|
634
|
-
syllable: 6
|
|
635
|
-
})
|
|
636
|
-
poem = poefy.poem
|
|
637
|
-
poem.map!(&:strip!)
|
|
638
|
-
expect(poem.count).to be @line_count
|
|
639
|
-
expect(poem[0]).to eq "[Chorus 1]"
|
|
640
|
-
expect(poem[8]).to eq "[Verse 1]"
|
|
641
|
-
expect(poem[5]).to eq poem[4]
|
|
642
|
-
expect(poem[6]).to eq poem[4]
|
|
643
|
-
poefy.close
|
|
644
|
-
end
|
|
645
|
-
|
|
646
|
-
it "should correctly modify the poetic form 2" do
|
|
647
|
-
poefy = Poefy::Poem.new :spec_whitman
|
|
648
|
-
poem = poefy.poem({
|
|
649
|
-
form_from_text: @text,
|
|
650
|
-
syllable: 6
|
|
651
|
-
})
|
|
652
|
-
poem.map!(&:strip!)
|
|
653
|
-
expect(poem.count).to be @line_count
|
|
654
|
-
expect(poem[0]).to eq "[Chorus 1]"
|
|
655
|
-
expect(poem[8]).to eq "[Verse 1]"
|
|
656
|
-
expect(poem[5]).to eq poem[4]
|
|
657
|
-
expect(poem[6]).to eq poem[4]
|
|
658
|
-
poefy.close
|
|
659
|
-
end
|
|
660
|
-
|
|
661
|
-
it "should correctly modify the poetic form 3" do
|
|
662
|
-
poefy = Poefy::Poem.new(:spec_whitman, {
|
|
663
|
-
form_from_text: @text
|
|
664
|
-
})
|
|
665
|
-
poem = poefy.poem({
|
|
666
|
-
syllable: 6
|
|
667
|
-
})
|
|
668
|
-
poem.map!(&:strip!)
|
|
669
|
-
expect(poem.count).to be @line_count
|
|
670
|
-
expect(poem[0]).to eq "[Chorus 1]"
|
|
671
|
-
expect(poem[8]).to eq "[Verse 1]"
|
|
672
|
-
expect(poem[5]).to eq poem[4]
|
|
673
|
-
expect(poem[6]).to eq poem[4]
|
|
674
|
-
poefy.close
|
|
675
|
-
end
|
|
676
|
-
|
|
677
|
-
it "should correctly replace the poetic form" do
|
|
678
|
-
poefy = Poefy::Poem.new(:spec_whitman, {
|
|
679
|
-
syllable: 6
|
|
680
|
-
})
|
|
681
|
-
poem = poefy.poem({
|
|
682
|
-
form_from_text: @text
|
|
683
|
-
})
|
|
684
|
-
poem.map!(&:strip!)
|
|
685
|
-
expect(poem.count).to be @line_count
|
|
686
|
-
expect(poem[0]).to eq "[Chorus 1]"
|
|
687
|
-
expect(poem[8]).to eq "[Verse 1]"
|
|
688
|
-
expect(poem[5]).to eq poem[4]
|
|
689
|
-
expect(poem[6]).to eq poem[4]
|
|
690
|
-
poefy.close
|
|
691
|
-
end
|
|
692
|
-
|
|
693
|
-
end
|
|
694
|
-
|
|
695
|
-
end
|
|
696
|
-
|
|
697
|
-
################################################################################
|
|
1
|
+
#!/usr/bin/env ruby
|
|
2
|
+
# Encoding: UTF-8
|
|
3
|
+
|
|
4
|
+
################################################################################
|
|
5
|
+
|
|
6
|
+
describe Poefy::Poem, "-- Postgres" do
|
|
7
|
+
|
|
8
|
+
before(:all) do
|
|
9
|
+
require 'poefy/pg'
|
|
10
|
+
@root = Poefy.root
|
|
11
|
+
dbs = %w{spec_test_tiny spec_shakespeare spec_whitman}
|
|
12
|
+
dbs.each do |db_name|
|
|
13
|
+
Poefy::Database.single_exec! <<-SQL
|
|
14
|
+
SET client_min_messages TO WARNING;
|
|
15
|
+
DROP TABLE IF EXISTS #{db_name};
|
|
16
|
+
SQL
|
|
17
|
+
end
|
|
18
|
+
end
|
|
19
|
+
|
|
20
|
+
after(:all) do
|
|
21
|
+
dbs = %w{spec_test_tiny spec_shakespeare spec_whitman}
|
|
22
|
+
dbs.each do |db_name|
|
|
23
|
+
Poefy::Database.single_exec! <<-SQL
|
|
24
|
+
SET client_min_messages TO WARNING;
|
|
25
|
+
DROP TABLE IF EXISTS #{db_name};
|
|
26
|
+
SQL
|
|
27
|
+
end
|
|
28
|
+
end
|
|
29
|
+
|
|
30
|
+
##############################################################################
|
|
31
|
+
|
|
32
|
+
describe "using tiny dataset 'spec_test_tiny'" do
|
|
33
|
+
corpus = "spec_test_tiny"
|
|
34
|
+
|
|
35
|
+
# Create a small corpus of a few rhymes.
|
|
36
|
+
text_array = %w{man plan flan can dan fish dish wish bee sea tree flea}
|
|
37
|
+
text_array.map!{ |i| 'a ' + i }
|
|
38
|
+
text_string = text_array.join("\n")
|
|
39
|
+
row_count = text_array.count
|
|
40
|
+
|
|
41
|
+
before(:each) do
|
|
42
|
+
@poefy = Poefy::Poem.new(corpus, { proper: false })
|
|
43
|
+
end
|
|
44
|
+
after(:each) do
|
|
45
|
+
@poefy.close
|
|
46
|
+
end
|
|
47
|
+
it "initialised object not nil" do
|
|
48
|
+
expect(@poefy).to_not be_nil
|
|
49
|
+
end
|
|
50
|
+
|
|
51
|
+
# Create corpora in the three different ways.
|
|
52
|
+
describe "@poefy#make_database!" do
|
|
53
|
+
|
|
54
|
+
it "Use array of strings" do
|
|
55
|
+
@poefy.make_database! text_array
|
|
56
|
+
expect(@poefy.corpus.exist?).to be true
|
|
57
|
+
expect(@poefy.corpus.count).to be row_count
|
|
58
|
+
poem = @poefy.poem({ rhyme: 'aabb' })
|
|
59
|
+
expect(poem.count).to be 4
|
|
60
|
+
end
|
|
61
|
+
|
|
62
|
+
it "Use one long newline delimited string" do
|
|
63
|
+
@poefy.make_database! text_string
|
|
64
|
+
expect(@poefy.corpus.exist?).to be true
|
|
65
|
+
expect(@poefy.corpus.count).to be row_count
|
|
66
|
+
poem = @poefy.poem({ rhyme: 'aabb' })
|
|
67
|
+
expect(poem.count).to be 4
|
|
68
|
+
end
|
|
69
|
+
|
|
70
|
+
it "Use text lines from a file" do
|
|
71
|
+
|
|
72
|
+
# Create a temp file.
|
|
73
|
+
tmp = Tempfile.new('spec-', Poefy.root + '/spec')
|
|
74
|
+
text_path = tmp.path
|
|
75
|
+
tmp.write text_string
|
|
76
|
+
tmp.close
|
|
77
|
+
|
|
78
|
+
@poefy.make_database! text_path
|
|
79
|
+
expect(@poefy.corpus.exist?).to be true
|
|
80
|
+
expect(@poefy.corpus.count).to be row_count
|
|
81
|
+
poem = @poefy.poem({ rhyme: 'aabb' })
|
|
82
|
+
expect(poem.count).to be 4
|
|
83
|
+
|
|
84
|
+
# Delete the temp file.
|
|
85
|
+
tmp.delete
|
|
86
|
+
end
|
|
87
|
+
end
|
|
88
|
+
|
|
89
|
+
# Make sure that the description can be updated as specified
|
|
90
|
+
# and that it doesn't cause SQL injection.
|
|
91
|
+
describe "corpus description using #desc=" do
|
|
92
|
+
it "@poefy.corpus.desc is initially empty" do
|
|
93
|
+
expect(@poefy.corpus.desc).to eq ''
|
|
94
|
+
end
|
|
95
|
+
|
|
96
|
+
values = [
|
|
97
|
+
"test",
|
|
98
|
+
" -- test",
|
|
99
|
+
"; -- test",
|
|
100
|
+
"test' -- ",
|
|
101
|
+
"test'' -- ",
|
|
102
|
+
"'test' -- ",
|
|
103
|
+
"'test'' -- ",
|
|
104
|
+
"Shakespeare's sonnets",
|
|
105
|
+
"Shakespeare's -- sonnets",
|
|
106
|
+
"Shakespeare's; -- sonnets",
|
|
107
|
+
"test' ; INSERT INTO spec_test_tiny VALUES('foo') -- ",
|
|
108
|
+
"105 OR 1=1",
|
|
109
|
+
"' or ''='"
|
|
110
|
+
]
|
|
111
|
+
values.each do |value|
|
|
112
|
+
it "@poefy.corpus.desc = #{value}" do
|
|
113
|
+
@poefy.corpus.desc = value
|
|
114
|
+
expect(@poefy.corpus.desc).to eq value
|
|
115
|
+
expect(@poefy.corpus.count).to be row_count
|
|
116
|
+
end
|
|
117
|
+
end
|
|
118
|
+
end
|
|
119
|
+
|
|
120
|
+
describe ":rhyme option" do
|
|
121
|
+
|
|
122
|
+
describe "should raise Poefy::MissingFormOrRhyme" do
|
|
123
|
+
it "blank, no argument" do
|
|
124
|
+
expect {
|
|
125
|
+
@poefy.poem
|
|
126
|
+
}.to raise_error(Poefy::MissingFormOrRhyme)
|
|
127
|
+
end
|
|
128
|
+
it "({ })" do
|
|
129
|
+
expect {
|
|
130
|
+
@poefy.poem ({ })
|
|
131
|
+
}.to raise_error(Poefy::MissingFormOrRhyme)
|
|
132
|
+
end
|
|
133
|
+
it "({ rhyme: nil })" do
|
|
134
|
+
expect {
|
|
135
|
+
@poefy.poem ({ rhyme: nil })
|
|
136
|
+
}.to raise_error(Poefy::MissingFormOrRhyme)
|
|
137
|
+
end
|
|
138
|
+
it "({ rhyme: ' ' })" do
|
|
139
|
+
expect {
|
|
140
|
+
@poefy.poem ({ rhyme: ' ' })
|
|
141
|
+
}.to raise_error(Poefy::MissingFormOrRhyme)
|
|
142
|
+
end
|
|
143
|
+
it "({ rhyme: '' })" do
|
|
144
|
+
expect {
|
|
145
|
+
@poefy.poem ({ rhyme: '' })
|
|
146
|
+
}.to raise_error(Poefy::MissingFormOrRhyme)
|
|
147
|
+
end
|
|
148
|
+
end
|
|
149
|
+
|
|
150
|
+
describe "should return correct number of lines" do
|
|
151
|
+
rhymes = %w{a b z A aa ab zz AA AB AA1 A1 B1 Z1 AB1 A1A1A1A1B1B1B1B1B1}
|
|
152
|
+
rhymes += ['A1A1A1 A1A1A1 B1B1B1B1B1B1','a b c a b c']
|
|
153
|
+
rhymes += [' abc','abc ',' abc ']
|
|
154
|
+
rhymes += ['n aaa n','n aXXXa N1']
|
|
155
|
+
rhymes.each do |i|
|
|
156
|
+
it "({ rhyme: '#{i}' })" do
|
|
157
|
+
poem = @poefy.poem ({ rhyme: i })
|
|
158
|
+
expect(poem.count).to be i.gsub(/[0-9]/,'').length
|
|
159
|
+
end
|
|
160
|
+
end
|
|
161
|
+
end
|
|
162
|
+
|
|
163
|
+
describe "should accept characters other than number" do
|
|
164
|
+
rhymes = %w{. , : .. ., ,, :: (()) @ ~ <<>< A1A1A1...a;}
|
|
165
|
+
rhymes.each do |i|
|
|
166
|
+
it "({ rhyme: '#{i}' })" do
|
|
167
|
+
poem = @poefy.poem ({ rhyme: i })
|
|
168
|
+
expect(poem.count).to be i.gsub(/[0-9]/,'').length
|
|
169
|
+
end
|
|
170
|
+
end
|
|
171
|
+
end
|
|
172
|
+
|
|
173
|
+
describe "should raise error if can't parse rhyme string" do
|
|
174
|
+
rhymes = %w{a1 b1 ab1 Ab1 AAAAABb1 1 1111 1122 11221 ;;::1. }
|
|
175
|
+
rhymes += ['AA Bb1','11 11','11 1 1','..1.']
|
|
176
|
+
rhymes.each do |i|
|
|
177
|
+
it "({ rhyme: '#{i}' })" do
|
|
178
|
+
expect {
|
|
179
|
+
@poefy.poem ({ rhyme: i })
|
|
180
|
+
}.to raise_error(Poefy::RhymeError)
|
|
181
|
+
end
|
|
182
|
+
end
|
|
183
|
+
end
|
|
184
|
+
|
|
185
|
+
describe "should raise error if can't complete rhyme string" do
|
|
186
|
+
rhymes = %w{aaaaaa abcd aaaaabbbbb}
|
|
187
|
+
rhymes.each do |i|
|
|
188
|
+
it "({ rhyme: '#{i}' })" do
|
|
189
|
+
expect {
|
|
190
|
+
@poefy.poem ({ rhyme: i })
|
|
191
|
+
}.to raise_error(Poefy::NotEnoughData)
|
|
192
|
+
end
|
|
193
|
+
end
|
|
194
|
+
end
|
|
195
|
+
|
|
196
|
+
describe "should correctly repeat uppercase lines" do
|
|
197
|
+
lines = 200
|
|
198
|
+
it "({ rhyme: 'A' * #{lines} })" do
|
|
199
|
+
poem = @poefy.poem ({ rhyme: 'A' * lines })
|
|
200
|
+
expect(poem.count).to be lines
|
|
201
|
+
expect(poem.uniq.count).to be 1
|
|
202
|
+
end
|
|
203
|
+
it "({ rhyme: ('A'..'C').to_a.map { |i| i * #{lines} }.join })" do
|
|
204
|
+
rhyme = ('A'..'C').to_a.map { |i| i * lines }.join
|
|
205
|
+
poem = @poefy.poem ({ rhyme: rhyme })
|
|
206
|
+
expect(poem.count).to be lines * 3
|
|
207
|
+
expect(poem.uniq.count).to be 3
|
|
208
|
+
end
|
|
209
|
+
end
|
|
210
|
+
|
|
211
|
+
describe "should raise error if can't complete repeating rhyme string" do
|
|
212
|
+
lines = 200
|
|
213
|
+
it "({ rhyme: ('A'..'D').to_a.map { |i| i * #{lines} }.join })" do
|
|
214
|
+
rhyme = ('A'..'D').to_a.map { |i| i * lines }.join
|
|
215
|
+
expect {
|
|
216
|
+
@poefy.poem ({ rhyme: rhyme })
|
|
217
|
+
}.to raise_error(Poefy::NotEnoughData)
|
|
218
|
+
end
|
|
219
|
+
end
|
|
220
|
+
|
|
221
|
+
end
|
|
222
|
+
|
|
223
|
+
describe ":form option" do
|
|
224
|
+
|
|
225
|
+
describe "should return correct number of lines" do
|
|
226
|
+
it "({ form: :default })" do
|
|
227
|
+
poem = @poefy.poem ({ form: :default })
|
|
228
|
+
expect(poem.count).to be 1
|
|
229
|
+
end
|
|
230
|
+
end
|
|
231
|
+
|
|
232
|
+
describe "should raise error if given a named form it can't fulfil" do
|
|
233
|
+
it "({ form: 'sonnet' })" do
|
|
234
|
+
expect {
|
|
235
|
+
@poefy.poem ({ form: 'sonnet' })
|
|
236
|
+
}.to raise_error(Poefy::NotEnoughData)
|
|
237
|
+
end
|
|
238
|
+
it "({ form: :villanelle })" do
|
|
239
|
+
expect {
|
|
240
|
+
@poefy.poem ({ form: 'villanelle' })
|
|
241
|
+
}.to raise_error(Poefy::NotEnoughData)
|
|
242
|
+
end
|
|
243
|
+
end
|
|
244
|
+
|
|
245
|
+
describe "should raise error if given a junk named form" do
|
|
246
|
+
it "({ form: 'sonnet_junk' })" do
|
|
247
|
+
expect {
|
|
248
|
+
@poefy.poem ({ form: 'sonnet_junk' })
|
|
249
|
+
}.to raise_error(Poefy::MissingFormOrRhyme)
|
|
250
|
+
end
|
|
251
|
+
it "({ form: :not_a_form })" do
|
|
252
|
+
expect {
|
|
253
|
+
@poefy.poem ({ form: :not_a_form })
|
|
254
|
+
}.to raise_error(Poefy::MissingFormOrRhyme)
|
|
255
|
+
end
|
|
256
|
+
it "({ form: :not_a_form, indent: '0010' })" do
|
|
257
|
+
expect {
|
|
258
|
+
@poefy.poem ({ form: :not_a_form, indent: '0010' })
|
|
259
|
+
}.to raise_error(Poefy::MissingFormOrRhyme)
|
|
260
|
+
end
|
|
261
|
+
end
|
|
262
|
+
|
|
263
|
+
describe "should be valid if given a junk named form, and a rhyme" do
|
|
264
|
+
it "({ form: :not_a_form, rhyme: 'abcb' })" do
|
|
265
|
+
poem = @poefy.poem ({ form: :not_a_form, rhyme: 'abcb' })
|
|
266
|
+
expect(poem.count).to be 4
|
|
267
|
+
end
|
|
268
|
+
end
|
|
269
|
+
|
|
270
|
+
describe "should overwrite a named form if another option is specified" do
|
|
271
|
+
it "({ form: 'default', rhyme: 'ab' })" do
|
|
272
|
+
poem = @poefy.poem ({ form: 'default', rhyme: 'ab' })
|
|
273
|
+
expect(poem.count).to be 2
|
|
274
|
+
end
|
|
275
|
+
it "({ form: :villanelle, rhyme: 'abcb' })" do
|
|
276
|
+
poem = @poefy.poem ({ form: :villanelle, rhyme: 'abcb' })
|
|
277
|
+
expect(poem.count).to be 4
|
|
278
|
+
end
|
|
279
|
+
end
|
|
280
|
+
end
|
|
281
|
+
end
|
|
282
|
+
|
|
283
|
+
##############################################################################
|
|
284
|
+
|
|
285
|
+
describe "using dataset 'spec_shakespeare'" do
|
|
286
|
+
|
|
287
|
+
file_txt = "shakespeare_sonnets.txt"
|
|
288
|
+
file_db = "spec_shakespeare"
|
|
289
|
+
|
|
290
|
+
# All the Shakespeare lines are pentameter, so some forms should fail.
|
|
291
|
+
forms = Poefy::PoeticForms::POETIC_FORMS
|
|
292
|
+
forms_fail = [:limerick, :haiku, :common, :ballad, :double_dactyl]
|
|
293
|
+
forms_pass = forms.keys - forms_fail
|
|
294
|
+
|
|
295
|
+
before(:each) do
|
|
296
|
+
@poefy = Poefy::Poem.new(file_db, { proper: false })
|
|
297
|
+
end
|
|
298
|
+
after(:each) do
|
|
299
|
+
@poefy.close
|
|
300
|
+
end
|
|
301
|
+
|
|
302
|
+
it "initialised object not nil" do
|
|
303
|
+
expect(@poefy).to_not be_nil
|
|
304
|
+
end
|
|
305
|
+
|
|
306
|
+
describe "#make_database( '#{@root}/data/#{file_txt}', true )" do
|
|
307
|
+
it "should make the database '#{@root}/data/#{file_db}" do
|
|
308
|
+
input = `sed '/[a-z]/!d' #{@root}/data/#{file_txt}`
|
|
309
|
+
@poefy.make_database! input
|
|
310
|
+
expect(@poefy.corpus.exist?).to be true
|
|
311
|
+
end
|
|
312
|
+
end
|
|
313
|
+
|
|
314
|
+
describe "using acrostic option" do
|
|
315
|
+
describe "should return correct number of lines" do
|
|
316
|
+
it "({ form: :sonnet, acrostic: 'pauldpthompson' })" do
|
|
317
|
+
poem = @poefy.poem ({ form: :sonnet,
|
|
318
|
+
acrostic: 'pauldpthompson' })
|
|
319
|
+
expect(poem.count).to be 14
|
|
320
|
+
end
|
|
321
|
+
end
|
|
322
|
+
describe "should raise Poefy::NotEnoughData" do
|
|
323
|
+
it "({ form: :sonnet, acrostic: 'qqqqqqqqqqqqqq' })" do
|
|
324
|
+
expect {
|
|
325
|
+
@poefy.poem ({ form: :sonnet,
|
|
326
|
+
acrostic: 'qqqqqqqqqqqqqq' })
|
|
327
|
+
}.to raise_error(Poefy::NotEnoughData)
|
|
328
|
+
end
|
|
329
|
+
end
|
|
330
|
+
end
|
|
331
|
+
|
|
332
|
+
describe "using form string" do
|
|
333
|
+
describe "should return correct number of lines" do
|
|
334
|
+
|
|
335
|
+
# Make sure each form's lines match the expected output.
|
|
336
|
+
# Generate a few to be sure.
|
|
337
|
+
forms_pass.each do |form|
|
|
338
|
+
it "({ form: #{form} })" do
|
|
339
|
+
10.times do
|
|
340
|
+
poem = @poefy.poem ({ form: form })
|
|
341
|
+
expect(poem.count).to satisfy do |c|
|
|
342
|
+
[*forms[form][:rhyme]].map do |r|
|
|
343
|
+
r.gsub(/[0-9]/,'').length
|
|
344
|
+
end.include?(c)
|
|
345
|
+
end
|
|
346
|
+
end
|
|
347
|
+
end
|
|
348
|
+
end
|
|
349
|
+
end
|
|
350
|
+
|
|
351
|
+
describe "should raise Poefy::NotEnoughData" do
|
|
352
|
+
forms_fail.each do |form|
|
|
353
|
+
it "({ form: #{form} })" do
|
|
354
|
+
4.times do
|
|
355
|
+
expect {
|
|
356
|
+
@poefy.poem ({ form: form })
|
|
357
|
+
}.to raise_error(Poefy::NotEnoughData)
|
|
358
|
+
end
|
|
359
|
+
end
|
|
360
|
+
end
|
|
361
|
+
end
|
|
362
|
+
end
|
|
363
|
+
|
|
364
|
+
describe "make sonnets" do
|
|
365
|
+
sonnet_options = [
|
|
366
|
+
{ rhyme: 'ababcdcdefefgg' },
|
|
367
|
+
{ rhyme: 'abab cdcd efef gg', indent: '0101 0101 0011 01' },
|
|
368
|
+
{ form: 'sonnet' },
|
|
369
|
+
{ form: :sonnet, syllable: 0 },
|
|
370
|
+
{ form: :sonnet, syllable: 10 },
|
|
371
|
+
{ form: :sonnet, regex: /^[A-Z].*$/ },
|
|
372
|
+
{ form: :sonnet, regex: '^[A-Z].*$' },
|
|
373
|
+
{ form: :sonnet, acrostic: 'pauldpthompson' },
|
|
374
|
+
{ form: 'sonnet', indent: '01010101001101' },
|
|
375
|
+
{ form: 'sonnet', proper: false }
|
|
376
|
+
]
|
|
377
|
+
sonnet_options.each do |option|
|
|
378
|
+
it "#{option}" do
|
|
379
|
+
4.times do
|
|
380
|
+
poem = @poefy.poem(option)
|
|
381
|
+
expect(poem).to_not be_nil
|
|
382
|
+
end
|
|
383
|
+
end
|
|
384
|
+
end
|
|
385
|
+
end
|
|
386
|
+
end
|
|
387
|
+
|
|
388
|
+
##############################################################################
|
|
389
|
+
|
|
390
|
+
describe "using dataset 'spec_whitman'" do
|
|
391
|
+
|
|
392
|
+
file_txt = "whitman_leaves.txt"
|
|
393
|
+
file_db = "spec_whitman"
|
|
394
|
+
|
|
395
|
+
# There's a good mix of syllable count, so all forms should pass.
|
|
396
|
+
forms = Poefy::PoeticForms::POETIC_FORMS
|
|
397
|
+
forms_pass = forms.keys
|
|
398
|
+
|
|
399
|
+
before(:each) do
|
|
400
|
+
@poefy = Poefy::Poem.new(file_db, { proper: false })
|
|
401
|
+
end
|
|
402
|
+
after(:each) do
|
|
403
|
+
@poefy.close
|
|
404
|
+
end
|
|
405
|
+
|
|
406
|
+
it "initialised object not nil" do
|
|
407
|
+
expect(@poefy).to_not be_nil
|
|
408
|
+
end
|
|
409
|
+
|
|
410
|
+
describe "#make_database( '#{@root}/data/#{file_txt}', true )" do
|
|
411
|
+
it "should make the database '#{@root}/data/#{file_db}" do
|
|
412
|
+
input = `sed '/[a-z]/!d' #{@root}/data/#{file_txt}`
|
|
413
|
+
@poefy.make_database! input
|
|
414
|
+
expect(@poefy.corpus.exist?).to be true
|
|
415
|
+
end
|
|
416
|
+
end
|
|
417
|
+
|
|
418
|
+
describe "using form string" do
|
|
419
|
+
describe "should return correct number of lines" do
|
|
420
|
+
|
|
421
|
+
# Make sure each form's lines match the expected output.
|
|
422
|
+
# Generate a few to be sure.
|
|
423
|
+
forms_pass.each do |form|
|
|
424
|
+
it "({ form: #{form} })" do
|
|
425
|
+
10.times do
|
|
426
|
+
poem = @poefy.poem ({ form: form })
|
|
427
|
+
expect(poem.count).to satisfy do |c|
|
|
428
|
+
[*forms[form][:rhyme]].map do |r|
|
|
429
|
+
r.gsub(/[0-9]/,'').length
|
|
430
|
+
end.include?(c)
|
|
431
|
+
end
|
|
432
|
+
end
|
|
433
|
+
end
|
|
434
|
+
end
|
|
435
|
+
end
|
|
436
|
+
end
|
|
437
|
+
|
|
438
|
+
describe "make sonnets" do
|
|
439
|
+
sonnet_options = [
|
|
440
|
+
{ rhyme: 'ababcdcdefefgg' },
|
|
441
|
+
{ rhyme: 'abab cdcd efef gg', indent: '0101 0101 0011 01' },
|
|
442
|
+
{ form: 'sonnet' },
|
|
443
|
+
{ form: :sonnet, syllable: 0 },
|
|
444
|
+
{ form: :sonnet, syllable: 10 },
|
|
445
|
+
{ form: :sonnet, regex: /^[A-Z].*$/ },
|
|
446
|
+
{ form: :sonnet, regex: '^[A-Z].*$' },
|
|
447
|
+
{ form: :sonnet, acrostic: 'pauldpthompson' },
|
|
448
|
+
{ form: 'sonnet', indent: '01010101001101' },
|
|
449
|
+
{ form: 'sonnet', proper: false }
|
|
450
|
+
]
|
|
451
|
+
sonnet_options.each do |option|
|
|
452
|
+
it "#{option}" do
|
|
453
|
+
4.times do
|
|
454
|
+
poem = @poefy.poem(option)
|
|
455
|
+
expect(poem).to_not be_nil
|
|
456
|
+
end
|
|
457
|
+
end
|
|
458
|
+
end
|
|
459
|
+
end
|
|
460
|
+
|
|
461
|
+
describe "using syllable string" do
|
|
462
|
+
|
|
463
|
+
it "({ rhyme: 'abcb defe', syllable: '[8,6,8,6,0,8,6,8,6]' })" do
|
|
464
|
+
options = {
|
|
465
|
+
rhyme: 'abcb defe',
|
|
466
|
+
syllable: '[8,6,8,6,0,8,6,8,6]'
|
|
467
|
+
}
|
|
468
|
+
poem = @poefy.poem (options)
|
|
469
|
+
expect(poem.count).to be options[:rhyme].length
|
|
470
|
+
end
|
|
471
|
+
|
|
472
|
+
it "({ rhyme: 'abcb defe', syllable: '[8,6,8,6,8,6,8,6]' })" do
|
|
473
|
+
options = {
|
|
474
|
+
rhyme: 'abcb defe',
|
|
475
|
+
syllable: '[8,6,8,6,8,6,8,6]'
|
|
476
|
+
}
|
|
477
|
+
poem = @poefy.poem (options)
|
|
478
|
+
expect(poem.count).to be options[:rhyme].length
|
|
479
|
+
end
|
|
480
|
+
end
|
|
481
|
+
end
|
|
482
|
+
|
|
483
|
+
##############################################################################
|
|
484
|
+
|
|
485
|
+
describe "reusing the same Poem instance" do
|
|
486
|
+
it "should correctly merge the option hashes" do
|
|
487
|
+
|
|
488
|
+
# Default to use rondeau poetic form, and proper sentence validation
|
|
489
|
+
poefy = Poefy::Poem.new(
|
|
490
|
+
'spec_shakespeare',
|
|
491
|
+
{ form: 'rondeau', proper: true }
|
|
492
|
+
)
|
|
493
|
+
|
|
494
|
+
# Generate a properly sentenced rondeau
|
|
495
|
+
poem = poefy.poem
|
|
496
|
+
expect(poem.count).to be 17
|
|
497
|
+
|
|
498
|
+
# Generate a rondeau without proper validation
|
|
499
|
+
poem = poefy.poem ({ proper: false })
|
|
500
|
+
expect(poem.count).to be 17
|
|
501
|
+
|
|
502
|
+
# Generate a proper rondeau with a certain indentation
|
|
503
|
+
poem = poefy.poem ({ indent: '01012 0012 010112' })
|
|
504
|
+
expect(poem.count).to be 17
|
|
505
|
+
|
|
506
|
+
# Generate other forms
|
|
507
|
+
poem = poefy.poem ({ rhyme: 'abbaabbacdecde' })
|
|
508
|
+
expect(poem.count).to be 14
|
|
509
|
+
poem = poefy.poem ({ form: 'sonnet' })
|
|
510
|
+
expect(poem.count).to be 14
|
|
511
|
+
poem = poefy.poem ({ form: 'ballade' })
|
|
512
|
+
expect(poem.count).to be 31
|
|
513
|
+
|
|
514
|
+
# Generate a default rondeau again
|
|
515
|
+
poem = poefy.poem
|
|
516
|
+
expect(poem.count).to be 17
|
|
517
|
+
|
|
518
|
+
poefy.close
|
|
519
|
+
end
|
|
520
|
+
end
|
|
521
|
+
|
|
522
|
+
##############################################################################
|
|
523
|
+
|
|
524
|
+
describe "using the transform option" do
|
|
525
|
+
|
|
526
|
+
it "should correctly transform the output 1" do
|
|
527
|
+
poefy = Poefy::Poem.new :spec_shakespeare
|
|
528
|
+
transform_hash = {
|
|
529
|
+
4 => proc { |line, num, poem| line.upcase },
|
|
530
|
+
12 => proc { |line, num, poem| line.upcase }
|
|
531
|
+
}
|
|
532
|
+
poem = poefy.poem({ form: :sonnet, transform: transform_hash })
|
|
533
|
+
expect(poem.count).to be 14
|
|
534
|
+
expect(poem[3]).to eq poem[3].upcase
|
|
535
|
+
expect(poem[11]).to eq poem[11].upcase
|
|
536
|
+
poefy.close
|
|
537
|
+
end
|
|
538
|
+
|
|
539
|
+
it "should correctly transform the output 2" do
|
|
540
|
+
poefy = Poefy::Poem.new :spec_shakespeare
|
|
541
|
+
transform_hash = {
|
|
542
|
+
4 => proc { |line, num, poem| poem.count },
|
|
543
|
+
-3 => proc { |line, num, poem| poem.count },
|
|
544
|
+
7 => proc { |line, num, poem| 'test string' }
|
|
545
|
+
}
|
|
546
|
+
poem = poefy.poem({ form: :sonnet, transform: transform_hash })
|
|
547
|
+
expect(poem.count).to be 14
|
|
548
|
+
expect(poem[3]).to eq '14'
|
|
549
|
+
expect(poem[11]).to eq '14'
|
|
550
|
+
expect(poem[6]).to eq 'test string'
|
|
551
|
+
poefy.close
|
|
552
|
+
end
|
|
553
|
+
|
|
554
|
+
it "should correctly transform the output 3" do
|
|
555
|
+
poefy = Poefy::Poem.new :spec_shakespeare
|
|
556
|
+
transform_proc = proc { |line, num, poem| line.downcase }
|
|
557
|
+
poem = poefy.poem({ form: :sonnet, transform: transform_proc })
|
|
558
|
+
expect(poem.count).to be 14
|
|
559
|
+
poem.each do |i|
|
|
560
|
+
expect(i).to eq i.downcase
|
|
561
|
+
end
|
|
562
|
+
poefy.close
|
|
563
|
+
end
|
|
564
|
+
|
|
565
|
+
it "should correctly transform the output 4" do
|
|
566
|
+
poefy = Poefy::Poem.new :spec_shakespeare
|
|
567
|
+
transform_proc = proc { |line, num, poem| "#{num} #{line.downcase}" }
|
|
568
|
+
poem = poefy.poem({ form: :sonnet, transform: transform_proc })
|
|
569
|
+
expect(poem.count).to be 14
|
|
570
|
+
poem.each.with_index do |line, index|
|
|
571
|
+
expect(line).to eq line.downcase
|
|
572
|
+
first_word = line.split(' ').first
|
|
573
|
+
expect(first_word).to eq (index + 1).to_s
|
|
574
|
+
end
|
|
575
|
+
poefy.close
|
|
576
|
+
end
|
|
577
|
+
end
|
|
578
|
+
|
|
579
|
+
##############################################################################
|
|
580
|
+
|
|
581
|
+
describe "using the form_from_text option" do
|
|
582
|
+
before(:all) do
|
|
583
|
+
@text = <<-TEXT
|
|
584
|
+
[Chorus 1]
|
|
585
|
+
Oh yeah, I'll tell you something
|
|
586
|
+
I think you'll understand
|
|
587
|
+
When I'll say that something
|
|
588
|
+
I want to hold your hand
|
|
589
|
+
I want to hold your hand
|
|
590
|
+
I want to hold your hand
|
|
591
|
+
|
|
592
|
+
[Verse 1]
|
|
593
|
+
Oh please, say to me
|
|
594
|
+
You'll let me be your man
|
|
595
|
+
And please, say to me
|
|
596
|
+
You'll let me hold your hand
|
|
597
|
+
I'll let me hold your hand
|
|
598
|
+
I want to hold your hand
|
|
599
|
+
TEXT
|
|
600
|
+
@line_count = @text.split("\n").count
|
|
601
|
+
end
|
|
602
|
+
|
|
603
|
+
it "should use the exact poetic form 1" do
|
|
604
|
+
poefy = Poefy::Poem.new(:spec_whitman, {
|
|
605
|
+
form_from_text: @text
|
|
606
|
+
})
|
|
607
|
+
poem = poefy.poem
|
|
608
|
+
poem.map!(&:strip!)
|
|
609
|
+
expect(poem.count).to be @line_count
|
|
610
|
+
expect(poem[0]).to eq "[Chorus 1]"
|
|
611
|
+
expect(poem[8]).to eq "[Verse 1]"
|
|
612
|
+
expect(poem[5]).to eq poem[4]
|
|
613
|
+
expect(poem[6]).to eq poem[4]
|
|
614
|
+
poefy.close
|
|
615
|
+
end
|
|
616
|
+
|
|
617
|
+
it "should use the exact poetic form 2" do
|
|
618
|
+
poefy = Poefy::Poem.new :spec_whitman
|
|
619
|
+
poem = poefy.poem({
|
|
620
|
+
form_from_text: @text
|
|
621
|
+
})
|
|
622
|
+
poem.map!(&:strip!)
|
|
623
|
+
expect(poem.count).to be @line_count
|
|
624
|
+
expect(poem[0]).to eq "[Chorus 1]"
|
|
625
|
+
expect(poem[8]).to eq "[Verse 1]"
|
|
626
|
+
expect(poem[5]).to eq poem[4]
|
|
627
|
+
expect(poem[6]).to eq poem[4]
|
|
628
|
+
poefy.close
|
|
629
|
+
end
|
|
630
|
+
|
|
631
|
+
it "should correctly modify the poetic form 1" do
|
|
632
|
+
poefy = Poefy::Poem.new(:spec_whitman, {
|
|
633
|
+
form_from_text: @text,
|
|
634
|
+
syllable: 6
|
|
635
|
+
})
|
|
636
|
+
poem = poefy.poem
|
|
637
|
+
poem.map!(&:strip!)
|
|
638
|
+
expect(poem.count).to be @line_count
|
|
639
|
+
expect(poem[0]).to eq "[Chorus 1]"
|
|
640
|
+
expect(poem[8]).to eq "[Verse 1]"
|
|
641
|
+
expect(poem[5]).to eq poem[4]
|
|
642
|
+
expect(poem[6]).to eq poem[4]
|
|
643
|
+
poefy.close
|
|
644
|
+
end
|
|
645
|
+
|
|
646
|
+
it "should correctly modify the poetic form 2" do
|
|
647
|
+
poefy = Poefy::Poem.new :spec_whitman
|
|
648
|
+
poem = poefy.poem({
|
|
649
|
+
form_from_text: @text,
|
|
650
|
+
syllable: 6
|
|
651
|
+
})
|
|
652
|
+
poem.map!(&:strip!)
|
|
653
|
+
expect(poem.count).to be @line_count
|
|
654
|
+
expect(poem[0]).to eq "[Chorus 1]"
|
|
655
|
+
expect(poem[8]).to eq "[Verse 1]"
|
|
656
|
+
expect(poem[5]).to eq poem[4]
|
|
657
|
+
expect(poem[6]).to eq poem[4]
|
|
658
|
+
poefy.close
|
|
659
|
+
end
|
|
660
|
+
|
|
661
|
+
it "should correctly modify the poetic form 3" do
|
|
662
|
+
poefy = Poefy::Poem.new(:spec_whitman, {
|
|
663
|
+
form_from_text: @text
|
|
664
|
+
})
|
|
665
|
+
poem = poefy.poem({
|
|
666
|
+
syllable: 6
|
|
667
|
+
})
|
|
668
|
+
poem.map!(&:strip!)
|
|
669
|
+
expect(poem.count).to be @line_count
|
|
670
|
+
expect(poem[0]).to eq "[Chorus 1]"
|
|
671
|
+
expect(poem[8]).to eq "[Verse 1]"
|
|
672
|
+
expect(poem[5]).to eq poem[4]
|
|
673
|
+
expect(poem[6]).to eq poem[4]
|
|
674
|
+
poefy.close
|
|
675
|
+
end
|
|
676
|
+
|
|
677
|
+
it "should correctly replace the poetic form" do
|
|
678
|
+
poefy = Poefy::Poem.new(:spec_whitman, {
|
|
679
|
+
syllable: 6
|
|
680
|
+
})
|
|
681
|
+
poem = poefy.poem({
|
|
682
|
+
form_from_text: @text
|
|
683
|
+
})
|
|
684
|
+
poem.map!(&:strip!)
|
|
685
|
+
expect(poem.count).to be @line_count
|
|
686
|
+
expect(poem[0]).to eq "[Chorus 1]"
|
|
687
|
+
expect(poem[8]).to eq "[Verse 1]"
|
|
688
|
+
expect(poem[5]).to eq poem[4]
|
|
689
|
+
expect(poem[6]).to eq poem[4]
|
|
690
|
+
poefy.close
|
|
691
|
+
end
|
|
692
|
+
|
|
693
|
+
end
|
|
694
|
+
|
|
695
|
+
end
|
|
696
|
+
|
|
697
|
+
################################################################################
|