mirror of https://github.com/inclusionAI/AReaL
23 lines
448 B
Python
23 lines
448 B
Python
import pytest
|
|
from sympy import I, Integer, Sum, Symbol
|
|
|
|
from .context import assert_equal
|
|
|
|
a = Symbol("a", real=True)
|
|
b = Symbol("b", real=True)
|
|
i = Symbol("i", real=True)
|
|
n = Symbol("n", real=True)
|
|
x = Symbol("x", real=True)
|
|
|
|
|
|
def test_complex():
|
|
assert_equal("a+Ib", a + I * b)
|
|
|
|
|
|
def test_complex_e():
|
|
assert_equal("e^{I\\pi}", Integer(-1))
|
|
|
|
|
|
def test_complex_sum():
|
|
assert_equal("\\sum_{i=0}^{n} i \\cdot x", Sum(i * x, (i, 0, n)))
|