import os.path import textwrap import unittest from import_issues import MAX_SIZE_COMMENT, split_long_str class TestStringSplitting(unittest.TestCase): maxDiff = None def setUp(self): __cur_dir = os.path.abspath(os.path.dirname(__file__)) with open(os.path.join(__cur_dir, "209_description.txt")) as inf: self.long_str = inf.read() def test_one_line(self): in_str = "Testing string." split_str = split_long_str(in_str, 100) self.assertIsInstance(split_str, list) self.assertEqual(len(split_str), 1) self.assertEqual(in_str, "".join(split_str)) def test_short_str(self): in_str = textwrap.dedent( """\ Reader, I married him. A quiet wedding we had: he and I, the parson and clerk, were alone present. When we got back from church, I went into the kitchen of the manor-house, where Mary was cooking the dinner and John cleaning the knives, and I said— “Mary, I have been married to Mr. Rochester this morning.” The housekeeper and her husband were both of that decent phlegmatic order of people, to whom one may at any time safely communicate a remarkable piece of news without incurring the danger of having one’s ears pierced by some shrill ejaculation, and subsequently stunned by a torrent of wordy wonderment. Mary did look up, and she did stare at me: the ladle with which she was basting a pair of chickens roasting at the fire, did for some three minutes hang suspended in air; and for the same space of time John’s knives also had rest from the polishing process: but Mary, bending again over the roast, said only— “Have you, Miss? Well, for sure!” """ ) split_str = split_long_str(in_str, 100) self.assertIsInstance(split_str, list) self.assertEqual(len(split_str), 6) self.assertEqual(in_str, "".join(split_str)) def test_long_str(self): split_str = split_long_str(self.long_str, MAX_SIZE_COMMENT) self.assertIsInstance(split_str, list) self.assertEqual(len(split_str), 4) self.assertEqual(self.long_str, "".join(split_str)) if __name__ == "__main__": unittest.main()