You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
68 lines
2.0 KiB
68 lines
2.0 KiB
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
# @Time : 2023/4/2 9:33
|
|
# @Author : old tom
|
|
# @File : test_fu_excel.py
|
|
# @Project : futool
|
|
from unittest import TestCase
|
|
from futool.poi.fu_excel import ExcelReader, SimpleExcelWriter
|
|
|
|
ex_reader = ExcelReader(file_path=r'D:\test\test3.xlsx')
|
|
|
|
|
|
# @Desc :
|
|
class TestExcelReader(TestCase):
|
|
|
|
def test_read_sheet(self):
|
|
sheet = ex_reader.load_sheet('Sheet1')
|
|
# for row in sheet:
|
|
# for cell in row:
|
|
# print(cell.value)
|
|
for i, row in enumerate(sheet):
|
|
if i == 0:
|
|
for cell in row:
|
|
print(cell.value)
|
|
|
|
def test_read_sheet_by_index(self):
|
|
sheet = ex_reader.load_sheet_by_index(0)
|
|
for row in sheet:
|
|
for cell in row:
|
|
print(cell.value)
|
|
|
|
def test_read_row(self):
|
|
ex_reader.load_sheet('Sheet1')
|
|
print(ex_reader.read_row(0))
|
|
|
|
def test_read_range_row(self):
|
|
ex_reader.load_sheet('Sheet1')
|
|
print(ex_reader.read_range_rows(1, 3))
|
|
|
|
def test_read_column(self):
|
|
ex_reader.load_sheet('Sheet1')
|
|
print(ex_reader.read_column(4))
|
|
|
|
def test_read_rows(self):
|
|
ex_reader.load_sheet('Sheet1')
|
|
print(ex_reader.read_rows(row_index=[0, 3]))
|
|
|
|
def test_read_all(self):
|
|
ex_reader.load_sheet('Sheet1')
|
|
print(ex_reader.read_all())
|
|
|
|
def test_read_range_column(self):
|
|
ex_reader.load_sheet('Sheet1')
|
|
print(ex_reader.read_range_column(1, 4))
|
|
|
|
|
|
class TestExcelWriter(TestCase):
|
|
def test_write(self):
|
|
ew = SimpleExcelWriter(r'D:\test\test3.xlsx')
|
|
# ew.write(head=['序号', '姓名', '年龄', '身份证', '住址'], data=[('1', '张三', '1', '101111', '特特特特特')])
|
|
ew.write(head=['序号', '姓名', '年龄', '身份证', '住址'], data=self.gen_big_excel())
|
|
|
|
def gen_big_excel(self):
|
|
big_data = []
|
|
for i in range(0, 999999):
|
|
big_data.append(('1', '张三', '1', '101111', '特特特特特'))
|
|
return big_data
|