def test_rewrite_norm(self):
"Test a simple re-write"
l = ast.Literal('foo')
r = ast.Literal('bar')
c = ast.LogicalOperator('or', l, r)
name = merge.node_name(c)
merge.rewrite_ast(c, name, l, True)
assert isinstance(c.left, ast.Constant)
def test_rewrite_norm(self):
"Test a simple re-write"
l = ast.Literal('foo')
r = ast.Literal('bar')
c = ast.LogicalOperator('or', l, r)
name = merge.node_name(c)
merge.rewrite_ast(c, name, l, True)
assert isinstance(c.left, ast.Constant)
def test_rewrite_compare(self):
"Checks that a compare rewrite uses compare module"
l = ast.Literal('foo')
r = ast.Literal('bar')
n = ast.CompareOperator('>', l, r)
name = merge.node_name(n)
with patch('pypred.merge.compare.compare_rewrite') as c:
merge.rewrite_ast(n, name, n, True)
assert c.called
def test_rewrite_contains(self):
"Checks that a contain rewrite uses contains module"
l = ast.LiteralSet([ast.Number(1), ast.Number(2)])
r = ast.Literal('bar')
c = ast.ContainsOperator(l, r)
name = merge.node_name(c)
with patch('pypred.merge.contains.contains_rewrite') as cr:
merge.rewrite_ast(c, name, c, True)
assert cr.called
def test_rewrite_contains(self):
"Checks that a contain rewrite uses contains module"
l = ast.LiteralSet([ast.Number(1), ast.Number(2)])
r = ast.Literal('bar')
c = ast.ContainsOperator(l, r)
name = merge.node_name(c)
with patch('pypred.merge.contains.contains_rewrite') as cr:
merge.rewrite_ast(c, name, c, True)
assert cr.called
def test_rewrite_compare(self):
"Checks that a compare rewrite uses compare module"
l = ast.Literal('foo')
r = ast.Literal('bar')
n = ast.CompareOperator('>', l, r)
name = merge.node_name(n)
with patch('pypred.merge.compare.compare_rewrite') as c:
merge.rewrite_ast(n, name, n, True)
assert c.called
