# Path: shopify_python/ast.py
# def count_tree_size(node):  # type: (astroid.NodeNG) -> int


import astroid
from shopify_python import ast



def test_count_tree_size():
    root = astroid.builder.parse("""
    def test(x, y):
        return x * y if x > 5 else 0
    """)

# Path: shopify_python/ast.py
# def count_tree_size(node):  # type: (astroid.NodeNG) -> int


# Path: tests/shopify_python/test_ast.py
import astroid
from shopify_python import ast




def test_count_tree_size():
    root = astroid.builder.parse("""
    def test(x, y):
        return x * y if x > 5 else 0
    """)

    assert ast.count_tree_size(root) == 14