def test_infix_to_postfix():
from pykern.pkcollections import PKDict
from pykern.pkunit import pkeq
from sirepo.template.code_variable import CodeVar, PurePythonEval
code_var = CodeVar([], PurePythonEval(PKDict()))
pkeq(code_var.infix_to_postfix('x + y * 2'), 'x y 2 * +')
pkeq(code_var.infix_to_postfix('-(x)'), 'x chs')
pkeq(code_var.infix_to_postfix('-(x + +x)'), 'x x + chs')
def test_eval():
from pykern.pkcollections import PKDict
from pykern.pkunit import pkeq
from sirepo.template.code_variable import CodeVar, PurePythonEval
code_var = CodeVar(
[
PKDict(
name='gamma',
value='(bend_energy * 1e-3 + EMASS) / EMASS',
),
PKDict(
name='bend_energy',
value=6.50762633,
),
],
PurePythonEval(PKDict(
CLIGHT=299792458,
EMASS=0.00051099892,
)),
)
pkeq(0.9973461123895662, code_var.eval_var('sqrt(1 - (1/pow(gamma, 2)))')[0])
pkeq('unknown token: abc', code_var.eval_var('abc + 123')[1])
pkeq('division by zero', code_var.eval_var('100 / 0')[1])
pkeq('1 1 gamma 2 pow / - sqrt', code_var.infix_to_postfix('sqrt(1 - (1/pow(gamma, 2)))'))
pkeq(True, code_var.is_var_value('abc'))
pkeq(False, code_var.is_var_value('-1.234e-6'))
pkeq(False, code_var.is_var_value('0'))
def _fix_value(value, names):
expr = CodeVar.infix_to_postfix(value.lower())
for v in expr.split(' '):
m = re.match(r'^(.*?)\-\>(.*)', v)
if m:
v = re.sub(r'\-\>', '.', v)
names[v] = [m.group(1), m.group(2)]
return re.sub(r'\-\>', '.', value)