97396e5c19ecf3d911f54fdbb9385eac5925983b,nltk/test/unit/model/test_models.py,NgramModelTextGenerationTests,setUp,#NgramModelTextGenerationTests#,365

Before Change

    Using MLE estimator, generate some text.

    def setUp(self):
        self.model = MLENgramModel(self.trigram_counter)

    def test_generate_one_no_context(self):
        generated = self.model.generate_one()
        self.assertIn(generated, self.model.ngram_counter.unigrams)

After Change

class NgramModelTextGenerationTests(unittest.TestCase):
    Using MLE estimator, generate some text.

    def setUp(self):
        vocab = NgramModelVocabulary(["a", "b", "c", "d", "z", "<s>", "</s>"], unk_cutoff=1)
        training_text = [list("abcd"), list("egadbe")]
        self.model = MleLanguageModel(3, vocabulary=vocab)
        self.model.fit(training_text)

    def test_generate_one_no_context(self):
        generated = self.model.generate_one()
        self.assertIn(generated, self.model.counts.unigrams)

Instances