milis-site21/test/test_truncate.py

import unittest
import makesite


class TruncateTest(unittest.TestCase):
    def test_truncate(self):
        long_text = '  \n'.join('word' + str(i) for i in range(50))
        expected_text = ' '.join('word' + str(i) for i in range(25))
        self.assertEqual(makesite.truncate(long_text), expected_text)