TestScopeTransformer

Complexity

Total Complexity

4

Size/Duplication

Total Lines	9
Duplicated Lines	0 %

Importance

Changes	4
Bugs	0	Features	0

1 Method

Rating	Name	Duplication	Size	Complexity
A	test_scope_added()	0	8	4

import ast
from py14.scope import add_scope_context
from py14.context import add_variable_context


def parse(*args):
    source = ast.parse("\n".join(args))
    add_scope_context(source)
    return source


class TestScopeTransformer:
    def test_scope_added(self):
        source = parse(
            "def foo():",
            "   return 10",
        )
        assert isinstance(source.scopes[-1], ast.Module)
        assert isinstance(source.body[0].scopes[-1], ast.FunctionDef)
        assert isinstance(source.body[0].body[0].scopes[-1], ast.FunctionDef)


class TestScopeList:
    def test_find_returns_most_upper_definition(self):
        source = parse(
            "x = 1",
            "def foo():",
            "   x = 2",
        )
        add_variable_context(source)
        definition = source.scopes.find("x")
        assert definition.lineno == 1