def test_csv_output_stringio2(self):
data = [[1, 2, 3], [4, 5, 6]]
r = pe.Sheet(data)
io = StringIO()
r.save_to_memory("csv", io)
r = pe.load_from_memory("csv", io.getvalue())
result = ['1', '2', '3', '4', '5', '6']
actual = pe.utils.to_array(r.enumerate())
assert actual == result
def test_csv_output_stringio(self):
data = [[1, 2, 3], [4, 5, 6]]
io = StringIO()
w = pe.Writer(("csv", io))
w.write_rows(data)
w.close()
r = pe.Reader(("csv", io.getvalue()))
result = ['1', '2', '3', '4', '5', '6']
actual = pe.utils.to_array(r.enumerate())
assert actual == result
def test_issue_76():
from pyexcel._compact import StringIO
tsv_stream = StringIO()
tsv_stream.write('1\t2\t3\t4\n')
tsv_stream.write('1\t2\t3\t4\n')
tsv_stream.seek(0)
sheet = pe.get_sheet(file_stream=tsv_stream,
file_type='csv',
delimiter='\t')
data = [[1, 2, 3, 4], [1, 2, 3, 4]]
eq_(sheet.array, data)
def test_issue_06():
import logging
logger = logging.getLogger("test")
logger.setLevel(logging.DEBUG)
ch = logging.StreamHandler()
ch.setLevel(logging.DEBUG)
logger.addHandler(ch)
output = StringIO()
book = pe.Book({'hoja1': [['datos', 'de', 'prueba'], [1, 2, 3]], })
book.save_to_memory('csv', output)
logger.debug(output.getvalue())
def test_issue_76():
from pyexcel._compact import StringIO
tsv_stream = StringIO()
tsv_stream.write("1\t2\t3\t4\n")
tsv_stream.write("1\t2\t3\t4\n")
tsv_stream.seek(0)
sheet = p.get_sheet(
file_stream=tsv_stream, file_type="csv", delimiter="\t"
)
data = [[1, 2, 3, 4], [1, 2, 3, 4]]
eq_(sheet.array, data)
