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