def test_no_split(self):
utils.split_file_by_num_lines(self.testfile, 20)
FAKEOPEN.assert_called_once_with('testfile-1.txt', 'w')
self.assertEqual(
FAKEFILE.method_calls,
[call.writelines(self.lines), call.close()]
)
def test_padding_with_num_total_lines(self):
utils.split_file_by_num_lines(
self.testfile,
2,
pad_file_names=True,
num_lines_total=20
)
self._test_expected_calls_made(self.lines, 10, 2,
'testfile-{:02}.txt')
def test_header(self):
utils.split_file_by_num_lines(self.testfile, 5,
header=True)
self._test_expected_calls_made(self.lines, 4, 5, header=True)
def test_split_by_five(self):
utils.split_file_by_num_lines(self.testfile, 5)
self._test_expected_calls_made(self.lines, 4, 5)
