summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorDmitry Pribysh <dmand@yandex.ru>2015-10-24 16:39:40 +0300
committerDmitry Pribysh <dmand@yandex.ru>2015-10-24 16:39:40 +0300
commitfbf6636bca20ffd4066206d3d72bc94e15dc7c02 (patch)
treee72b02d5658a8b79747f26e3e9f701efaf621fa4
parent63d38d6b6d41d2259f909faa51a8272220dd2de7 (diff)
downloadastroid-fbf6636bca20ffd4066206d3d72bc94e15dc7c02.tar.gz
Use safe_infer for inference of elements in list multiplication operationfix-list-multiplication
Now inference yields YES instead of list item value when multiple values are inferred for one list item.
-rw-r--r--astroid/protocols.py7
1 files changed, 4 insertions, 3 deletions
diff --git a/astroid/protocols.py b/astroid/protocols.py
index c51464d..7c9e4b9 100644
--- a/astroid/protocols.py
+++ b/astroid/protocols.py
@@ -143,9 +143,10 @@ def _multiply_seq_by_int(self, other, context):
node = self.__class__()
elts = []
for elt in self.elts:
- infered = next(elt.infer(context))
- if not infered is util.YES:
- elts.append(infered)
+ infered = helpers.safe_infer(elt, context)
+ if infered is None:
+ infered = util.YES
+ elts.append(infered)
node.elts = elts * other.value
return node