def test_create_answer_paragraph_is_in_past_tense_if_has_tag_simple_past(self):
base_sentences = [Sentence([Noun('dog'), Verb('like'), Noun('cat'), Punctuation.PERIOD])]
base_paragraph = Paragraph(base_sentences, Tags([StatusTag.SIMPLE_PAST]))
paragaph_str = 'A dog liked cats,'
answer = create_answer_paragraph(paragaph_str, base_paragraph)
with_plurals = PluralsAssignment(base_paragraph).assign_plural([Noun('cat')])
expected = Grammarizer(with_plurals).grammarize_to_past_tense()
self.assertEqual(answer, expected)
def test_create_answer_paragraph_can_pick_plural_when_paragraph_str_has_capital_letters(self):
base_sentences = [Sentence([Noun('dog'), Verb('like'), Noun('cat'), Punctuation.PERIOD])]
base_paragraph = Paragraph(base_sentences)
paragaph_str = 'DOGS like CATS'
answer = create_answer_paragraph(paragaph_str, base_paragraph)
with_plurals = PluralsAssignment(base_paragraph).assign_plural([Noun('dog'), Noun('cat')])
expected = Grammarizer(with_plurals).grammarize_to_present_tense()
self.assertEqual(answer, expected)
def test_create_answer_paragraph_picks_plural_when_plural_and_singular_present(self):
base_sentences = [Sentence([Noun('dog'), Verb('like'), Noun('dog'), Punctuation.PERIOD])]
base_paragraph = Paragraph(base_sentences)
with_plurals = PluralsAssignment(base_paragraph).assign_plural([Noun('dog')])
expected = Grammarizer(with_plurals).grammarize_to_present_tense()
for paragraph_Str in ('dogs like a dog.', 'a dog like dogs.'):
answer = create_answer_paragraph(paragraph_Str, base_paragraph)
self.assertEqual(answer, expected)
def test_create_answer_paragraph_makes_paragraph_according_to_plurals_in_paragraph_str(self):
base_sentences = [Sentence([Noun('dog'), Verb('like'), Noun('cat'), Punctuation.PERIOD])]
base_paragraph = Paragraph(base_sentences)
paragaph_str = 'A dog like cats,'
answer = create_answer_paragraph(paragaph_str, base_paragraph)
with_plurals = PluralsAssignment(base_paragraph).assign_plural([Noun('cat')])
expected = Grammarizer(with_plurals).grammarize_to_present_tense()
self.assertEqual(answer, expected)
def test_create_answer_paragraph_makes_grammatically_correct_paragraph_if_base_paragraph_word_order_is_ok(self):
base_sentences = [Sentence([Pronoun.HE, Verb('like'), Noun('dog'), Punctuation.PERIOD]),
Sentence([Noun('dog'), Verb('go'), BasicWord.preposition('to'), Noun('house'),
Punctuation.PERIOD])]
base_paragraph = Paragraph(base_sentences)
answer = create_answer_paragraph('', base_paragraph)
with_plurals = PluralsAssignment(base_paragraph).assign_plural([])
grammatical = Grammarizer(with_plurals).grammarize_to_present_tense()
self.assertEqual(answer, grammatical)
def test_create_answer_paragraph_uses_base_paragraph_for_word_order_and_punctuation(self):
base_sentences = [
Sentence([Noun('dog'), Verb('play'), BasicWord.preposition('with'), Noun('dog'), Punctuation.PERIOD])]
base_paragraph = Paragraph(base_sentences)
combinations = ('dog dog with play', 'dog dog play with', 'dog with dog play', 'dog with play dog',
'play with dog dog', 'play dog with dog', 'with dog play dog')
with_plurals = PluralsAssignment(base_paragraph).assign_plural([])
expected = Grammarizer(with_plurals).grammarize_to_present_tense()
for paragraph_str in combinations:
answer = create_answer_paragraph(paragraph_str, base_paragraph)
self.assertEqual(answer, expected)
def test_create_answer_paragraph_fully_formed_paragraph_singular_now_assigned_plural(self):
base_sentences = [Sentence([Noun('dog'), Verb('like'), Noun('cat'), Punctuation.PERIOD])]
paragraph = Paragraph(base_sentences)
with_negatives = assign_random_negatives(paragraph, 1.0)
without_plurals = PluralsAssignment(with_negatives).assign_random_plurals(0.0)
grammatical = Grammarizer(without_plurals).grammarize_to_present_tense()
self.assertEqual(str(grammatical), "A dog doesn't like a cat.")
paragraph_str = "dogs cats"
answer = create_answer_paragraph(paragraph_str, grammatical)
with_plurals = PluralsAssignment(with_negatives).assign_plural([Noun('dog'), Noun('cat')])
expected = Grammarizer(with_plurals).grammarize_to_present_tense()
self.assertEqual(answer, expected)
self.assertEqual(str(answer), "Dogs don't like cats.")
def _get_comparitor(self):
answer_paragraph = create_answer_paragraph(self._submission,
self._original)
comparison = ParagraphComparison(answer_paragraph, self._submission)
return comparison