561 lines
24 KiB
Python
561 lines
24 KiB
Python
from __future__ import annotations
|
|
|
|
import io
|
|
from pathlib import Path
|
|
import platform
|
|
import re
|
|
import shlex
|
|
from xml.etree import ElementTree as ET
|
|
from typing import Any
|
|
|
|
import numpy as np
|
|
from packaging.version import parse as parse_version
|
|
import pyparsing
|
|
import pytest
|
|
|
|
|
|
import matplotlib as mpl
|
|
from matplotlib.testing.decorators import check_figures_equal, image_comparison
|
|
import matplotlib.pyplot as plt
|
|
from matplotlib import mathtext, _mathtext
|
|
|
|
pyparsing_version = parse_version(pyparsing.__version__)
|
|
|
|
|
|
# If test is removed, use None as placeholder
|
|
math_tests = [
|
|
r'$a+b+\dot s+\dot{s}+\ldots$',
|
|
r'$x\hspace{-0.2}\doteq\hspace{-0.2}y$',
|
|
r'\$100.00 $\alpha \_$',
|
|
r'$\frac{\$100.00}{y}$',
|
|
r'$x y$',
|
|
r'$x+y\ x=y\ x<y\ x:y\ x,y\ x@y$',
|
|
r'$100\%y\ x*y\ x/y x\$y$',
|
|
r'$x\leftarrow y\ x\forall y\ x-y$',
|
|
r'$x \sf x \bf x {\cal X} \rm x$',
|
|
r'$x\ x\,x\;x\quad x\qquad x\!x\hspace{ 0.5 }y$',
|
|
r'$\{ \rm braces \}$',
|
|
r'$\left[\left\lfloor\frac{5}{\frac{\left(3\right)}{4}} y\right)\right]$',
|
|
r'$\left(x\right)$',
|
|
r'$\sin(x)$',
|
|
r'$x_2$',
|
|
r'$x^2$',
|
|
r'$x^2_y$',
|
|
r'$x_y^2$',
|
|
(r'$\sum _{\genfrac{}{}{0}{}{0\leq i\leq m}{0<j<n}}f\left(i,j\right)'
|
|
r'\mathcal{R}\prod_{i=\alpha_{i+1}}^\infty a_i \sin(2 \pi f x_i)'
|
|
r"\sqrt[2]{\prod^\frac{x}{2\pi^2}_\infty}$"),
|
|
r'$x = \frac{x+\frac{5}{2}}{\frac{y+3}{8}}$',
|
|
r'$dz/dt = \gamma x^2 + {\rm sin}(2\pi y+\phi)$',
|
|
r'Foo: $\alpha_{i+1}^j = {\rm sin}(2\pi f_j t_i) e^{-5 t_i/\tau}$',
|
|
None,
|
|
r'Variable $i$ is good',
|
|
r'$\Delta_i^j$',
|
|
r'$\Delta^j_{i+1}$',
|
|
r'$\ddot{o}\acute{e}\grave{e}\hat{O}\breve{\imath}\tilde{n}\vec{q}$',
|
|
r"$\arccos((x^i))$",
|
|
r"$\gamma = \frac{x=\frac{6}{8}}{y} \delta$",
|
|
r'$\limsup_{x\to\infty}$',
|
|
None,
|
|
r"$f'\quad f'''(x)\quad ''/\mathrm{yr}$",
|
|
r'$\frac{x_2888}{y}$',
|
|
r"$\sqrt[3]{\frac{X_2}{Y}}=5$",
|
|
None,
|
|
r"$\sqrt[3]{x}=5$",
|
|
r'$\frac{X}{\frac{X}{Y}}$',
|
|
r"$W^{3\beta}_{\delta_1 \rho_1 \sigma_2} = U^{3\beta}_{\delta_1 \rho_1} + \frac{1}{8 \pi 2} \int^{\alpha_2}_{\alpha_2} d \alpha^\prime_2 \left[\frac{ U^{2\beta}_{\delta_1 \rho_1} - \alpha^\prime_2U^{1\beta}_{\rho_1 \sigma_2} }{U^{0\beta}_{\rho_1 \sigma_2}}\right]$",
|
|
r'$\mathcal{H} = \int d \tau \left(\epsilon E^2 + \mu H^2\right)$',
|
|
r'$\widehat{abc}\widetilde{def}$',
|
|
'$\\Gamma \\Delta \\Theta \\Lambda \\Xi \\Pi \\Sigma \\Upsilon \\Phi \\Psi \\Omega$',
|
|
'$\\alpha \\beta \\gamma \\delta \\epsilon \\zeta \\eta \\theta \\iota \\lambda \\mu \\nu \\xi \\pi \\kappa \\rho \\sigma \\tau \\upsilon \\phi \\chi \\psi$',
|
|
|
|
# The following examples are from the MathML torture test here:
|
|
# https://www-archive.mozilla.org/projects/mathml/demo/texvsmml.xhtml
|
|
r'${x}^{2}{y}^{2}$',
|
|
r'${}_{2}F_{3}$',
|
|
r'$\frac{x+{y}^{2}}{k+1}$',
|
|
r'$x+{y}^{\frac{2}{k+1}}$',
|
|
r'$\frac{a}{b/2}$',
|
|
r'${a}_{0}+\frac{1}{{a}_{1}+\frac{1}{{a}_{2}+\frac{1}{{a}_{3}+\frac{1}{{a}_{4}}}}}$',
|
|
r'${a}_{0}+\frac{1}{{a}_{1}+\frac{1}{{a}_{2}+\frac{1}{{a}_{3}+\frac{1}{{a}_{4}}}}}$',
|
|
r'$\binom{n}{k/2}$',
|
|
r'$\binom{p}{2}{x}^{2}{y}^{p-2}-\frac{1}{1-x}\frac{1}{1-{x}^{2}}$',
|
|
r'${x}^{2y}$',
|
|
r'$\sum _{i=1}^{p}\sum _{j=1}^{q}\sum _{k=1}^{r}{a}_{ij}{b}_{jk}{c}_{ki}$',
|
|
r'$\sqrt{1+\sqrt{1+\sqrt{1+\sqrt{1+\sqrt{1+\sqrt{1+\sqrt{1+x}}}}}}}$',
|
|
r'$\left(\frac{{\partial }^{2}}{\partial {x}^{2}}+\frac{{\partial }^{2}}{\partial {y}^{2}}\right){|\varphi \left(x+iy\right)|}^{2}=0$',
|
|
r'${2}^{{2}^{{2}^{x}}}$',
|
|
r'${\int }_{1}^{x}\frac{\mathrm{dt}}{t}$',
|
|
r'$\int {\int }_{D}\mathrm{dx} \mathrm{dy}$',
|
|
# mathtex doesn't support array
|
|
# 'mmltt18' : r'$f\left(x\right)=\left\{\begin{array}{cc}\hfill 1/3\hfill & \text{if_}0\le x\le 1;\hfill \\ \hfill 2/3\hfill & \hfill \text{if_}3\le x\le 4;\hfill \\ \hfill 0\hfill & \text{elsewhere.}\hfill \end{array}$',
|
|
# mathtex doesn't support stackrel
|
|
# 'mmltt19' : r'$\stackrel{\stackrel{k\text{times}}{\ufe37}}{x+...+x}$',
|
|
r'${y}_{{x}^{2}}$',
|
|
# mathtex doesn't support the "\text" command
|
|
# 'mmltt21' : r'$\sum _{p\text{\prime}}f\left(p\right)={\int }_{t>1}f\left(t\right) d\pi \left(t\right)$',
|
|
# mathtex doesn't support array
|
|
# 'mmltt23' : r'$\left(\begin{array}{cc}\hfill \left(\begin{array}{cc}\hfill a\hfill & \hfill b\hfill \\ \hfill c\hfill & \hfill d\hfill \end{array}\right)\hfill & \hfill \left(\begin{array}{cc}\hfill e\hfill & \hfill f\hfill \\ \hfill g\hfill & \hfill h\hfill \end{array}\right)\hfill \\ \hfill 0\hfill & \hfill \left(\begin{array}{cc}\hfill i\hfill & \hfill j\hfill \\ \hfill k\hfill & \hfill l\hfill \end{array}\right)\hfill \end{array}\right)$',
|
|
# mathtex doesn't support array
|
|
# 'mmltt24' : r'$det|\begin{array}{ccccc}\hfill {c}_{0}\hfill & \hfill {c}_{1}\hfill & \hfill {c}_{2}\hfill & \hfill \dots \hfill & \hfill {c}_{n}\hfill \\ \hfill {c}_{1}\hfill & \hfill {c}_{2}\hfill & \hfill {c}_{3}\hfill & \hfill \dots \hfill & \hfill {c}_{n+1}\hfill \\ \hfill {c}_{2}\hfill & \hfill {c}_{3}\hfill & \hfill {c}_{4}\hfill & \hfill \dots \hfill & \hfill {c}_{n+2}\hfill \\ \hfill \u22ee\hfill & \hfill \u22ee\hfill & \hfill \u22ee\hfill & \hfill \hfill & \hfill \u22ee\hfill \\ \hfill {c}_{n}\hfill & \hfill {c}_{n+1}\hfill & \hfill {c}_{n+2}\hfill & \hfill \dots \hfill & \hfill {c}_{2n}\hfill \end{array}|>0$',
|
|
r'${y}_{{x}_{2}}$',
|
|
r'${x}_{92}^{31415}+\pi $',
|
|
r'${x}_{{y}_{b}^{a}}^{{z}_{c}^{d}}$',
|
|
r'${y}_{3}^{\prime \prime \prime }$',
|
|
# End of the MathML torture tests.
|
|
|
|
r"$\left( \xi \left( 1 - \xi \right) \right)$", # Bug 2969451
|
|
r"$\left(2 \, a=b\right)$", # Sage bug #8125
|
|
r"$? ! &$", # github issue #466
|
|
None,
|
|
None,
|
|
r"$\left\Vert \frac{a}{b} \right\Vert \left\vert \frac{a}{b} \right\vert \left\| \frac{a}{b}\right\| \left| \frac{a}{b} \right| \Vert a \Vert \vert b \vert \| a \| | b |$",
|
|
r'$\mathring{A} \AA$',
|
|
r'$M \, M \thinspace M \/ M \> M \: M \; M \ M \enspace M \quad M \qquad M \! M$',
|
|
r'$\Cap$ $\Cup$ $\leftharpoonup$ $\barwedge$ $\rightharpoonup$',
|
|
r'$\hspace{-0.2}\dotplus\hspace{-0.2}$ $\hspace{-0.2}\doteq\hspace{-0.2}$ $\hspace{-0.2}\doteqdot\hspace{-0.2}$ $\ddots$',
|
|
r'$xyz^kx_kx^py^{p-2} d_i^jb_jc_kd x^j_i E^0 E^0_u$', # github issue #4873
|
|
r'${xyz}^k{x}_{k}{x}^{p}{y}^{p-2} {d}_{i}^{j}{b}_{j}{c}_{k}{d} {x}^{j}_{i}{E}^{0}{E}^0_u$',
|
|
r'${\int}_x^x x\oint_x^x x\int_{X}^{X}x\int_x x \int^x x \int_{x} x\int^{x}{\int}_{x} x{\int}^{x}_{x}x$',
|
|
r'testing$^{123}$',
|
|
None,
|
|
r'$6-2$; $-2$; $ -2$; ${-2}$; ${ -2}$; $20^{+3}_{-2}$',
|
|
r'$\overline{\omega}^x \frac{1}{2}_0^x$', # github issue #5444
|
|
r'$,$ $.$ $1{,}234{, }567{ , }890$ and $1,234,567,890$', # github issue 5799
|
|
r'$\left(X\right)_{a}^{b}$', # github issue 7615
|
|
r'$\dfrac{\$100.00}{y}$', # github issue #1888
|
|
]
|
|
# 'svgastext' tests switch svg output to embed text as text (rather than as
|
|
# paths).
|
|
svgastext_math_tests = [
|
|
r'$-$-',
|
|
]
|
|
# 'lightweight' tests test only a single fontset (dejavusans, which is the
|
|
# default) and only png outputs, in order to minimize the size of baseline
|
|
# images.
|
|
lightweight_math_tests = [
|
|
r'$\sqrt[ab]{123}$', # github issue #8665
|
|
r'$x \overset{f}{\rightarrow} \overset{f}{x} \underset{xx}{ff} \overset{xx}{ff} \underset{f}{x} \underset{f}{\leftarrow} x$', # github issue #18241
|
|
r'$\sum x\quad\sum^nx\quad\sum_nx\quad\sum_n^nx\quad\prod x\quad\prod^nx\quad\prod_nx\quad\prod_n^nx$', # GitHub issue 18085
|
|
r'$1.$ $2.$ $19680801.$ $a.$ $b.$ $mpl.$',
|
|
r'$\text{text}_{\text{sub}}^{\text{sup}} + \text{\$foo\$} + \frac{\text{num}}{\mathbf{\text{den}}}\text{with space, curly brackets \{\}, and dash -}$',
|
|
r'$\boldsymbol{abcde} \boldsymbol{+} \boldsymbol{\Gamma + \Omega} \boldsymbol{01234} \boldsymbol{\alpha * \beta}$',
|
|
r'$\left\lbrace\frac{\left\lbrack A^b_c\right\rbrace}{\left\leftbrace D^e_f \right\rbrack}\right\rightbrace\ \left\leftparen\max_{x} \left\lgroup \frac{A}{B}\right\rgroup \right\rightparen$',
|
|
r'$\left( a\middle. b \right)$ $\left( \frac{a}{b} \middle\vert x_i \in P^S \right)$ $\left[ 1 - \middle| a\middle| + \left( x - \left\lfloor \dfrac{a}{b}\right\rfloor \right) \right]$',
|
|
r'$\sum_{\substack{k = 1\\ k \neq \lfloor n/2\rfloor}}^{n}P(i,j) \sum_{\substack{i \neq 0\\ -1 \leq i \leq 3\\ 1 \leq j \leq 5}} F^i(x,y) \sum_{\substack{\left \lfloor \frac{n}{2} \right\rfloor}} F(n)$',
|
|
]
|
|
|
|
digits = "0123456789"
|
|
uppercase = "ABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
|
lowercase = "abcdefghijklmnopqrstuvwxyz"
|
|
uppergreek = ("\\Gamma \\Delta \\Theta \\Lambda \\Xi \\Pi \\Sigma \\Upsilon \\Phi \\Psi "
|
|
"\\Omega")
|
|
lowergreek = ("\\alpha \\beta \\gamma \\delta \\epsilon \\zeta \\eta \\theta \\iota "
|
|
"\\lambda \\mu \\nu \\xi \\pi \\kappa \\rho \\sigma \\tau \\upsilon "
|
|
"\\phi \\chi \\psi")
|
|
all = [digits, uppercase, lowercase, uppergreek, lowergreek]
|
|
|
|
# Use stubs to reserve space if tests are removed
|
|
# stub should be of the form (None, N) where N is the number of strings that
|
|
# used to be tested
|
|
# Add new tests at the end.
|
|
font_test_specs: list[tuple[None | list[str], Any]] = [
|
|
([], all),
|
|
(['mathrm'], all),
|
|
(['mathbf'], all),
|
|
(['mathit'], all),
|
|
(['mathtt'], [digits, uppercase, lowercase]),
|
|
(None, 3),
|
|
(None, 3),
|
|
(None, 3),
|
|
(['mathbb'], [digits, uppercase, lowercase,
|
|
r'\Gamma \Pi \Sigma \gamma \pi']),
|
|
(['mathrm', 'mathbb'], [digits, uppercase, lowercase,
|
|
r'\Gamma \Pi \Sigma \gamma \pi']),
|
|
(['mathbf', 'mathbb'], [digits, uppercase, lowercase,
|
|
r'\Gamma \Pi \Sigma \gamma \pi']),
|
|
(['mathcal'], [uppercase]),
|
|
(['mathfrak'], [uppercase, lowercase]),
|
|
(['mathbf', 'mathfrak'], [uppercase, lowercase]),
|
|
(['mathscr'], [uppercase, lowercase]),
|
|
(['mathsf'], [digits, uppercase, lowercase]),
|
|
(['mathrm', 'mathsf'], [digits, uppercase, lowercase]),
|
|
(['mathbf', 'mathsf'], [digits, uppercase, lowercase]),
|
|
(['mathbfit'], all),
|
|
]
|
|
|
|
font_tests: list[None | str] = []
|
|
for fonts, chars in font_test_specs:
|
|
if fonts is None:
|
|
font_tests.extend([None] * chars)
|
|
else:
|
|
wrapper = ''.join([
|
|
' '.join(fonts),
|
|
' $',
|
|
*(r'\%s{' % font for font in fonts),
|
|
'%s',
|
|
*('}' for font in fonts),
|
|
'$',
|
|
])
|
|
for set in chars:
|
|
font_tests.append(wrapper % set)
|
|
|
|
|
|
@pytest.fixture
|
|
def baseline_images(request, fontset, index, text):
|
|
if text is None:
|
|
pytest.skip("test has been removed")
|
|
return ['%s_%s_%02d' % (request.param, fontset, index)]
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
'index, text', enumerate(math_tests), ids=range(len(math_tests)))
|
|
@pytest.mark.parametrize(
|
|
'fontset', ['cm', 'stix', 'stixsans', 'dejavusans', 'dejavuserif'])
|
|
@pytest.mark.parametrize('baseline_images', ['mathtext'], indirect=True)
|
|
@image_comparison(baseline_images=None,
|
|
tol=0.011 if platform.machine() in ('ppc64le', 's390x') else 0)
|
|
def test_mathtext_rendering(baseline_images, fontset, index, text):
|
|
mpl.rcParams['mathtext.fontset'] = fontset
|
|
fig = plt.figure(figsize=(5.25, 0.75))
|
|
fig.text(0.5, 0.5, text,
|
|
horizontalalignment='center', verticalalignment='center')
|
|
|
|
|
|
@pytest.mark.parametrize('index, text', enumerate(svgastext_math_tests),
|
|
ids=range(len(svgastext_math_tests)))
|
|
@pytest.mark.parametrize('fontset', ['cm', 'dejavusans'])
|
|
@pytest.mark.parametrize('baseline_images', ['mathtext0'], indirect=True)
|
|
@image_comparison(
|
|
baseline_images=None, extensions=['svg'],
|
|
savefig_kwarg={'metadata': { # Minimize image size.
|
|
'Creator': None, 'Date': None, 'Format': None, 'Type': None}})
|
|
def test_mathtext_rendering_svgastext(baseline_images, fontset, index, text):
|
|
mpl.rcParams['mathtext.fontset'] = fontset
|
|
mpl.rcParams['svg.fonttype'] = 'none' # Minimize image size.
|
|
fig = plt.figure(figsize=(5.25, 0.75))
|
|
fig.patch.set(visible=False) # Minimize image size.
|
|
fig.text(0.5, 0.5, text,
|
|
horizontalalignment='center', verticalalignment='center')
|
|
|
|
|
|
@pytest.mark.parametrize('index, text', enumerate(lightweight_math_tests),
|
|
ids=range(len(lightweight_math_tests)))
|
|
@pytest.mark.parametrize('fontset', ['dejavusans'])
|
|
@pytest.mark.parametrize('baseline_images', ['mathtext1'], indirect=True)
|
|
@image_comparison(baseline_images=None, extensions=['png'])
|
|
def test_mathtext_rendering_lightweight(baseline_images, fontset, index, text):
|
|
fig = plt.figure(figsize=(5.25, 0.75))
|
|
fig.text(0.5, 0.5, text, math_fontfamily=fontset,
|
|
horizontalalignment='center', verticalalignment='center')
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
'index, text', enumerate(font_tests), ids=range(len(font_tests)))
|
|
@pytest.mark.parametrize(
|
|
'fontset', ['cm', 'stix', 'stixsans', 'dejavusans', 'dejavuserif'])
|
|
@pytest.mark.parametrize('baseline_images', ['mathfont'], indirect=True)
|
|
@image_comparison(baseline_images=None, extensions=['png'],
|
|
tol=0.011 if platform.machine() in ('ppc64le', 's390x') else 0)
|
|
def test_mathfont_rendering(baseline_images, fontset, index, text):
|
|
mpl.rcParams['mathtext.fontset'] = fontset
|
|
fig = plt.figure(figsize=(5.25, 0.75))
|
|
fig.text(0.5, 0.5, text,
|
|
horizontalalignment='center', verticalalignment='center')
|
|
|
|
|
|
@check_figures_equal(extensions=["png"])
|
|
def test_short_long_accents(fig_test, fig_ref):
|
|
acc_map = _mathtext.Parser._accent_map
|
|
short_accs = [s for s in acc_map if len(s) == 1]
|
|
corresponding_long_accs = []
|
|
for s in short_accs:
|
|
l, = [l for l in acc_map if len(l) > 1 and acc_map[l] == acc_map[s]]
|
|
corresponding_long_accs.append(l)
|
|
fig_test.text(0, .5, "$" + "".join(rf"\{s}a" for s in short_accs) + "$")
|
|
fig_ref.text(
|
|
0, .5, "$" + "".join(fr"\{l} a" for l in corresponding_long_accs) + "$")
|
|
|
|
|
|
def test_fontinfo():
|
|
fontpath = mpl.font_manager.findfont("DejaVu Sans")
|
|
font = mpl.ft2font.FT2Font(fontpath)
|
|
table = font.get_sfnt_table("head")
|
|
assert table is not None
|
|
assert table['version'] == (1, 0)
|
|
|
|
|
|
# See gh-26152 for more context on this xfail
|
|
@pytest.mark.xfail(pyparsing_version.release == (3, 1, 0),
|
|
reason="Error messages are incorrect for this version")
|
|
@pytest.mark.parametrize(
|
|
'math, msg',
|
|
[
|
|
(r'$\hspace{}$', r'Expected \hspace{space}'),
|
|
(r'$\hspace{foo}$', r'Expected \hspace{space}'),
|
|
(r'$\sinx$', r'Unknown symbol: \sinx'),
|
|
(r'$\dotx$', r'Unknown symbol: \dotx'),
|
|
(r'$\frac$', r'Expected \frac{num}{den}'),
|
|
(r'$\frac{}{}$', r'Expected \frac{num}{den}'),
|
|
(r'$\binom$', r'Expected \binom{num}{den}'),
|
|
(r'$\binom{}{}$', r'Expected \binom{num}{den}'),
|
|
(r'$\genfrac$',
|
|
r'Expected \genfrac{ldelim}{rdelim}{rulesize}{style}{num}{den}'),
|
|
(r'$\genfrac{}{}{}{}{}{}$',
|
|
r'Expected \genfrac{ldelim}{rdelim}{rulesize}{style}{num}{den}'),
|
|
(r'$\sqrt$', r'Expected \sqrt{value}'),
|
|
(r'$\sqrt f$', r'Expected \sqrt{value}'),
|
|
(r'$\overline$', r'Expected \overline{body}'),
|
|
(r'$\overline{}$', r'Expected \overline{body}'),
|
|
(r'$\leftF$', r'Expected a delimiter'),
|
|
(r'$\rightF$', r'Unknown symbol: \rightF'),
|
|
(r'$\left(\right$', r'Expected a delimiter'),
|
|
# PyParsing 2 uses double quotes, PyParsing 3 uses single quotes and an
|
|
# extra backslash.
|
|
(r'$\left($', re.compile(r'Expected ("|\'\\)\\right["\']')),
|
|
(r'$\dfrac$', r'Expected \dfrac{num}{den}'),
|
|
(r'$\dfrac{}{}$', r'Expected \dfrac{num}{den}'),
|
|
(r'$\overset$', r'Expected \overset{annotation}{body}'),
|
|
(r'$\underset$', r'Expected \underset{annotation}{body}'),
|
|
(r'$\foo$', r'Unknown symbol: \foo'),
|
|
(r'$a^2^2$', r'Double superscript'),
|
|
(r'$a_2_2$', r'Double subscript'),
|
|
(r'$a^2_a^2$', r'Double superscript'),
|
|
(r'$a = {b$', r"Expected '}'"),
|
|
],
|
|
ids=[
|
|
'hspace without value',
|
|
'hspace with invalid value',
|
|
'function without space',
|
|
'accent without space',
|
|
'frac without parameters',
|
|
'frac with empty parameters',
|
|
'binom without parameters',
|
|
'binom with empty parameters',
|
|
'genfrac without parameters',
|
|
'genfrac with empty parameters',
|
|
'sqrt without parameters',
|
|
'sqrt with invalid value',
|
|
'overline without parameters',
|
|
'overline with empty parameter',
|
|
'left with invalid delimiter',
|
|
'right with invalid delimiter',
|
|
'unclosed parentheses with sizing',
|
|
'unclosed parentheses without sizing',
|
|
'dfrac without parameters',
|
|
'dfrac with empty parameters',
|
|
'overset without parameters',
|
|
'underset without parameters',
|
|
'unknown symbol',
|
|
'double superscript',
|
|
'double subscript',
|
|
'super on sub without braces',
|
|
'unclosed group',
|
|
]
|
|
)
|
|
def test_mathtext_exceptions(math, msg):
|
|
parser = mathtext.MathTextParser('agg')
|
|
match = re.escape(msg) if isinstance(msg, str) else msg
|
|
with pytest.raises(ValueError, match=match):
|
|
parser.parse(math)
|
|
|
|
|
|
def test_get_unicode_index_exception():
|
|
with pytest.raises(ValueError):
|
|
_mathtext.get_unicode_index(r'\foo')
|
|
|
|
|
|
def test_single_minus_sign():
|
|
fig = plt.figure()
|
|
fig.text(0.5, 0.5, '$-$')
|
|
fig.canvas.draw()
|
|
t = np.asarray(fig.canvas.renderer.buffer_rgba())
|
|
assert (t != 0xff).any() # assert that canvas is not all white.
|
|
|
|
|
|
@check_figures_equal(extensions=["png"])
|
|
def test_spaces(fig_test, fig_ref):
|
|
fig_test.text(.5, .5, r"$1\,2\>3\ 4$")
|
|
fig_ref.text(.5, .5, r"$1\/2\:3~4$")
|
|
|
|
|
|
@check_figures_equal(extensions=["png"])
|
|
def test_operator_space(fig_test, fig_ref):
|
|
fig_test.text(0.1, 0.1, r"$\log 6$")
|
|
fig_test.text(0.1, 0.2, r"$\log(6)$")
|
|
fig_test.text(0.1, 0.3, r"$\arcsin 6$")
|
|
fig_test.text(0.1, 0.4, r"$\arcsin|6|$")
|
|
fig_test.text(0.1, 0.5, r"$\operatorname{op} 6$") # GitHub issue #553
|
|
fig_test.text(0.1, 0.6, r"$\operatorname{op}[6]$")
|
|
fig_test.text(0.1, 0.7, r"$\cos^2$")
|
|
fig_test.text(0.1, 0.8, r"$\log_2$")
|
|
fig_test.text(0.1, 0.9, r"$\sin^2 \cos$") # GitHub issue #17852
|
|
|
|
fig_ref.text(0.1, 0.1, r"$\mathrm{log\,}6$")
|
|
fig_ref.text(0.1, 0.2, r"$\mathrm{log}(6)$")
|
|
fig_ref.text(0.1, 0.3, r"$\mathrm{arcsin\,}6$")
|
|
fig_ref.text(0.1, 0.4, r"$\mathrm{arcsin}|6|$")
|
|
fig_ref.text(0.1, 0.5, r"$\mathrm{op\,}6$")
|
|
fig_ref.text(0.1, 0.6, r"$\mathrm{op}[6]$")
|
|
fig_ref.text(0.1, 0.7, r"$\mathrm{cos}^2$")
|
|
fig_ref.text(0.1, 0.8, r"$\mathrm{log}_2$")
|
|
fig_ref.text(0.1, 0.9, r"$\mathrm{sin}^2 \mathrm{\,cos}$")
|
|
|
|
|
|
@check_figures_equal(extensions=["png"])
|
|
def test_inverted_delimiters(fig_test, fig_ref):
|
|
fig_test.text(.5, .5, r"$\left)\right($", math_fontfamily="dejavusans")
|
|
fig_ref.text(.5, .5, r"$)($", math_fontfamily="dejavusans")
|
|
|
|
|
|
@check_figures_equal(extensions=["png"])
|
|
def test_genfrac_displaystyle(fig_test, fig_ref):
|
|
fig_test.text(0.1, 0.1, r"$\dfrac{2x}{3y}$")
|
|
|
|
thickness = _mathtext.TruetypeFonts.get_underline_thickness(
|
|
None, None, fontsize=mpl.rcParams["font.size"],
|
|
dpi=mpl.rcParams["savefig.dpi"])
|
|
fig_ref.text(0.1, 0.1, r"$\genfrac{}{}{%f}{0}{2x}{3y}$" % thickness)
|
|
|
|
|
|
def test_mathtext_fallback_valid():
|
|
for fallback in ['cm', 'stix', 'stixsans', 'None']:
|
|
mpl.rcParams['mathtext.fallback'] = fallback
|
|
|
|
|
|
def test_mathtext_fallback_invalid():
|
|
for fallback in ['abc', '']:
|
|
with pytest.raises(ValueError, match="not a valid fallback font name"):
|
|
mpl.rcParams['mathtext.fallback'] = fallback
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"fallback,fontlist",
|
|
[("cm", ['DejaVu Sans', 'mpltest', 'STIXGeneral', 'cmr10', 'STIXGeneral']),
|
|
("stix", ['DejaVu Sans', 'mpltest', 'STIXGeneral'])])
|
|
def test_mathtext_fallback(fallback, fontlist):
|
|
mpl.font_manager.fontManager.addfont(
|
|
str(Path(__file__).resolve().parent / 'mpltest.ttf'))
|
|
mpl.rcParams["svg.fonttype"] = 'none'
|
|
mpl.rcParams['mathtext.fontset'] = 'custom'
|
|
mpl.rcParams['mathtext.rm'] = 'mpltest'
|
|
mpl.rcParams['mathtext.it'] = 'mpltest:italic'
|
|
mpl.rcParams['mathtext.bf'] = 'mpltest:bold'
|
|
mpl.rcParams['mathtext.bfit'] = 'mpltest:italic:bold'
|
|
mpl.rcParams['mathtext.fallback'] = fallback
|
|
|
|
test_str = r'a$A\AA\breve\gimel$'
|
|
|
|
buff = io.BytesIO()
|
|
fig, ax = plt.subplots()
|
|
fig.text(.5, .5, test_str, fontsize=40, ha='center')
|
|
fig.savefig(buff, format="svg")
|
|
tspans = (ET.fromstring(buff.getvalue())
|
|
.findall(".//{http://www.w3.org/2000/svg}tspan[@style]"))
|
|
# Getting the last element of the style attrib is a close enough
|
|
# approximation for parsing the font property.
|
|
char_fonts = [shlex.split(tspan.attrib["style"])[-1] for tspan in tspans]
|
|
assert char_fonts == fontlist
|
|
mpl.font_manager.fontManager.ttflist.pop()
|
|
|
|
|
|
def test_math_to_image(tmpdir):
|
|
mathtext.math_to_image('$x^2$', str(tmpdir.join('example.png')))
|
|
mathtext.math_to_image('$x^2$', io.BytesIO())
|
|
mathtext.math_to_image('$x^2$', io.BytesIO(), color='Maroon')
|
|
|
|
|
|
@image_comparison(baseline_images=['math_fontfamily_image.png'],
|
|
savefig_kwarg={'dpi': 40})
|
|
def test_math_fontfamily():
|
|
fig = plt.figure(figsize=(10, 3))
|
|
fig.text(0.2, 0.7, r"$This\ text\ should\ have\ one\ font$",
|
|
size=24, math_fontfamily='dejavusans')
|
|
fig.text(0.2, 0.3, r"$This\ text\ should\ have\ another$",
|
|
size=24, math_fontfamily='stix')
|
|
|
|
|
|
def test_default_math_fontfamily():
|
|
mpl.rcParams['mathtext.fontset'] = 'cm'
|
|
test_str = r'abc$abc\alpha$'
|
|
fig, ax = plt.subplots()
|
|
|
|
text1 = fig.text(0.1, 0.1, test_str, font='Arial')
|
|
prop1 = text1.get_fontproperties()
|
|
assert prop1.get_math_fontfamily() == 'cm'
|
|
text2 = fig.text(0.2, 0.2, test_str, fontproperties='Arial')
|
|
prop2 = text2.get_fontproperties()
|
|
assert prop2.get_math_fontfamily() == 'cm'
|
|
|
|
fig.draw_without_rendering()
|
|
|
|
|
|
def test_argument_order():
|
|
mpl.rcParams['mathtext.fontset'] = 'cm'
|
|
test_str = r'abc$abc\alpha$'
|
|
fig, ax = plt.subplots()
|
|
|
|
text1 = fig.text(0.1, 0.1, test_str,
|
|
math_fontfamily='dejavusans', font='Arial')
|
|
prop1 = text1.get_fontproperties()
|
|
assert prop1.get_math_fontfamily() == 'dejavusans'
|
|
text2 = fig.text(0.2, 0.2, test_str,
|
|
math_fontfamily='dejavusans', fontproperties='Arial')
|
|
prop2 = text2.get_fontproperties()
|
|
assert prop2.get_math_fontfamily() == 'dejavusans'
|
|
text3 = fig.text(0.3, 0.3, test_str,
|
|
font='Arial', math_fontfamily='dejavusans')
|
|
prop3 = text3.get_fontproperties()
|
|
assert prop3.get_math_fontfamily() == 'dejavusans'
|
|
text4 = fig.text(0.4, 0.4, test_str,
|
|
fontproperties='Arial', math_fontfamily='dejavusans')
|
|
prop4 = text4.get_fontproperties()
|
|
assert prop4.get_math_fontfamily() == 'dejavusans'
|
|
|
|
fig.draw_without_rendering()
|
|
|
|
|
|
def test_mathtext_cmr10_minus_sign():
|
|
# cmr10 does not contain a minus sign and used to issue a warning
|
|
# RuntimeWarning: Glyph 8722 missing from current font.
|
|
mpl.rcParams['font.family'] = 'cmr10'
|
|
mpl.rcParams['axes.formatter.use_mathtext'] = True
|
|
fig, ax = plt.subplots()
|
|
ax.plot(range(-1, 1), range(-1, 1))
|
|
# draw to make sure we have no warnings
|
|
fig.canvas.draw()
|
|
|
|
|
|
def test_mathtext_operators():
|
|
test_str = r'''
|
|
\increment \smallin \notsmallowns
|
|
\smallowns \QED \rightangle
|
|
\smallintclockwise \smallvarointclockwise
|
|
\smallointctrcclockwise
|
|
\ratio \minuscolon \dotsminusdots
|
|
\sinewave \simneqq \nlesssim
|
|
\ngtrsim \nlessgtr \ngtrless
|
|
\cupleftarrow \oequal \rightassert
|
|
\rightModels \hermitmatrix \barvee
|
|
\measuredrightangle \varlrtriangle
|
|
\equalparallel \npreccurlyeq \nsucccurlyeq
|
|
\nsqsubseteq \nsqsupseteq \sqsubsetneq
|
|
\sqsupsetneq \disin \varisins
|
|
\isins \isindot \varisinobar
|
|
\isinobar \isinvb \isinE
|
|
\nisd \varnis \nis
|
|
\varniobar \niobar \bagmember
|
|
\triangle'''.split()
|
|
|
|
fig = plt.figure()
|
|
for x, i in enumerate(test_str):
|
|
fig.text(0.5, (x + 0.5)/len(test_str), r'${%s}$' % i)
|
|
|
|
fig.draw_without_rendering()
|
|
|
|
|
|
@check_figures_equal(extensions=["png"])
|
|
def test_boldsymbol(fig_test, fig_ref):
|
|
fig_test.text(0.1, 0.2, r"$\boldsymbol{\mathrm{abc0123\alpha}}$")
|
|
fig_ref.text(0.1, 0.2, r"$\mathrm{abc0123\alpha}$")
|