def test_one_part(self):
utils.split_file_by_parts(self.testfile, 1)
FAKEOPEN.assert_called_once_with('testfile-1.txt', 'w')
self.assertEqual(
FAKEFILE.method_calls,
[call.writelines(self.lines), call.close()]
)
def _make_expected_write_calls(self, lines, lines_per_file,
header=False):
expected_write_calls = []
start_line = 1 if header else 0
for i in range(start_line, len(lines), lines_per_file):
if header:
expected_write_calls.append(call.write(self.lines[0]))
expected_write_calls.append(
call.writelines(lines[i:i+lines_per_file]))
expected_write_calls.append(call.close())
return expected_write_calls