Skip to content

Commit c518db1

Browse files
committed
Added String.random_paragraph unit test
1 parent fde73e7 commit c518db1

File tree

2 files changed

+7
-1
lines changed

2 files changed

+7
-1
lines changed

cyaron/str.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -54,6 +54,8 @@ def random_paragraph(sentence_count_range, **kwargs):
5454
if list_like(sentence_count_range):
5555
sentence_count = random.randint(sentence_count_range[0], sentence_count_range[1])
5656

57+
word_count_range = kwargs.get("word_count_range", (6, 10))
58+
5759
first_letter_uppercase = kwargs.get("first_letter_uppercase", True)
5860
kwargs["first_letter_uppercase"] = False
5961

@@ -77,7 +79,7 @@ def random_paragraph(sentence_count_range, **kwargs):
7779
sentences = []
7880
capitalize_next_sentence = True
7981
for i in range(sentence_count):
80-
string = String.random_sentence(**kwargs)
82+
string = String.random_sentence(word_count_range, **kwargs)
8183
sep_or_term = random.random()
8284

8385
if capitalize_next_sentence and first_letter_uppercase:

cyaron/tests/str_test.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,3 +18,7 @@ def test_random_sentence(self):
1818
self.assertTrue(sentence[0].isupper())
1919
self.assertTrue(sentence[-1] == ".")
2020
self.assertTrue(sentence.count(" ") == 9)
21+
22+
def test_random_paragraph(self):
23+
# Only test for Errors
24+
String.random_paragraph(10)

0 commit comments

Comments
 (0)