def test_stateful_eval_variable_name_sanitization():
assert stateful_eval("`data|a` / `data|b`", {
'data|a': 1,
'data|b': 2,
'data_b': 3
}, None, None, None) == 0.5
assert stateful_eval("`2data|a` / `2data|b`", {
'2data|a': 1,
'2data|b': 2,
'2data_b': 3
}, None, None, None) == 0.5
def test_stateful_eval():
state = {}
assert stateful_eval("dummy_transform(data)", {
'dummy_transform': dummy_transform,
'data': 1
}, None, state, None) == 1
assert state == {"dummy_transform(data)": {'data': 1}}
assert stateful_eval("dummy_transform(data)", {
'dummy_transform': dummy_transform,
'data': 2
}, None, state, None) == 1
def test_stateful_eval_distribution():
state = {}
assert stateful_eval(
"dummy_transform(data)", {
'dummy_transform': dummy_transform,
'data': {
'__property__': 'value',
'a': 1,
'b': 2
}
}, None, state, None) == {
'__property__': 'value',
'a': 1,
'b': 2
}
assert state == {
"dummy_transform(data)": {
'a': {
'data': 1
},
'b': {
'data': 2
}
}
}
assert stateful_eval(
"dummy_transform(data)", {
'dummy_transform': dummy_transform,
'data': {
'__property__': 'value2',
'a': 3,
'b': 4
}
}, None, state, None) == {
'__property__': 'value2',
'a': 1,
'b': 2
}
def _evaluate(self, expr, metadata, transform_state):
return stateful_eval(expr, self.layered_context, {expr: metadata}, transform_state, self.config)