93 lines
2.7 KiB
Python
93 lines
2.7 KiB
Python
import io
|
|
import copy
|
|
import pickle
|
|
import tempfile
|
|
from fontTools.ttLib import TTFont
|
|
from fontTools.ttLib.sfnt import calcChecksum, SFNTReader, WOFFFlavorData
|
|
from pathlib import Path
|
|
import pytest
|
|
|
|
TEST_DATA = Path(__file__).parent / "data"
|
|
|
|
|
|
@pytest.fixture
|
|
def ttfont_path():
|
|
font = TTFont()
|
|
font.importXML(TEST_DATA / "TestTTF-Regular.ttx")
|
|
with tempfile.NamedTemporaryFile(suffix=".ttf", delete=False) as fp:
|
|
font_path = Path(fp.name)
|
|
font.save(font_path)
|
|
yield font_path
|
|
font_path.unlink()
|
|
|
|
def test_calcChecksum():
|
|
assert calcChecksum(b"abcd") == 1633837924
|
|
assert calcChecksum(b"abcdxyz") == 3655064932
|
|
|
|
|
|
EMPTY_SFNT = b"\x00\x01\x00\x00" + b"\x00" * 8
|
|
|
|
|
|
def pickle_unpickle(obj):
|
|
return pickle.loads(pickle.dumps(obj))
|
|
|
|
|
|
class SFNTReaderTest:
|
|
@pytest.mark.parametrize("deepcopy", [copy.deepcopy, pickle_unpickle])
|
|
def test_pickle_protocol_FileIO(self, deepcopy, tmp_path):
|
|
fontfile = tmp_path / "test.ttf"
|
|
fontfile.write_bytes(EMPTY_SFNT)
|
|
reader = SFNTReader(fontfile.open("rb"))
|
|
|
|
reader2 = deepcopy(reader)
|
|
|
|
assert reader2 is not reader
|
|
assert reader2.file is not reader.file
|
|
|
|
assert isinstance(reader2.file, io.BufferedReader)
|
|
assert isinstance(reader2.file.raw, io.FileIO)
|
|
assert reader2.file.name == reader.file.name
|
|
assert reader2.file.tell() == reader.file.tell()
|
|
|
|
for k, v in reader.__dict__.items():
|
|
if k == "file":
|
|
continue
|
|
assert getattr(reader2, k) == v
|
|
|
|
@pytest.mark.parametrize("deepcopy", [copy.deepcopy, pickle_unpickle])
|
|
def test_pickle_protocol_BytesIO(self, deepcopy, tmp_path):
|
|
buf = io.BytesIO(EMPTY_SFNT)
|
|
reader = SFNTReader(buf)
|
|
|
|
reader2 = deepcopy(reader)
|
|
|
|
assert reader2 is not reader
|
|
assert reader2.file is not reader.file
|
|
|
|
assert isinstance(reader2.file, io.BytesIO)
|
|
assert reader2.file.tell() == reader.file.tell()
|
|
assert reader2.file.getvalue() == reader.file.getvalue()
|
|
|
|
for k, v in reader.__dict__.items():
|
|
if k == "file":
|
|
continue
|
|
assert getattr(reader2, k) == v
|
|
|
|
def test_ttLib_sfnt_write_privData(tmp_path, ttfont_path):
|
|
output_path = tmp_path / "TestTTF-Regular.woff"
|
|
font = TTFont(ttfont_path)
|
|
|
|
privData = "Private Eyes".encode()
|
|
|
|
data = WOFFFlavorData()
|
|
head = font['head']
|
|
data.majorVersion, data.minorVersion =map(int, format(head.fontRevision, '.3f').split('.'))
|
|
|
|
data.privData = privData
|
|
font.flavor = "woff"
|
|
font.flavorData = data
|
|
font.save(output_path)
|
|
|
|
assert output_path.exists()
|
|
assert TTFont(output_path).flavorData.privData == privData
|