def test_tostring_c(self):
language = Language.C
x = as_symbol('x')
y = as_symbol('y')
z = as_symbol('z')
n = as_number(123)
assert Expr(Op.FACTORS, {x: 2}).tostring(language=language) == 'x * x'
assert Expr(Op.FACTORS, {
x + y: 2
}).tostring(language=language) == '(x + y) * (x + y)'
assert Expr(Op.FACTORS, {
x: 12
}).tostring(language=language) == 'pow(x, 12)'
assert as_apply(ArithOp.DIV, x,
y).tostring(language=language) == 'x / y'
assert as_apply(ArithOp.DIV, x,
x + y).tostring(language=language) == 'x / (x + y)'
assert as_apply(ArithOp.DIV, x - y, x +
y).tostring(language=language) == '(x - y) / (x + y)'
assert (x + (x - y) / (x + y) +
n).tostring(language=language) == '123 + x + (x - y) / (x + y)'
assert as_ternary(x, y, z).tostring(language=language) == '(x ? y : z)'
assert as_eq(x, y).tostring(language=language) == 'x == y'
assert as_ne(x, y).tostring(language=language) == 'x != y'
assert as_lt(x, y).tostring(language=language) == 'x < y'
assert as_le(x, y).tostring(language=language) == 'x <= y'
assert as_gt(x, y).tostring(language=language) == 'x > y'
assert as_ge(x, y).tostring(language=language) == 'x >= y'
def test_tostring_c(self):
language = Language.C
x = as_symbol("x")
y = as_symbol("y")
z = as_symbol("z")
n = as_number(123)
assert Expr(Op.FACTORS, {x: 2}).tostring(language=language) == "x * x"
assert (Expr(Op.FACTORS, {
x + y: 2
}).tostring(language=language) == "(x + y) * (x + y)")
assert Expr(Op.FACTORS, {
x: 12
}).tostring(language=language) == "pow(x, 12)"
assert as_apply(ArithOp.DIV, x,
y).tostring(language=language) == "x / y"
assert (as_apply(ArithOp.DIV, x,
x + y).tostring(language=language) == "x / (x + y)")
assert (as_apply(ArithOp.DIV, x - y, x +
y).tostring(language=language) == "(x - y) / (x + y)")
assert (x + (x - y) / (x + y) +
n).tostring(language=language) == "123 + x + (x - y) / (x + y)"
assert as_ternary(x, y, z).tostring(language=language) == "(x?y:z)"
assert as_eq(x, y).tostring(language=language) == "x == y"
assert as_ne(x, y).tostring(language=language) == "x != y"
assert as_lt(x, y).tostring(language=language) == "x < y"
assert as_le(x, y).tostring(language=language) == "x <= y"
assert as_gt(x, y).tostring(language=language) == "x > y"
assert as_ge(x, y).tostring(language=language) == "x >= y"
def test_fromstring(self):
x = as_symbol('x')
y = as_symbol('y')
z = as_symbol('z')
f = as_symbol('f')
s = as_string('"ABC"')
t = as_string('"123"')
a = as_array((x, y))
assert fromstring('x') == x
assert fromstring('+ x') == x
assert fromstring('- x') == -x
assert fromstring('x + y') == x + y
assert fromstring('x + 1') == x + 1
assert fromstring('x * y') == x * y
assert fromstring('x * 2') == x * 2
assert fromstring('x / y') == x / y
assert fromstring('x ** 2', language=Language.Python) == x**2
assert fromstring('x ** 2 ** 3', language=Language.Python) == x**2**3
assert fromstring('(x + y) * z') == (x + y) * z
assert fromstring('f(x)') == f(x)
assert fromstring('f(x,y)') == f(x, y)
assert fromstring('f[x]') == f[x]
assert fromstring('f[x][y]') == f[x][y]
assert fromstring('"ABC"') == s
assert normalize(
fromstring('"ABC" // "123" ', language=Language.Fortran)) == s // t
assert fromstring('f("ABC")') == f(s)
assert fromstring('MYSTRKIND_"ABC"') == as_string('"ABC"', 'MYSTRKIND')
assert fromstring('(/x, y/)') == a, fromstring('(/x, y/)')
assert fromstring('f((/x, y/))') == f(a)
assert fromstring('(/(x+y)*z/)') == as_array(((x + y) * z, ))
assert fromstring('123') == as_number(123)
assert fromstring('123_2') == as_number(123, 2)
assert fromstring('123_myintkind') == as_number(123, 'myintkind')
assert fromstring('123.0') == as_number(123.0, 4)
assert fromstring('123.0_4') == as_number(123.0, 4)
assert fromstring('123.0_8') == as_number(123.0, 8)
assert fromstring('123.0e0') == as_number(123.0, 4)
assert fromstring('123.0d0') == as_number(123.0, 8)
assert fromstring('123d0') == as_number(123.0, 8)
assert fromstring('123e-0') == as_number(123.0, 4)
assert fromstring('123d+0') == as_number(123.0, 8)
assert fromstring('123.0_myrealkind') == as_number(123.0, 'myrealkind')
assert fromstring('3E4') == as_number(30000.0, 4)
assert fromstring('(1, 2)') == as_complex(1, 2)
assert fromstring('(1e2, PI)') == as_complex(as_number(100.0),
as_symbol('PI'))
assert fromstring('[1, 2]') == as_array((as_number(1), as_number(2)))
assert fromstring('POINT(x, y=1)') == as_apply(as_symbol('POINT'),
x,
y=as_number(1))
assert (fromstring('PERSON(name="John", age=50, shape=(/34, 23/))') ==
as_apply(as_symbol('PERSON'),
name=as_string('"John"'),
age=as_number(50),
shape=as_array((as_number(34), as_number(23)))))
assert fromstring('x?y:z') == as_ternary(x, y, z)
assert fromstring('*x') == as_deref(x)
assert fromstring('**x') == as_deref(as_deref(x))
assert fromstring('&x') == as_ref(x)
assert fromstring('(*x) * (*y)') == as_deref(x) * as_deref(y)
assert fromstring('(*x) * *y') == as_deref(x) * as_deref(y)
assert fromstring('*x * *y') == as_deref(x) * as_deref(y)
assert fromstring('*x**y') == as_deref(x) * as_deref(y)
assert fromstring('x == y') == as_eq(x, y)
assert fromstring('x != y') == as_ne(x, y)
assert fromstring('x < y') == as_lt(x, y)
assert fromstring('x > y') == as_gt(x, y)
assert fromstring('x <= y') == as_le(x, y)
assert fromstring('x >= y') == as_ge(x, y)
assert fromstring('x .eq. y', language=Language.Fortran) == as_eq(x, y)
assert fromstring('x .ne. y', language=Language.Fortran) == as_ne(x, y)
assert fromstring('x .lt. y', language=Language.Fortran) == as_lt(x, y)
assert fromstring('x .gt. y', language=Language.Fortran) == as_gt(x, y)
assert fromstring('x .le. y', language=Language.Fortran) == as_le(x, y)
assert fromstring('x .ge. y', language=Language.Fortran) == as_ge(x, y)
def test_operations(self):
x = as_symbol('x')
y = as_symbol('y')
z = as_symbol('z')
assert x + x == Expr(Op.TERMS, {x: 2})
assert x - x == Expr(Op.INTEGER, (0, 4))
assert x + y == Expr(Op.TERMS, {x: 1, y: 1})
assert x - y == Expr(Op.TERMS, {x: 1, y: -1})
assert x * x == Expr(Op.FACTORS, {x: 2})
assert x * y == Expr(Op.FACTORS, {x: 1, y: 1})
assert +x == x
assert -x == Expr(Op.TERMS, {x: -1}), repr(-x)
assert 2 * x == Expr(Op.TERMS, {x: 2})
assert 2 + x == Expr(Op.TERMS, {x: 1, as_number(1): 2})
assert 2 * x + 3 * y == Expr(Op.TERMS, {x: 2, y: 3})
assert (x + y) * 2 == Expr(Op.TERMS, {x: 2, y: 2})
assert x**2 == Expr(Op.FACTORS, {x: 2})
assert (x + y)**2 == Expr(
Op.TERMS, {
Expr(Op.FACTORS, {x: 2}): 1,
Expr(Op.FACTORS, {y: 2}): 1,
Expr(Op.FACTORS, {
x: 1,
y: 1
}): 2
})
assert (x + y) * x == x**2 + x * y
assert (x + y)**2 == x**2 + 2 * x * y + y**2
assert (x + y)**2 + (x - y)**2 == 2 * x**2 + 2 * y**2
assert (x + y) * z == x * z + y * z
assert z * (x + y) == x * z + y * z
assert (x / 2) == as_apply(ArithOp.DIV, x, as_number(2))
assert (2 * x / 2) == x
assert (3 * x / 2) == as_apply(ArithOp.DIV, 3 * x, as_number(2))
assert (4 * x / 2) == 2 * x
assert (5 * x / 2) == as_apply(ArithOp.DIV, 5 * x, as_number(2))
assert (6 * x / 2) == 3 * x
assert ((3 * 5) * x / 6) == as_apply(ArithOp.DIV, 5 * x, as_number(2))
assert (30 * x**2 * y**4 / (24 * x**3 * y**3)) == as_apply(
ArithOp.DIV, 5 * y, 4 * x)
assert ((15 * x / 6) / 5) == as_apply(ArithOp.DIV, x,
as_number(2)), ((15 * x / 6) / 5)
assert (x / (5 / x)) == as_apply(ArithOp.DIV, x**2, as_number(5))
assert (x / 2.0) == Expr(Op.TERMS, {x: 0.5})
s = as_string('"ABC"')
t = as_string('"123"')
assert s // t == Expr(Op.STRING, ('"ABC123"', 1))
assert s // x == Expr(Op.CONCAT, (s, x))
assert x // s == Expr(Op.CONCAT, (x, s))
c = as_complex(1., 2.)
assert -c == as_complex(-1., -2.)
assert c + c == as_expr((1 + 2j) * 2)
assert c * c == as_expr((1 + 2j)**2)
def test_fromstring(self):
x = as_symbol("x")
y = as_symbol("y")
z = as_symbol("z")
f = as_symbol("f")
s = as_string('"ABC"')
t = as_string('"123"')
a = as_array((x, y))
assert fromstring("x") == x
assert fromstring("+ x") == x
assert fromstring("- x") == -x
assert fromstring("x + y") == x + y
assert fromstring("x + 1") == x + 1
assert fromstring("x * y") == x * y
assert fromstring("x * 2") == x * 2
assert fromstring("x / y") == x / y
assert fromstring("x ** 2", language=Language.Python) == x**2
assert fromstring("x ** 2 ** 3", language=Language.Python) == x**2**3
assert fromstring("(x + y) * z") == (x + y) * z
assert fromstring("f(x)") == f(x)
assert fromstring("f(x,y)") == f(x, y)
assert fromstring("f[x]") == f[x]
assert fromstring("f[x][y]") == f[x][y]
assert fromstring('"ABC"') == s
assert (normalize(
fromstring('"ABC" // "123" ',
language=Language.Fortran)) == s // t)
assert fromstring('f("ABC")') == f(s)
assert fromstring('MYSTRKIND_"ABC"') == as_string('"ABC"', "MYSTRKIND")
assert fromstring("(/x, y/)") == a, fromstring("(/x, y/)")
assert fromstring("f((/x, y/))") == f(a)
assert fromstring("(/(x+y)*z/)") == as_array(((x + y) * z, ))
assert fromstring("123") == as_number(123)
assert fromstring("123_2") == as_number(123, 2)
assert fromstring("123_myintkind") == as_number(123, "myintkind")
assert fromstring("123.0") == as_number(123.0, 4)
assert fromstring("123.0_4") == as_number(123.0, 4)
assert fromstring("123.0_8") == as_number(123.0, 8)
assert fromstring("123.0e0") == as_number(123.0, 4)
assert fromstring("123.0d0") == as_number(123.0, 8)
assert fromstring("123d0") == as_number(123.0, 8)
assert fromstring("123e-0") == as_number(123.0, 4)
assert fromstring("123d+0") == as_number(123.0, 8)
assert fromstring("123.0_myrealkind") == as_number(123.0, "myrealkind")
assert fromstring("3E4") == as_number(30000.0, 4)
assert fromstring("(1, 2)") == as_complex(1, 2)
assert fromstring("(1e2, PI)") == as_complex(as_number(100.0),
as_symbol("PI"))
assert fromstring("[1, 2]") == as_array((as_number(1), as_number(2)))
assert fromstring("POINT(x, y=1)") == as_apply(as_symbol("POINT"),
x,
y=as_number(1))
assert fromstring(
'PERSON(name="John", age=50, shape=(/34, 23/))') == as_apply(
as_symbol("PERSON"),
name=as_string('"John"'),
age=as_number(50),
shape=as_array((as_number(34), as_number(23))),
)
assert fromstring("x?y:z") == as_ternary(x, y, z)
assert fromstring("*x") == as_deref(x)
assert fromstring("**x") == as_deref(as_deref(x))
assert fromstring("&x") == as_ref(x)
assert fromstring("(*x) * (*y)") == as_deref(x) * as_deref(y)
assert fromstring("(*x) * *y") == as_deref(x) * as_deref(y)
assert fromstring("*x * *y") == as_deref(x) * as_deref(y)
assert fromstring("*x**y") == as_deref(x) * as_deref(y)
assert fromstring("x == y") == as_eq(x, y)
assert fromstring("x != y") == as_ne(x, y)
assert fromstring("x < y") == as_lt(x, y)
assert fromstring("x > y") == as_gt(x, y)
assert fromstring("x <= y") == as_le(x, y)
assert fromstring("x >= y") == as_ge(x, y)
assert fromstring("x .eq. y", language=Language.Fortran) == as_eq(x, y)
assert fromstring("x .ne. y", language=Language.Fortran) == as_ne(x, y)
assert fromstring("x .lt. y", language=Language.Fortran) == as_lt(x, y)
assert fromstring("x .gt. y", language=Language.Fortran) == as_gt(x, y)
assert fromstring("x .le. y", language=Language.Fortran) == as_le(x, y)
assert fromstring("x .ge. y", language=Language.Fortran) == as_ge(x, y)