def test_objects_pow_expression_one(patch, tree):
"""
Ensures ExpressionVisitor.pow_expression works with one node
"""
patch.many(ExpressionVisitor, ['primary_expression'])
tree.child(0).data = 'primary_expression'
tree.children = [1]
r = ExpressionVisitor().pow_expression(tree)
ExpressionVisitor.primary_expression.assert_called_with(tree.child(0))
assert r == ExpressionVisitor.primary_expression()
def test_objects_pow_expression_two(patch, tree):
"""
Ensures ExpressionVisitor.pow_expression works with two nodes
"""
patch.many(ExpressionVisitor, ['nary_expression',
'primary_expression',
'unary_expression'])
tree.child(1).child(0).type = 'POWER'
tree.children = [1, '+', 2]
tree.child(1).data = 'pow_operator'
r = ExpressionVisitor().pow_expression(tree)
ExpressionVisitor.nary_expression.assert_called_with(
tree, tree.child(1).child(0), [
ExpressionVisitor.primary_expression(tree.child(0)),
ExpressionVisitor.unary_expression(tree.child(2))
])
assert r == ExpressionVisitor.nary_expression()