sympy/printing/tests/test_latex.py

   1 import py
   2
   3 from sympy import *
   4 from sympy.printing.latex import latex
   5 from sympy.utilities.pytest import XFAIL
   6
   7 x,y = symbols('xy')
   8
   9 def test_latex_basic():
  10     assert latex(1+x) == "$1+x$"
  11     assert latex(x**2) == "${x}^{2}$"
  12     assert latex(x**(1+x)) == "${x}^{(1+x)}$"
  13
  14 def test_latex_symbols():
  15     Gamma, lmbda, rho = map(Symbol, ('Gamma', 'lambda', 'rho'))
  16     mass, volume = map(Symbol, ('mass', 'volume'))
  17     assert latex(Gamma + lmbda) == "$\Gamma+\lambda$"
  18     assert latex(Gamma * lmbda) == "$\Gamma \lambda$"
  19     assert latex(volume * rho == mass) == r"$\rho \cdot \mathrm{volume} = \mathrm{mass}$"
  20     assert latex(volume / mass * rho == 1) == r"$\rho \cdot \mathrm{volume} \cdot {\mathrm{mass}}^{(-1)} = 1$"
  21     assert latex(mass**3 * volume**3) == r"${\mathrm{mass}}^{3} \cdot {\mathrm{volume}}^{3}$"
  22
  23 @XFAIL
  24 def test_latex_functions():
  25     #assert latex(exp(x)) == "${e}^{x}$"
  26
  27     f = Function('f')
  28     beta = Function('beta')
  29     assert latex(f(x)) == r"$f\left(x\right)$"
  30     assert latex(beta(x)) == r"$\beta\left(x\right)$"
  31     #assert latex(sin(x)) == r"$\mathrm{sin}\left(x\right)$"
  32
  33     #assert latex(factorial(x, evaluate=False)) == "$x!$"
  34     #assert latex(factorial(-4, evaluate=False)) == "$(-4)!$"
  35     #assert latex(factorial(-x, evaluate=False)) == "$(- x)!$"
  36
  37 def test_latex_derivatives():
  38     assert latex(diff(x**3, x, evaluate=False)) == \
  39         r"$\frac{\partial}{\partial x} {x}^{3}$"
  40     #assert latex(diff(sin(x)+x**2, x, evaluate=False)) == \
  41     #    r"$\frac{\partial}{\partial x} \left({x}^{2}+\mathrm{sin}\left(x\right)\right)$"
  42
  43 def test_latex_integrals():
  44     #assert latex(Integral(log(x), x)) == r"$\int \mathrm{log}\left(x\right)\,dx$"
  45     assert latex(Integral(x**2, (x,0,1))) == r"$\int_{0}^{1} {x}^{2}\,dx$"
  46     assert latex(Integral(x**2, (x,10,20))) == r"$\int_{10}^{20} {x}^{2}\,dx$"
  47     assert latex(Integral(y*x**2, (x,0,1), y)) == r"$\int \int_{0}^{1} y {x}^{2}\,dx\,dy$"
  48
  49 #def test_latex_limits():
  50 #    assert latex(limit(x, x, oo, evaluate=False)) == r"$\lim_{x \to \infty}x$"