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_06(self): 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_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_06(): import logging logger = logging.getLogger("test") logger.setLevel(logging.DEBUG) ch = logging.StreamHandler() ch.setLevel(logging.DEBUG) logger.addHandler(ch) output = StringIO() book = p.Book({"hoja1": [["datos", "de", "prueba"], [1, 2, 3]]}) book.save_to_memory("csv", output) logger.debug(output.getvalue())
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
class TestColumnFormatter: def setUp(self): self.data = { "1": [1, 2, 3, 4, 5, 6, 7, 8], "2": ["1", "2", "3", "4", "5", "6", "7", "8"], "3": [1.1, 2.2, 3.3, 4.4, 5.5, 6.6, 7.7, 8.8], "4": ["1.1", "2.2", "3.3", "4.4", "5.5", "6.6", "7,7", "8.8"], "5": [2, 3, 4, 5, 6, 7, 8, 9], "6": ["2", "3", "4", "5", "6", "7", "8", "9"] } self.io = StringIO() w = pe.Writer(("csv", self.io)) w.write_dict(self.data) #w.close() self.test_tuple = ("csv", self.io.getvalue()) def test_general_usage(self): r = pe.Reader(self.test_tuple) r.add_formatter(pe.formatters.ColumnFormatter( 0, str)) c1 = r.column_at(0)[1:] c2 = self.data["2"] for i in range(0, len(c1)): assert c1[i] == c2[i] def test_one_formatter_for_two_columns(self): r = pe.Reader(self.test_tuple) r.add_formatter(pe.formatters.ColumnFormatter( [0,5], str)) c1 = r.column_at(0)[1:] c2 = self.data["2"] for i in range(0, len(c1)): assert c1[i] == c2[i] c1 = r.column_at(5)[1:] c2 = self.data["6"] for i in range(0, len(c1)): assert c1[i] == c2[i] @raises(NotImplementedError) def test_invalid_input(self): pe.formatters.ColumnFormatter("world", str) @raises(IndexError) def test_invalid_input2(self): """Empty list""" pe.formatters.ColumnFormatter([], str) @raises(IndexError) def test_float_in_list(self): pe.formatters.ColumnFormatter([1, 1.1], str) def test_two_formatters(self): r = pe.Reader(self.test_tuple) r.add_formatter(pe.formatters.ColumnFormatter( 0, str)) c1 = r.column_at(0)[1:] c2 = self.data["2"] for i in range(0, len(c1)): assert c1[i] == c2[i] r.add_formatter(pe.formatters.ColumnFormatter( 0, int)) c1 = r.column_at(0)[1:] c2 = self.data["1"] for i in range(0, len(c1)): assert type(c1[i]) == int assert c1[i] == c2[i] def test_custom_func(self): r = pe.Reader(self.test_tuple) f = lambda x: int(x) + 1 r.add_formatter(pe.formatters.ColumnFormatter( 0, int, f)) c1 = r.column_at(0)[1:] c2 = self.data["5"] for i in range(0, len(c1)): assert c1[i] == c2[i] def test_custom_func_with_a_general_converter(self): r = pe.Reader(self.test_tuple) f = lambda x: int(x) + 1 r.add_formatter(pe.formatters.ColumnFormatter( 0, int, f)) c1 = r.column_at(0)[1:] c2 = self.data["5"] for i in range(0, len(c1)): assert c1[i] == c2[i] r.add_formatter(pe.formatters.ColumnFormatter( 0, str)) c1 = r.column_at(0)[1:] c2 = self.data["6"] for i in range(0, len(c1)): assert c1[i] == c2[i]