def test_all2(self):
t = self.s0 + '\n'
res = _split_lines(3, t)
self.assertEqual(res[0], 3)
self.assertEqual(res[1], 'ab\ncd\nef\n')
self.assertEqual(res[2], '')
def test_all(self):
res = _split_lines(3, self.s0)
self.assertEqual(res[0], 3)
self.assertEqual(res[1], 'ab\ncd\nef')
self.assertEqual(res[2], '')
def test_short(self):
res = _split_lines(10, self.s0)
# print(res[0])
self.assertEqual(res[0], 3)
self.assertEqual(res[1], self.s0)
self.assertEqual(res[2], '')
def test_part(self):
res = _split_lines(2, self.s0)
self.assertEqual(res[0], 2)
self.assertEqual(res[1], 'ab\ncd\n')
self.assertEqual(res[2], 'ef')
def test_zero_lines(self):
res = _split_lines(0, self.s0)
self.assertEqual(res[0], 0)
self.assertEqual(res[1], '')
self.assertEqual(res[2], self.s0)
