113 lines
3.0 KiB
Python
113 lines
3.0 KiB
Python
|
from sympy.core.symbol import symbols
|
||
|
|
from sympy.printing.codeprinter import ccode
|
||
|
|
from sympy.codegen.ast import Declaration, Variable, float64, int64, String, CodeBlock
|
||
|
|
from sympy.codegen.cnodes import (
|
||
|
|
alignof, CommaOperator, goto, Label, PreDecrement, PostDecrement, PreIncrement, PostIncrement,
|
||
|
|
sizeof, union, struct
|
||
|
|
)
|
||
|
|
|
||
|
|
x, y = symbols('x y')
|
||
|
|
|
||
|
|
|
||
|
|
def test_alignof():
|
||
|
|
ax = alignof(x)
|
||
|
|
assert ccode(ax) == 'alignof(x)'
|
||
|
|
assert ax.func(*ax.args) == ax
|
||
|
|
|
||
|
|
|
||
|
|
def test_CommaOperator():
|
||
|
|
expr = CommaOperator(PreIncrement(x), 2*x)
|
||
|
|
assert ccode(expr) == '(++(x), 2*x)'
|
||
|
|
assert expr.func(*expr.args) == expr
|
||
|
|
|
||
|
|
|
||
|
|
def test_goto_Label():
|
||
|
|
s = 'early_exit'
|
||
|
|
g = goto(s)
|
||
|
|
assert g.func(*g.args) == g
|
||
|
|
assert g != goto('foobar')
|
||
|
|
assert ccode(g) == 'goto early_exit'
|
||
|
|
|
||
|
|
l1 = Label(s)
|
||
|
|
assert ccode(l1) == 'early_exit:'
|
||
|
|
assert l1 == Label('early_exit')
|
||
|
|
assert l1 != Label('foobar')
|
||
|
|
|
||
|
|
body = [PreIncrement(x)]
|
||
|
|
l2 = Label(s, body)
|
||
|
|
assert l2.name == String("early_exit")
|
||
|
|
assert l2.body == CodeBlock(PreIncrement(x))
|
||
|
|
assert ccode(l2) == ("early_exit:\n"
|
||
|
|
"++(x);")
|
||
|
|
|
||
|
|
body = [PreIncrement(x), PreDecrement(y)]
|
||
|
|
l2 = Label(s, body)
|
||
|
|
assert l2.name == String("early_exit")
|
||
|
|
assert l2.body == CodeBlock(PreIncrement(x), PreDecrement(y))
|
||
|
|
assert ccode(l2) == ("early_exit:\n"
|
||
|
|
"{\n ++(x);\n --(y);\n}")
|
||
|
|
|
||
|
|
|
||
|
|
def test_PreDecrement():
|
||
|
|
p = PreDecrement(x)
|
||
|
|
assert p.func(*p.args) == p
|
||
|
|
assert ccode(p) == '--(x)'
|
||
|
|
|
||
|
|
|
||
|
|
def test_PostDecrement():
|
||
|
|
p = PostDecrement(x)
|
||
|
|
assert p.func(*p.args) == p
|
||
|
|
assert ccode(p) == '(x)--'
|
||
|
|
|
||
|
|
|
||
|
|
def test_PreIncrement():
|
||
|
|
p = PreIncrement(x)
|
||
|
|
assert p.func(*p.args) == p
|
||
|
|
assert ccode(p) == '++(x)'
|
||
|
|
|
||
|
|
|
||
|
|
def test_PostIncrement():
|
||
|
|
p = PostIncrement(x)
|
||
|
|
assert p.func(*p.args) == p
|
||
|
|
assert ccode(p) == '(x)++'
|
||
|
|
|
||
|
|
|
||
|
|
def test_sizeof():
|
||
|
|
typename = 'unsigned int'
|
||
|
|
sz = sizeof(typename)
|
||
|
|
assert ccode(sz) == 'sizeof(%s)' % typename
|
||
|
|
assert sz.func(*sz.args) == sz
|
||
|
|
assert not sz.is_Atom
|
||
|
|
assert sz.atoms() == {String('unsigned int'), String('sizeof')}
|
||
|
|
|
||
|
|
|
||
|
|
def test_struct():
|
||
|
|
vx, vy = Variable(x, type=float64), Variable(y, type=float64)
|
||
|
|
s = struct('vec2', [vx, vy])
|
||
|
|
assert s.func(*s.args) == s
|
||
|
|
assert s == struct('vec2', (vx, vy))
|
||
|
|
assert s != struct('vec2', (vy, vx))
|
||
|
|
assert str(s.name) == 'vec2'
|
||
|
|
assert len(s.declarations) == 2
|
||
|
|
assert all(isinstance(arg, Declaration) for arg in s.declarations)
|
||
|
|
assert ccode(s) == (
|
||
|
|
"struct vec2 {\n"
|
||
|
|
" double x;\n"
|
||
|
|
" double y;\n"
|
||
|
|
"}")
|
||
|
|
|
||
|
|
|
||
|
|
def test_union():
|
||
|
|
vx, vy = Variable(x, type=float64), Variable(y, type=int64)
|
||
|
|
u = union('dualuse', [vx, vy])
|
||
|
|
assert u.func(*u.args) == u
|
||
|
|
assert u == union('dualuse', (vx, vy))
|
||
|
|
assert str(u.name) == 'dualuse'
|
||
|
|
assert len(u.declarations) == 2
|
||
|
|
assert all(isinstance(arg, Declaration) for arg in u.declarations)
|
||
|
|
assert ccode(u) == (
|
||
|
|
"union dualuse {\n"
|
||
|
|
" double x;\n"
|
||
|
|
" int64_t y;\n"
|
||
|
|
"}")
|