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
|
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
|