def test_write_to_file2():
f = BaseOutputFile('test.txt', file_type=FileType.TEXT)
f.add_row('12')
f.write_to_file()
with open(f.file_path, 'r') as tf:
assert '12\n' == tf.read()
f.headers = ['These headers will be skipped for FileType.TEXT']
assert f.write_to_file()
with open(f.file_path, 'r') as tf:
assert '12\n' == tf.read()
os.remove(f.file_path)
def test_write_to_file():
f = BaseOutputFile('test.csv')
f.add_row([1, 2])
f.write_to_file()
with open(f.file_path, 'r') as tf:
assert '1,2\n' == tf.read()
f.headers = ['First', 'Second']
assert f.write_to_file()
with open(f.file_path, 'r') as tf:
assert '"First","Second"\n1,2\n' == tf.read()
os.remove(f.file_path)
def test_write_to_file4():
f = BaseOutputFile('test.csv')
f.file_type = 'This is not a file type'
f.data = [1, 2]
assert not f.write_to_file()
assert not os.path.exists(f.file_path)
def test_write_to_file3():
f = BaseOutputFile('test.csv')
assert not f.write_to_file(
) # Don't write anything if there is nothing to write
assert not os.path.exists(f.file_path)