def test_case_insensitive():
# tests case insensitive and attribute like variables
from pykern.pkcollections import PKDict
from pykern.pkunit import pkeq
from sirepo.template.code_variable import CodeVar, PurePythonEval
code_var = CodeVar(
[
PKDict(
name='x.x7.x',
value='123',
),
PKDict(
name='Y',
value='x.X7.x + x.x7.x',
),
],
PurePythonEval(),
case_insensitive=True,
)
pkeq(code_var.compute_cache(
_empty_data(),
PKDict(),
), PKDict({
'x.x7.x': 123,
'y': 246,
'x.x7.x + x.x7.x': 246,
}))
pkeq(
code_var.get_expr_dependencies('Y y +'),
['x.x7.x', 'y'],
)
pkeq(
code_var.validate_var_delete(
'X.X7.X',
_empty_data(),
PKDict(),
),
'"X.X7.X" is in use in variable(s): y',
)
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'))
pkeq(
code_var.validate_var_delete(
'bend_energy',
_empty_data(),
PKDict(),
),
'"bend_energy" is in use in variable(s): gamma',
)