def test_write_output(): builtin_open_patcher = patch('__builtin__.open') builtin_open_mock = builtin_open_patcher.start() builtin_open_mock.return_value = fp = Mock() fp.__exit__ = fp.__enter__ = lambda *args: fp _write_output('/output.txt', 'sample content') builtin_open_mock.assert_called_once_with('/output.txt', 'w') fp.write.assert_called_once_with('sample content') builtin_open_patcher.stop()
def analysis_transactions(input_file, output_file): reader = _get_reader(input_file) writer = _get_writer(output_file) content = analysis_transactions_with_reader_writer(reader, writer) _write_output(output_file, content)