def get_book_stream(**keywords): """ Get an instance of BookStream from an excel source Where the dictionary should have text as keys and two dimensional array as values. """ a_source = SOURCE.get_book_source(**keywords) sheets = a_source.get_data() filename, path = a_source.get_source_info() return BookStream(sheets, filename=filename, path=path)
def test_load_from_empty_sheets(): bs = BookStream() bs.load_from_sheets(None) assert bs.number_of_sheets() == 0
def test_book_stream(): bs = BookStream() assert bs.number_of_sheets() == 0
def test_key_sorting(): adict = {"cd": [[1, 3]], "ab": [[2, 3]]} bs = BookStream(adict) # bs[0] should be 'ab' : SheetStream([[2,3]]) assert bs[0].payload == [[2, 3]]
def test_get_book_headers_in_array(): data = iter([["a", "b", "c"], [1, 2, 3]]) book_stream = BookStream({"test": data}) colnames_array = get_book_headers_in_array(book_stream) eq_(colnames_array, [["a", "b", "c"]])
def test_get_book_headers_in_array(): data = iter([['a', 'b', 'c'], [1, 2, 3]]) book_stream = BookStream({'test': data}) colnames_array = get_book_headers_in_array(book_stream) eq_(colnames_array, [['a', 'b', 'c']])