def test_compiler_if_block(patch, compiler):
patch.object(Objects, 'expression')
patch.many(Compiler, ['add_line', 'subtree'])
tree = Tree('if_block', [Tree('if_statement', []),
Tree('nested_block', [])])
compiler.if_block(tree)
Objects.expression.assert_called_with(tree.node('if_statement'))
nested_block = tree.node('nested_block')
args = Objects.expression()
compiler.add_line.assert_called_with('if', tree.line(), args=args,
enter=nested_block.line(),
parent=None)
compiler.subtree.assert_called_with(nested_block, parent=tree.line())
def test_tree_line():
tree = Tree('outer', [Tree('path', [Token('WORD', 'word', line=1)])])
assert tree.line() == '1'
def test_tree_line():
tree = Tree("outer", [Tree("path", [Token("WORD", "word", line=1)])])
assert tree.line() == "1"
