diff --git a/test/completion/decorators.py b/test/completion/decorators.py index 702c3f39..685f1f65 100644 --- a/test/completion/decorators.py +++ b/test/completion/decorators.py @@ -304,3 +304,15 @@ class A(): #? int() A().ret() + + +# ----------------- +# On decorator completions +# ----------------- + +import abc +#? ['abc'] +@abc + +#? ['abstractmethod'] +@abc.abstractmethod''') diff --git a/test/test_evaluate/test_compiled.py b/test/test_evaluate/test_compiled.py index 00a84d7f..a8bfe65e 100644 --- a/test/test_evaluate/test_compiled.py +++ b/test/test_evaluate/test_compiled.py @@ -83,19 +83,3 @@ def test_method_completion(): else: result = ['__func__'] assert [c.name for c in Script(code).completions()] == result - - -def test_decorator_global(): - code = dedent(''' - import abc - @abc''') - result = ['abc'] - assert [c.name for c in Script(code).completions()] == result - - -def test_decorator_dotted(): - code = dedent(''' - import abc - @abc.abstractmethod''') - result = ['abstractmethod'] - assert [c.name for c in Script(code).completions()] == result