From 37d7b85ed18539c4175072334d0e5ca0e7254a2e Mon Sep 17 00:00:00 2001
From: Mathias Rav <rav@cs.au.dk>
Date: Sat, 4 Feb 2017 13:18:41 +0100
Subject: [PATCH] Add tests for decorator completion

---
 test/test_evaluate/test_compiled.py | 16 ++++++++++++++++
 1 file changed, 16 insertions(+)

diff --git a/test/test_evaluate/test_compiled.py b/test/test_evaluate/test_compiled.py
index a8bfe65e..00a84d7f 100644
--- a/test/test_evaluate/test_compiled.py
+++ b/test/test_evaluate/test_compiled.py
@@ -83,3 +83,19 @@ 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