def get_sheet_stream(**keywords):
"""
Get an instance of SheetStream from an excel source
"""
source = factory.get_source(**keywords)
sheets = source.get_data()
sheet_name, data = one_sheet_tuple(sheets.items())
return SheetStream(sheet_name, data)
def test_save_to():
file_type = 'csv'
io = RWManager.get_io(file_type)
g = (i for i in [[1, 2], [3, 4]])
ss = WriteSheetToMemory(file_type=file_type,
file_stream=io,
lineterminator='\n')
sheet_stream = SheetStream("test", g)
ss.write_data(sheet_stream)
content = io.getvalue()
expected = dedent("""\
1,2
3,4
""")
assert content == expected
def get_sheet_stream(**keywords):
source = factory.get_source(**keywords)
sheets = source.get_data()
sheet_name, data = one_sheet_tuple(sheets.items())
return SheetStream(sheet_name, data)