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)
