From e488dcd6909b1d32a7a0c10e42f6c9ee90e6901b Mon Sep 17 00:00:00 2001 From: Mikhail Golubev Date: Sun, 21 Jan 2018 21:26:09 +0300 Subject: [PATCH] PY-16412 A more complex case of a type comment for unpacked assignment target --- .../typeCommentComplexUnpackedAssignmentTarget.py | 3 +++ .../typeCommentComplexUnpackedAssignmentTarget_after.py | 3 +++ .../intentions/PyAnnotateVariableTypeIntentionTest.java | 4 ++++ 3 files changed, 10 insertions(+) create mode 100644 python/testData/intentions/PyAnnotateVariableTypeIntentionTest/typeCommentComplexUnpackedAssignmentTarget.py create mode 100644 python/testData/intentions/PyAnnotateVariableTypeIntentionTest/typeCommentComplexUnpackedAssignmentTarget_after.py diff --git a/python/testData/intentions/PyAnnotateVariableTypeIntentionTest/typeCommentComplexUnpackedAssignmentTarget.py b/python/testData/intentions/PyAnnotateVariableTypeIntentionTest/typeCommentComplexUnpackedAssignmentTarget.py new file mode 100644 index 000000000000..dd16b7031640 --- /dev/null +++ b/python/testData/intentions/PyAnnotateVariableTypeIntentionTest/typeCommentComplexUnpackedAssignmentTarget.py @@ -0,0 +1,3 @@ +def func(): + ((var, _), _) = ('foo', 1), 2 + var diff --git a/python/testData/intentions/PyAnnotateVariableTypeIntentionTest/typeCommentComplexUnpackedAssignmentTarget_after.py b/python/testData/intentions/PyAnnotateVariableTypeIntentionTest/typeCommentComplexUnpackedAssignmentTarget_after.py new file mode 100644 index 000000000000..85868318ee04 --- /dev/null +++ b/python/testData/intentions/PyAnnotateVariableTypeIntentionTest/typeCommentComplexUnpackedAssignmentTarget_after.py @@ -0,0 +1,3 @@ +def func(): + ((var, _), _) = ('foo', 1), 2 # type: ((str, int), int) + var diff --git a/python/testSrc/com/jetbrains/python/intentions/PyAnnotateVariableTypeIntentionTest.java b/python/testSrc/com/jetbrains/python/intentions/PyAnnotateVariableTypeIntentionTest.java index 7f090cb19a51..680708151998 100644 --- a/python/testSrc/com/jetbrains/python/intentions/PyAnnotateVariableTypeIntentionTest.java +++ b/python/testSrc/com/jetbrains/python/intentions/PyAnnotateVariableTypeIntentionTest.java @@ -74,6 +74,10 @@ public class PyAnnotateVariableTypeIntentionTest extends PyIntentionTestCase { doTestTypeComment(); } + public void testTypeCommentComplexUnpackedAssignmentTarget() { + doTestTypeComment(); + } + public void testTypeCommentLocalChainedAssignmentTarget() { doTestTypeComment(); }