def test_select_expr_contains(self):
"Checks that a contains operation 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.select_rewrite_expression') as cr:
merge.select_rewrite_expression(DEEP, name, [c])
assert cr.called
def test_select_expr_compare(self):
"Checks that a compare operation 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.select_rewrite_expression') as c:
merge.select_rewrite_expression(name, [n])
assert c.called
def test_select_expr_contains(self):
"Checks that a contains operation 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.select_rewrite_expression') as cr:
merge.select_rewrite_expression(DEEP, name, [c])
assert cr.called
def test_select_expr_compare(self):
"Checks that a compare operation 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.select_rewrite_expression') as c:
merge.select_rewrite_expression(DEEP, name, [n])
assert c.called
def test_select_expr_logical(self):
"Checks that a logical operation is not selected"
l = ast.Literal('foo')
r = ast.Literal('bar')
n = ast.LogicalOperator('or', l, r)
name = merge.node_name(n)
expr = merge.select_rewrite_expression(name, [n])
assert expr == l
n.left = None
expr = merge.select_rewrite_expression(name, [n])
assert expr == r
def test_select_expr_logical(self):
"Checks that a logical operation is not selected"
l = ast.Literal('foo')
r = ast.Literal('bar')
n = ast.LogicalOperator('or', l, r)
name = merge.node_name(n)
expr = merge.select_rewrite_expression(DEEP, name, [n])
assert expr == l
n.left = None
expr = merge.select_rewrite_expression(DEEP, name, [n])
assert expr == r
def test_select_expr_first(self):
"Checks that first expression is selected"
l = ast.Literal('foo')
r = ast.Literal('bar')
name = merge.node_name(l)
expr = merge.select_rewrite_expression(name, [l, r])
assert expr == l
def test_select_expr_negate(self):
"Checks that a negate operation is not selected"
l = ast.Literal('foo')
n = ast.NegateOperator(l)
name = merge.node_name(n)
expr = merge.select_rewrite_expression(name, [n])
assert expr == l
def test_select_expr_first(self):
"Checks that first expression is selected"
l = ast.Literal('foo')
r = ast.Literal('bar')
name = merge.node_name(l)
expr = merge.select_rewrite_expression(DEEP, name, [l, r])
assert expr == l
def test_select_expr_negate(self):
"Checks that a negate operation is not selected"
l = ast.Literal('foo')
n = ast.NegateOperator(l)
name = merge.node_name(n)
expr = merge.select_rewrite_expression(DEEP, name, [n])
assert expr == l