You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
204 lines
7.0 KiB
Python
204 lines
7.0 KiB
Python
from io import BytesIO
|
|
import multiprocessing
|
|
import os
|
|
from pathlib import Path
|
|
import shutil
|
|
import sys
|
|
import warnings
|
|
|
|
import numpy as np
|
|
import pytest
|
|
|
|
from matplotlib import font_manager as fm
|
|
from matplotlib.font_manager import (
|
|
findfont, findSystemFonts, FontProperties, fontManager, json_dump,
|
|
json_load, get_font, get_fontconfig_fonts, is_opentype_cff_font,
|
|
MSUserFontDirectories, _call_fc_list)
|
|
from matplotlib import pyplot as plt, rc_context
|
|
|
|
has_fclist = shutil.which('fc-list') is not None
|
|
|
|
|
|
def test_font_priority():
|
|
with rc_context(rc={
|
|
'font.sans-serif':
|
|
['cmmi10', 'Bitstream Vera Sans']}):
|
|
font = findfont(FontProperties(family=["sans-serif"]))
|
|
assert Path(font).name == 'cmmi10.ttf'
|
|
|
|
# Smoketest get_charmap, which isn't used internally anymore
|
|
font = get_font(font)
|
|
cmap = font.get_charmap()
|
|
assert len(cmap) == 131
|
|
assert cmap[8729] == 30
|
|
|
|
|
|
def test_score_weight():
|
|
assert 0 == fontManager.score_weight("regular", "regular")
|
|
assert 0 == fontManager.score_weight("bold", "bold")
|
|
assert (0 < fontManager.score_weight(400, 400) <
|
|
fontManager.score_weight("normal", "bold"))
|
|
assert (0 < fontManager.score_weight("normal", "regular") <
|
|
fontManager.score_weight("normal", "bold"))
|
|
assert (fontManager.score_weight("normal", "regular") ==
|
|
fontManager.score_weight(400, 400))
|
|
|
|
|
|
def test_json_serialization(tmpdir):
|
|
# Can't open a NamedTemporaryFile twice on Windows, so use a temporary
|
|
# directory instead.
|
|
path = Path(tmpdir, "fontlist.json")
|
|
json_dump(fontManager, path)
|
|
copy = json_load(path)
|
|
with warnings.catch_warnings():
|
|
warnings.filterwarnings('ignore', 'findfont: Font family.*not found')
|
|
for prop in ({'family': 'STIXGeneral'},
|
|
{'family': 'Bitstream Vera Sans', 'weight': 700},
|
|
{'family': 'no such font family'}):
|
|
fp = FontProperties(**prop)
|
|
assert (fontManager.findfont(fp, rebuild_if_missing=False) ==
|
|
copy.findfont(fp, rebuild_if_missing=False))
|
|
|
|
|
|
def test_otf():
|
|
fname = '/usr/share/fonts/opentype/freefont/FreeMono.otf'
|
|
if Path(fname).exists():
|
|
assert is_opentype_cff_font(fname)
|
|
for f in fontManager.ttflist:
|
|
if 'otf' in f.fname:
|
|
with open(f.fname, 'rb') as fd:
|
|
res = fd.read(4) == b'OTTO'
|
|
assert res == is_opentype_cff_font(f.fname)
|
|
|
|
|
|
@pytest.mark.skipif(not has_fclist, reason='no fontconfig installed')
|
|
def test_get_fontconfig_fonts():
|
|
assert len(get_fontconfig_fonts()) > 1
|
|
|
|
|
|
@pytest.mark.parametrize('factor', [2, 4, 6, 8])
|
|
def test_hinting_factor(factor):
|
|
font = findfont(FontProperties(family=["sans-serif"]))
|
|
|
|
font1 = get_font(font, hinting_factor=1)
|
|
font1.clear()
|
|
font1.set_size(12, 100)
|
|
font1.set_text('abc')
|
|
expected = font1.get_width_height()
|
|
|
|
hinted_font = get_font(font, hinting_factor=factor)
|
|
hinted_font.clear()
|
|
hinted_font.set_size(12, 100)
|
|
hinted_font.set_text('abc')
|
|
# Check that hinting only changes text layout by a small (10%) amount.
|
|
np.testing.assert_allclose(hinted_font.get_width_height(), expected,
|
|
rtol=0.1)
|
|
|
|
|
|
@pytest.mark.skipif(sys.platform != "win32",
|
|
reason="Need Windows font to test against")
|
|
def test_utf16m_sfnt():
|
|
segoe_ui_semibold = None
|
|
for f in fontManager.ttflist:
|
|
# seguisbi = Microsoft Segoe UI Semibold
|
|
if f.fname[-12:] == "seguisbi.ttf":
|
|
segoe_ui_semibold = f
|
|
break
|
|
else:
|
|
pytest.xfail(reason="Couldn't find font to test against.")
|
|
|
|
# Check that we successfully read the "semibold" from the font's
|
|
# sfnt table and set its weight accordingly
|
|
assert segoe_ui_semibold.weight == "semibold"
|
|
|
|
|
|
@pytest.mark.xfail(not (os.environ.get("TRAVIS") and sys.platform == "linux"),
|
|
reason="Font may be missing.")
|
|
def test_find_ttc():
|
|
fp = FontProperties(family=["WenQuanYi Zen Hei"])
|
|
if Path(findfont(fp)).name != "wqy-zenhei.ttc":
|
|
# Travis appears to fail to pick up the ttc file sometimes. Try to
|
|
# rebuild the cache and try again.
|
|
fm._rebuild()
|
|
assert Path(findfont(fp)).name == "wqy-zenhei.ttc"
|
|
|
|
fig, ax = plt.subplots()
|
|
ax.text(.5, .5, "\N{KANGXI RADICAL DRAGON}", fontproperties=fp)
|
|
fig.savefig(BytesIO(), format="raw")
|
|
fig.savefig(BytesIO(), format="svg")
|
|
with pytest.raises(RuntimeError):
|
|
fig.savefig(BytesIO(), format="pdf")
|
|
with pytest.raises(RuntimeError):
|
|
fig.savefig(BytesIO(), format="ps")
|
|
|
|
|
|
@pytest.mark.skipif(sys.platform != 'linux', reason='Linux only')
|
|
def test_user_fonts_linux(tmpdir, monkeypatch):
|
|
font_test_file = 'mpltest.ttf'
|
|
|
|
# Precondition: the test font should not be available
|
|
fonts = findSystemFonts()
|
|
if any(font_test_file in font for font in fonts):
|
|
pytest.skip(f'{font_test_file} already exists in system fonts')
|
|
|
|
# Prepare a temporary user font directory
|
|
user_fonts_dir = tmpdir.join('fonts')
|
|
user_fonts_dir.ensure(dir=True)
|
|
shutil.copyfile(Path(__file__).parent / font_test_file,
|
|
user_fonts_dir.join(font_test_file))
|
|
|
|
with monkeypatch.context() as m:
|
|
m.setenv('XDG_DATA_HOME', str(tmpdir))
|
|
_call_fc_list.cache_clear()
|
|
# Now, the font should be available
|
|
fonts = findSystemFonts()
|
|
assert any(font_test_file in font for font in fonts)
|
|
|
|
# Make sure the temporary directory is no longer cached.
|
|
_call_fc_list.cache_clear()
|
|
|
|
|
|
@pytest.mark.skipif(sys.platform != 'win32', reason='Windows only')
|
|
def test_user_fonts_win32():
|
|
if not (os.environ.get('APPVEYOR', False) or
|
|
os.environ.get('TF_BUILD', False)):
|
|
pytest.xfail("This test should only run on CI (appveyor or azure) "
|
|
"as the developer's font directory should remain "
|
|
"unchanged.")
|
|
|
|
font_test_file = 'mpltest.ttf'
|
|
|
|
# Precondition: the test font should not be available
|
|
fonts = findSystemFonts()
|
|
if any(font_test_file in font for font in fonts):
|
|
pytest.skip(f'{font_test_file} already exists in system fonts')
|
|
|
|
user_fonts_dir = MSUserFontDirectories[0]
|
|
|
|
# Make sure that the user font directory exists (this is probably not the
|
|
# case on Windows versions < 1809)
|
|
os.makedirs(user_fonts_dir)
|
|
|
|
# Copy the test font to the user font directory
|
|
shutil.copyfile(os.path.join(os.path.dirname(__file__), font_test_file),
|
|
os.path.join(user_fonts_dir, font_test_file))
|
|
|
|
# Now, the font should be available
|
|
fonts = findSystemFonts()
|
|
assert any(font_test_file in font for font in fonts)
|
|
|
|
|
|
def _model_handler(_):
|
|
fig, ax = plt.subplots()
|
|
fig.savefig(BytesIO(), format="pdf")
|
|
plt.close()
|
|
|
|
|
|
@pytest.mark.skipif(not hasattr(os, "register_at_fork"),
|
|
reason="Cannot register at_fork handlers")
|
|
def test_fork():
|
|
_model_handler(0) # Make sure the font cache is filled.
|
|
ctx = multiprocessing.get_context("fork")
|
|
with ctx.Pool(processes=2) as pool:
|
|
pool.map(_model_handler, range(2))
|