Update tests for showing the module name for both unpacking-non-sequence and unbalanced-tuple-unpacking.

5 files changed, 33 insertions, 10 deletions

diff --git a/test/input/func_unbalanced_tuple_unpacking.py b/test/input/func_unbalanced_tuple_unpacking.py
index 280d812..c2e01e9 100644
--- a/test/input/func_unbalanced_tuple_unpacking.py
+++ b/test/input/func_unbalanced_tuple_unpacking.py
@@ -1,5 +1,7 @@
 """Check possible unbalanced tuple unpacking """
 
+from input.unpacking import unpack
+
 __revision__ = 0
 
 def do_stuff():
@@ -61,3 +63,8 @@ def do_stuff8():
     """ This is correct """
     first, second = temp2()
     return first + second
+
+def do_stuff9():
+    """ This is not correct """
+    first, second = unpack()
+    return first + second
diff --git a/test/input/func_unpacking_non_sequence.py b/test/input/func_unpacking_non_sequence.py
index 506d60c..6f2e83c 100644
--- a/test/input/func_unpacking_non_sequence.py
+++ b/test/input/func_unpacking_non_sequence.py
@@ -2,6 +2,8 @@
 
 # pylint: disable=too-few-public-methods, invalid-name
 
+from input.unpacking import nonseq
+
 __revision__ = 0
 
 # Working
@@ -59,5 +61,6 @@ a, b = NonSeq()
 a, b = ValueError
 a, b = None
 a, b = 1
-a, b = locals
+a, b = nonseq
+a, b = nonseq()
 a, b = bad_unpacking()
diff --git a/test/input/unpacking.py b/test/input/unpacking.py
new file mode 100644
index 0000000..807d9b9
--- /dev/null
+++ b/test/input/unpacking.py
@@ -0,0 +1,11 @@
+""" Code for checking the display of the module

+for unbalanced-tuple-unpacking and unpacking-non-sequence

+"""

+

+def unpack():

+    """ Return something"""

+    return (1, 2, 3)

+

+def nonseq():

+    """ Return non sequence """

+    return 1
\ No newline at end of file
diff --git a/test/messages/func_unbalanced_tuple_unpacking.txt b/test/messages/func_unbalanced_tuple_unpacking.txt
index 43c6bb3..91880e7 100644
--- a/test/messages/func_unbalanced_tuple_unpacking.txt
+++ b/test/messages/func_unbalanced_tuple_unpacking.txt
@@ -1,4 +1,5 @@
-W:  7:do_stuff: Possible unbalanced tuple unpacking with sequence at line 7: left side has 2 label(s), right side has 3 value(s)
-W: 12:do_stuff1: Possible unbalanced tuple unpacking with sequence at line 12: left side has 2 label(s), right side has 3 value(s)
-W: 17:do_stuff2: Possible unbalanced tuple unpacking with sequence at line 17: left side has 2 label(s), right side has 3 value(s)
-W: 48:do_stuff7: Possible unbalanced tuple unpacking with sequence at line 44: left side has 2 label(s), right side has 3 value(s)
\ No newline at end of file
+W:  9:do_stuff: Possible unbalanced tuple unpacking with sequence at line 9: left side has 2 label(s), right side has 3 value(s)
+W: 14:do_stuff1: Possible unbalanced tuple unpacking with sequence at line 14: left side has 2 label(s), right side has 3 value(s)
+W: 19:do_stuff2: Possible unbalanced tuple unpacking with sequence at line 19: left side has 2 label(s), right side has 3 value(s)
+W: 50:do_stuff7: Possible unbalanced tuple unpacking with sequence at line 46: left side has 2 label(s), right side has 3 value(s)
+W: 69:do_stuff9: Possible unbalanced tuple unpacking with sequence at line 7 (input.unpacking): left side has 2 label(s), right side has 3 value(s)
\ No newline at end of file
diff --git a/test/messages/func_unpacking_non_sequence.txt b/test/messages/func_unpacking_non_sequence.txt
index de6c789..f0ea5f7 100644
--- a/test/messages/func_unpacking_non_sequence.txt
+++ b/test/messages/func_unpacking_non_sequence.txt
@@ -1,5 +1,6 @@
-W: 58: Attempting to unpack a non-sequence with non-sequence at line 49

-W: 60: Attempting to unpack a non-sequence with non-sequence at line 60

-W: 61: Attempting to unpack a non-sequence with non-sequence at line 61

-W: 62: Attempting to unpack a non-sequence with non-sequence at line 62 (__builtin__)

-W: 63: Attempting to unpack a non-sequence with non-sequence at line 55
\ No newline at end of file
+W: 60: Attempting to unpack a non-sequence with non-sequence at line 51

+W: 62: Attempting to unpack a non-sequence with non-sequence at line 62

+W: 63: Attempting to unpack a non-sequence with non-sequence at line 63

+W: 64: Attempting to unpack a non-sequence with non-sequence at line 9 (input.unpacking)

+W: 65: Attempting to unpack a non-sequence with non-sequence at line 11 (input.unpacking)

+W: 66: Attempting to unpack a non-sequence with non-sequence at line 57
\ No newline at end of file