def test_grammarizer_set_nouns_never_sets_uncountable_nouns_to_plural(self):
paragraph = [
[Noun.uncountable_noun('money'), Verb('grab'), Noun.uncountable_noun('tea'), EXCLAMATION],
[Noun('witch'), Verb('have', 'had'), Noun('watch'), PERIOD]
]
grammarizer = Grammarizer(paragraph, probability_plural_noun=1.0)
grammarizer.set_nouns()
noun_info = {
Noun.uncountable_noun('tea'): {'plural': False, 'definite': False, 'countable': False},
Noun('watch'): {'plural': True, 'definite': False, 'countable': True},
Noun('witch'): {'plural': True, 'definite': False, 'countable': True},
Noun.uncountable_noun('money'): {'plural': False, 'definite': False, 'countable': False},
}
self.assertEqual(grammarizer.noun_info, noun_info)
def test_grammarizer_set_nouns(self):
paragraph = [
[Noun.uncountable_noun('money'), Verb('grab'), Noun('witch'), EXCLAMATION],
[Noun('witch'), Verb('play'), Noun('watch'), PERIOD]
]
grammarizer = Grammarizer(paragraph, probability_plural_noun=0.5)
random.seed(10)
grammarizer.set_nouns()
noun_info = {
Noun('watch'): {'plural': True, 'definite': False, 'countable': True},
Noun('witch'): {'plural': False, 'definite': False, 'countable': True},
Noun.uncountable_noun('money'): {'plural': False, 'definite': False, 'countable': False},
}
self.assertEqual(grammarizer.noun_info, noun_info)
grammarizer.set_nouns()
noun_info = {
Noun('watch'): {'plural': True, 'definite': False, 'countable': True},
Noun('witch'): {'plural': False, 'definite': False, 'countable': True},
Noun.uncountable_noun('money'): {'plural': False, 'definite': False, 'countable': False},
}
self.assertEqual(grammarizer.noun_info, noun_info)
grammarizer.set_nouns()
noun_info = {
Noun('watch'): {'plural': False, 'definite': False, 'countable': True},
Noun('witch'): {'plural': False, 'definite': False, 'countable': True},
Noun.uncountable_noun('money'): {'plural': False, 'definite': False, 'countable': False},
}
self.assertEqual(grammarizer.noun_info, noun_info)