/usr/lib/python2.7/dist-packages/openpyxl/reader/tests/test_excel.py is in python-openpyxl 2.4.9-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 | from __future__ import absolute_import
# Copyright (c) 2010-2017 openpyxl
from io import BytesIO
from tempfile import NamedTemporaryFile
from zipfile import BadZipfile, ZipFile
from openpyxl.packaging.manifest import Manifest, Override
from openpyxl.utils.exceptions import InvalidFileException
from openpyxl.xml.functions import fromstring
from openpyxl.xml.constants import (
ARC_WORKBOOK,
XLSM,
XLSX,
XLTM,
XLTX,
)
import pytest
@pytest.fixture
def load_workbook():
from ..excel import load_workbook
return load_workbook
def test_read_empty_file(datadir, load_workbook):
datadir.chdir()
with pytest.raises(BadZipfile):
load_workbook('null_file.xlsx')
def test_load_workbook_from_fileobj(datadir, load_workbook):
""" can a workbook be loaded from a file object without exceptions
This tests for regressions of
https://bitbucket.org/openpyxl/openpyxl/issue/433
"""
datadir.chdir()
with open('empty_with_no_properties.xlsx', 'rb') as f:
load_workbook(f)
def test_repair_central_directory():
from ..excel import repair_central_directory, CENTRAL_DIRECTORY_SIGNATURE
data_a = b"foobarbaz" + CENTRAL_DIRECTORY_SIGNATURE
data_b = b"bazbarfoo1234567890123456890"
# The repair_central_directory looks for a magic set of bytes
# (CENTRAL_DIRECTORY_SIGNATURE) and strips off everything 18 bytes past the sequence
f = repair_central_directory(BytesIO(data_a + data_b), True)
assert f.read() == data_a + data_b[:18]
f = repair_central_directory(BytesIO(data_b), True)
assert f.read() == data_b
@pytest.mark.parametrize('wb_type, wb_name', [
(ct, name) for ct in [XLSX, XLSM, XLTX, XLTM]
for name in ['/' + ARC_WORKBOOK, '/xl/spqr.xml']
])
def test_find_standard_workbook_part(datadir, wb_type, wb_name):
from ..excel import _find_workbook_part
src = """
<Types xmlns="http://schemas.openxmlformats.org/package/2006/content-types">
<Override ContentType="{0}"
PartName="{1}"/>
</Types>
""".format(wb_type, wb_name)
node = fromstring(src)
package = Manifest.from_tree(node)
assert _find_workbook_part(package) == Override(wb_name, wb_type)
def test_no_workbook():
from ..excel import _find_workbook_part
with pytest.raises(IOError):
part = _find_workbook_part(Manifest())
def test_overwritten_default():
from ..excel import _find_workbook_part
src = """
<Types xmlns="http://schemas.openxmlformats.org/package/2006/content-types">
<Default Extension="xml" ContentType="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet.main+xml"/>
</Types>
"""
node = fromstring(src)
package = Manifest.from_tree(node)
assert _find_workbook_part(package) == Override("/xl/workbook.xml", XLSX)
@pytest.mark.parametrize("extension",
['.xlsb', '.xls', 'no-format']
)
def test_invalid_file_extension(extension, load_workbook):
tmp = NamedTemporaryFile(suffix=extension)
with pytest.raises(InvalidFileException):
load_workbook(filename=tmp.name)
def test_style_assignment(datadir, load_workbook):
datadir.chdir()
wb = load_workbook("complex-styles.xlsx")
assert len(wb._alignments) == 9
assert len(wb._fills) == 6
assert len(wb._fonts) == 8
assert len(wb._borders) == 7
assert len(wb._number_formats) == 0
assert len(wb._protections) == 1
@pytest.mark.parametrize("ro", [False, True])
def test_close_read(datadir, load_workbook, ro):
datadir.chdir()
wb = load_workbook("complex-styles.xlsx", read_only=ro)
assert hasattr(wb, '_archive') is ro
wb.close()
if ro:
assert wb._archive.fp is None
@pytest.mark.parametrize("wo", [False, True])
def test_close_write(wo):
from openpyxl.workbook import Workbook
wb = Workbook(write_only=wo)
wb.close()
def test_read_stringio(load_workbook):
filelike = BytesIO(b"certainly not a valid XSLX content")
# Test invalid file-like objects are detected and not handled as regular files
with pytest.raises(BadZipfile):
load_workbook(filelike)
def test_load_workbook_with_vba(datadir, load_workbook):
datadir.chdir()
test_file = 'legacy_drawing.xlsm'
# open the workbook directly from the file
wb1 = load_workbook(test_file, keep_vba=True)
# open again from a BytesIO copy
with open(test_file, 'rb') as f:
wb2 = load_workbook(BytesIO(f.read()), keep_vba=True)
assert wb1.vba_archive.namelist() == wb2.vba_archive.namelist()
assert wb1.vba_archive.namelist() == ZipFile(test_file, 'r').namelist()
def test_no_external_links(datadir, load_workbook):
datadir.chdir()
wb = load_workbook("bug137.xlsx", keep_links=False)
assert wb.keep_links is False
|