summaryrefslogtreecommitdiff
path: root/astroid/inference.py
diff options
context:
space:
mode:
authorClaudiu Popa <pcmanticore@gmail.com>2018-07-24 10:22:22 +0200
committerClaudiu Popa <pcmanticore@gmail.com>2018-07-24 10:22:22 +0200
commitfcbf25697d613c37bcc9d38004d576ce97f2cd78 (patch)
tree3bc5a31b9c0f32b63fc729ea0116659dc021a0be /astroid/inference.py
parent32c96a687ad0e5c59f7c1800bc2e00af26b16d0b (diff)
downloadastroid-git-fcbf25697d613c37bcc9d38004d576ce97f2cd78.tar.gz
Rename method for making it clearer
Diffstat (limited to 'astroid/inference.py')
-rw-r--r--astroid/inference.py30
1 files changed, 15 insertions, 15 deletions
diff --git a/astroid/inference.py b/astroid/inference.py
index 3b00f689..609b7948 100644
--- a/astroid/inference.py
+++ b/astroid/inference.py
@@ -57,17 +57,7 @@ nodes.Const._infer = infer_end
nodes.Slice._infer = infer_end
-def infer_seq(self, context=None):
- if not any(isinstance(e, nodes.Starred) for e in self.elts):
- yield self
- else:
- values = _infer_seq(self, context)
- new_seq = type(self)(self.lineno, self.col_offset, self.parent)
- new_seq.postinit(values)
- yield new_seq
-
-
-def _infer_seq(node, context=None):
+def _infer_sequence_helper(node, context=None):
"""Infer all values based on _BaseContainer.elts"""
values = []
@@ -80,15 +70,25 @@ def _infer_seq(node, context=None):
if not hasattr(starred, 'elts'):
raise exceptions.InferenceError(node=node,
context=context)
- values.extend(_infer_seq(starred))
+ values.extend(_infer_sequence_helper(starred))
else:
values.append(elt)
return values
-nodes.List._infer = infer_seq
-nodes.Tuple._infer = infer_seq
-nodes.Set._infer = infer_seq
+def infer_sequence(self, context=None):
+ if not any(isinstance(e, nodes.Starred) for e in self.elts):
+ yield self
+ else:
+ values = _infer_sequence_helper(self, context)
+ new_seq = type(self)(self.lineno, self.col_offset, self.parent)
+ new_seq.postinit(values)
+ yield new_seq
+
+
+nodes.List._infer = infer_sequence
+nodes.Tuple._infer = infer_sequence
+nodes.Set._infer = infer_sequence
def infer_map(self, context=None):