def test_abnormal(self, invalid_excel_file_path, table_name, start_row, expected): loader = ptr.ExcelTableFileLoader(invalid_excel_file_path) loader.table_name = table_name loader.start_row = start_row for tabletuple in loader.load(): assert tabletuple == []
def test_null_table_name(self, valid_excel_file_path, table_name, expected): loader = ptr.ExcelTableFileLoader(valid_excel_file_path) loader.table_name = table_name with pytest.raises(expected): for _tabletuple in loader.load(): pass
def test_normal(self, monkeypatch, value, source, expected): loader = ptr.ExcelTableFileLoader(source) loader.table_name = value monkeypatch.setattr(ptr.ExcelTableFileLoader, "_sheet_name", self.monkey_property) assert loader.make_table_name() == expected
def test_normal(self, valid_excel_file_path, table_name, start_row, expected_list): loader = ptr.ExcelTableFileLoader(valid_excel_file_path) loader.table_name = table_name loader.start_row = start_row for table_data in loader.load(): print("[actual]\n{}".format(dumps_tabledata(table_data))) assert table_data.in_tabledata_list(expected_list)
def test_normal(self, valid_excel_file_path, table_name, start_row, expected_list): loader = ptr.ExcelTableFileLoader(valid_excel_file_path) loader.table_name = table_name loader.start_row = start_row for table_data in loader.load(): assert table_data.in_tabledata_list(expected_list)
def test_normal(self, valid_excel_file_path, table_name, start_row, expected_tabledata): loader = ptr.ExcelTableFileLoader(valid_excel_file_path) loader.table_name = table_name loader.start_row = start_row for tabledata, expected in zip(loader.load(), expected_tabledata): assert tabledata == expected
def main(): file_path = "sample_data.xlsx" # create sample data file --- workbook = xlsxwriter.Workbook(file_path) worksheet = workbook.add_worksheet("samplesheet1") table = [ ["", "", "", ""], ["", "a", "b", "c"], ["", 1, 1.1, "a"], ["", 2, 2.2, "bb"], ["", 3, 3.3, "cc"], ] for row_idx, row in enumerate(table): for col_idx, item in enumerate(row): worksheet.write(row_idx, col_idx, item) worksheet = workbook.add_worksheet("samplesheet2") worksheet = workbook.add_worksheet("samplesheet3") table = [ ["", "", ""], ["", "", ""], ["aa", "ab", "ac"], [1, "hoge", "a"], [2, "", "bb"], [3, "foo", ""], ] for row_idx, row in enumerate(table): for col_idx, item in enumerate(row): worksheet.write(row_idx, col_idx, item) workbook.close() # create table --- con = simplesqlite.SimpleSQLite("sample.sqlite", "w") loader = pytablereader.ExcelTableFileLoader(file_path) for table_data in loader.load(): con.create_table_from_tabledata(table_data) # output --- for table_name in con.fetch_table_names(): print("table: " + table_name) print(con.fetch_attr_names(table_name)) result = con.select(select="*", table_name=table_name) for record in result.fetchall(): print(record) print()
def test_normal( self, tmpdir, writer_class, table, header, value, expected): test_file_path = tmpdir.join("test.xlsx") writer = writer_class() writer.open(str(test_file_path)) writer.make_worksheet(table) writer.header_list = header writer.value_matrix = value writer.iteration_length = len(value) writer.write_table_iter() writer.close() assert writer.first_data_row == 1 assert writer.last_data_row == 7 loader = ptr.ExcelTableFileLoader(str(test_file_path)) for table_data in loader.load(): assert table_data == expected
def test_normal( self, tmpdir, writer_class, table, header, value, expected): test_file_path = tmpdir.join("test.xlsx") writer = writer_class() writer.open(str(test_file_path)) writer.make_worksheet(table) writer.header_list = header writer.value_matrix = value writer.write_table() writer.close() loader = ptr.ExcelTableFileLoader(str(test_file_path)) for table_data in loader.load(): expected_dump = ptw.dump_tabledata(expected) actual_dump = ptw.dump_tabledata(table_data) print("[expected]\n{}".format(expected_dump)) print("[actual]\n{}".format(actual_dump)) assert actual_dump == expected_dump
def test_normal(self, tmpdir, writer_class, table, header, value, expected): if writer_class == ptw.ExcelXlsTableWriter and not HAS_XLWT: pytest.skip() test_file_path = tmpdir.join("test.xlsx") writer = writer_class() writer.open(str(test_file_path)) writer.make_worksheet(table) writer.header_list = header writer.value_matrix = value writer.write_table() writer.close() loader = ptr.ExcelTableFileLoader(str(test_file_path)) for table_data in loader.load(): expected_dump = ptw.dump_tabledata(expected) actual_dump = ptw.dump_tabledata(table_data) print_test_result(expected=expected_dump, actual=actual_dump) assert actual_dump == expected_dump
def test_null_file_path(self, source, expected): loader = ptr.ExcelTableFileLoader(source) with pytest.raises(expected): for _tabletuple in loader.load(): pass
def test_exception(self, value, source, expected): loader = ptr.ExcelTableFileLoader(source) loader.table_name = value with pytest.raises(expected): loader.make_table_name()
worksheet = workbook.add_worksheet("samplesheet3") table = [ ["", "", ""], ["", "", ""], ["aa", "ab", "ac"], [1, "hoge", "a"], [2, "", "bb"], [3, "foo", ""], ] for row_idx, row in enumerate(table): for col_idx, item in enumerate(row): worksheet.write(row_idx, col_idx, item) workbook.close() # create table --- con = simplesqlite.SimpleSQLite("sample.sqlite", "w") loader = pytablereader.ExcelTableFileLoader(file_path) for table_data in loader.load(): con.create_table_from_tabledata(table_data) # output --- for table_name in con.fetch_table_names(): print("table: " + table_name) print(con.fetch_attr_names(table_name)) result = con.select(select="*", table_name=table_name) for record in result.fetchall(): print(record) print()