def test_write_file(self): data = ['a', 'b', 'c'] # TODO: Do tempfile.mkstemp stuff tempfilepath = tempfile.mktemp() transform = utils.write_file(tempfilepath) result = list(transform(data)) self.assertEqual(result, data) with open(tempfilepath, 'r') as temp_f: self.assertEqual(temp_f.read(), 'a\nb\nc\n') # FIXME This probably doesn't remove the file if the assertion fails. os.remove(tempfilepath)
def test_write_file_append(self): data = ['a', 'b', 'c'] tempfilepath = tempfile.mktemp() with open(tempfilepath, 'w') as temp_f: temp_f.write('zzz\n') transform = utils.write_file(tempfilepath, append=True) result = list(transform(data)) self.assertEqual(result, data) with open(tempfilepath, 'r') as temp_f: self.assertEqual(temp_f.read(), 'zzz\na\nb\nc\n') # FIXME This probably doesn't remove the file if the assertion fails. os.remove(tempfilepath)