def test_Piecewise():
g = Piecewise((0, Or(x <= -1, x >= 1)), (1 - x, x > 0), (1 + x, True))
assert (mathematica_code(g) ==
'Piecewise[{{0, x >= 1 || x <= -1}, '
'{-x + 1, x > 0}, {x + 1, True}}]')
def test_Relational():
assert mathematica_code(Eq(x, y)) == 'x == y'
assert mathematica_code(Ne(x, y/(1 + y**2))) == 'x != (y/(y^2 + 1))'
assert mathematica_code(Le(0, x**2)) == '0 <= x^2'
assert mathematica_code(Gt(pi, 3, evaluate=False)) == 'Pi > 3'
def test_Booleans():
assert mathematica_code(true) == 'True'
assert mathematica_code(false) == 'False'
def test_Function():
assert mathematica_code(f(x, y, z)) == 'f[x, y, z]'
assert mathematica_code(sin(x) ** cos(x)) == 'Sin[x]^Cos[x]'
assert mathematica_code(sign(x)) == 'Sign[x]'
assert mathematica_code(atanh(x), user_functions={'atanh': 'ArcTanh'}) == 'ArcTanh[x]'
assert (mathematica_code(meijerg(((1, 1), (3, 4)), ((1,), ()), x)) ==
'MeijerG[{{1, 1}, {3, 4}}, {{1}, {}}, x]')
assert (mathematica_code(hyper((1, 2, 3), (3, 4), x)) ==
'HypergeometricPFQ[{1, 2, 3}, {3, 4}, x]')
assert mathematica_code(Min(x, y)) == 'Min[x, y]'
assert mathematica_code(Max(x, y)) == 'Max[x, y]'
assert mathematica_code(Max(x, 2)) == 'Max[2, x]' # issue sympy/sympy#15344
assert mathematica_code(binomial(x, y)) == 'Binomial[x, y]'
assert mathematica_code(log(x)) == 'Log[x]'
assert mathematica_code(tan(x)) == 'Tan[x]'
assert mathematica_code(cot(x)) == 'Cot[x]'
assert mathematica_code(asin(x)) == 'ArcSin[x]'
assert mathematica_code(acos(x)) == 'ArcCos[x]'
assert mathematica_code(atan(x)) == 'ArcTan[x]'
assert mathematica_code(acot(x)) == 'ArcCot[x]'
assert mathematica_code(sinh(x)) == 'Sinh[x]'
assert mathematica_code(cosh(x)) == 'Cosh[x]'
assert mathematica_code(tanh(x)) == 'Tanh[x]'
assert mathematica_code(coth(x)) == 'Coth[x]'
assert mathematica_code(asinh(x)) == 'ArcSinh[x]'
assert mathematica_code(acosh(x)) == 'ArcCosh[x]'
assert mathematica_code(atanh(x)) == 'ArcTanh[x]'
assert mathematica_code(acoth(x)) == 'ArcCoth[x]'
assert mathematica_code(sech(x)) == 'Sech[x]'
assert mathematica_code(csch(x)) == 'Csch[x]'
assert mathematica_code(erfc(x)) == 'Erfc[x]'
assert mathematica_code(conjugate(x)) == 'Conjugate[x]'
assert mathematica_code(re(x)) == 'Re[x]'
assert mathematica_code(im(x)) == 'Im[x]'
assert mathematica_code(polygamma(x, y)) == 'PolyGamma[x, y]'
assert mathematica_code(factorial(x)) == 'Factorial[x]'
assert mathematica_code(factorial2(x)) == 'Factorial2[x]'
assert mathematica_code(rf(x, y)) == 'Pochhammer[x, y]'
assert mathematica_code(gamma(x)) == 'Gamma[x]'
assert mathematica_code(zeta(x)) == 'Zeta[x]'
assert mathematica_code(Heaviside(x)) == 'UnitStep[x]'
assert mathematica_code(fibonacci(x)) == 'Fibonacci[x]'
assert mathematica_code(polylog(x, y)) == 'PolyLog[x, y]'
assert mathematica_code(loggamma(x)) == 'LogGamma[x]'
class MyFunc1(Function):
@classmethod
def eval(cls, x):
pass
class MyFunc2(Function):
@classmethod
def eval(cls, x, y):
pass
pytest.raises(ValueError,
lambda: mathematica_code(MyFunc1(x),
user_functions={'MyFunc1':
['Myfunc1']}))
assert mathematica_code(MyFunc1(x),
user_functions={'MyFunc1':
'Myfunc1'}) == 'Myfunc1[x]'
assert mathematica_code(MyFunc2(x, y),
user_functions={'MyFunc2':
[(lambda *x: False,
'Myfunc2')]}) == 'MyFunc2[x, y]'
def test_Lambda():
f1 = Lambda(x, x**2)
assert mathematica_code(f1) == 'Function[{x}, x^2]'
f2 = Lambda((x, y), x + 2*y)
assert mathematica_code(f2) == 'Function[{x, y}, x + 2*y]'
def test_symbols():
assert mathematica_code(x) == 'x'
d = Dummy('d')
assert mathematica_code(d) == f'd{d.dummy_index}'
def test_AlgebraicElement():
r = RootOf(x**7 + 3*x - 1, 3)
K = QQ.algebraic_field(r)
a = K([1, 2, 3, 0, 1])
assert mathematica_code(a) == ('AlgebraicNumber[Root[#^7 + 3*# - 1 &, 4],'
' {1, 0, 3, 2, 1}]')
def test_RootSum():
r = RootSum(x**3 + x + 3, Lambda(y, log(y*z)))
assert mathematica_code(r) == ('RootSum[Function[{x}, x^3 + x + 3], '
'Function[{y}, Log[y*z]]]')
def test_Integer():
assert mathematica_code(Integer(67)) == '67'
assert mathematica_code(Integer(-1)) == '-1'
def test_RootOf():
p = Poly(x**3 + y*x + 1, x)
assert mathematica_code(RootOf(p, 0)) == 'Root[#^3 + #*y + 1 &, 1]'