mirror of
https://github.com/jmcnamara/libxlsxwriter
synced 2025-03-28 21:13:14 +00:00
84 lines
2.4 KiB
Python
84 lines
2.4 KiB
Python
###############################################################################
|
|
#
|
|
# Tests for XlsxWriter.
|
|
#
|
|
# Copyright (c), 2013-2021, John McNamara, jmcnamara@cpan.org
|
|
#
|
|
|
|
from ..excel_comparison_test import ExcelComparisonTest
|
|
from ...workbook import Workbook
|
|
|
|
|
|
class TestCompareXLSXFiles(ExcelComparisonTest):
|
|
"""
|
|
Test file created by XlsxWriter against a file created by Excel.
|
|
|
|
"""
|
|
|
|
def setUp(self):
|
|
|
|
self.set_filename('autofilter11.xlsx')
|
|
self.set_text_file('autofilter_data.txt')
|
|
|
|
def test_create_file(self):
|
|
"""
|
|
Test the creation of a simple XlsxWriter file with an autofilter.
|
|
"""
|
|
|
|
workbook = Workbook(self.got_filename)
|
|
worksheet = workbook.add_worksheet()
|
|
|
|
# Set the autofilter.
|
|
worksheet.autofilter('A1:D51')
|
|
|
|
# Add filter criteria.
|
|
worksheet.filter_column_list('C', [3000, 5000, 8000])
|
|
|
|
# Open a text file with autofilter example data.
|
|
textfile = open(self.txt_filename)
|
|
|
|
# Read the headers from the first line of the input file.
|
|
headers = textfile.readline().strip("\n").split()
|
|
|
|
# Write out the headers.
|
|
worksheet.write_row('A1', headers)
|
|
|
|
# Start writing data after the headers.
|
|
row = 1
|
|
|
|
# Read the rest of the text file and write it to the worksheet.
|
|
for line in textfile:
|
|
|
|
# Split the input data based on whitespace.
|
|
data = line.strip("\n").split()
|
|
|
|
# Convert the number data from the text file.
|
|
for i, item in enumerate(data):
|
|
try:
|
|
data[i] = float(item)
|
|
except ValueError:
|
|
pass
|
|
|
|
# Get some of the field data.
|
|
region = data[0]
|
|
volume = int(data[2])
|
|
|
|
# Check for rows that match the filter.
|
|
if volume == 3000 or volume == 5000 or volume == 8000:
|
|
# Row matches the filter, no further action required.
|
|
pass
|
|
else:
|
|
# We need to hide rows that don't match the filter.
|
|
worksheet.set_row(row, options={'hidden': True})
|
|
|
|
# Write out the row data.
|
|
worksheet.write_row(row, 0, data)
|
|
|
|
# Move on to the next worksheet row.
|
|
row += 1
|
|
|
|
textfile.close()
|
|
workbook.close()
|
|
|
|
self.assertExcelEqual()
|