class BodyCopyTest(unittest.TestCase): def setUp(self): with open('corpus.txt') as f: text = f.read() self.model = BodyCopy(text) def test_make_sentences(self): sen = self.model.sentences(sentence_count=2) assert(len(sen) == 2) def test_make_paragraphs(self): para = self.model.paragraphs(paragraph_count=2) assert(len(para) == 2)
def setUp(self): with open('corpus.txt') as f: text = f.read() self.model = BodyCopy(text)