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"