def test_sub_for_comps(self):
cases = [
('not x < 3 and True', {
'x < 3': 'True'
}, False),
('not x < 3 and x < 3', {
'x < 3': 'True'
}, False),
('x < 3 and x < 3', {
'x < 3': 'True'
}, True),
('x < 3 and y > 4', {
'x < 3': 'True',
'y > 4': 'False'
}, False),
('x < 3 and not y > 4', {
'x < 3': 'True',
'y > 4': 'False'
}, True),
]
for expr, mapping, result in cases:
subbed = ExprManip.sub_for_comps(expr, mapping)
assert eval(subbed) == result
