Factorize functions inference system

1 files changed, 21 insertions, 0 deletions

diff --git a/astroid/brain/brain_numpy_utils.py b/astroid/brain/brain_numpy_utils.py
new file mode 100644
index 00000000..4de4ad46
--- /dev/null
+++ b/astroid/brain/brain_numpy_utils.py
@@ -0,0 +1,21 @@
+# Copyright (c) 2018-2019 hippo91 <guillaume.peillex@gmail.com>
+
+# Licensed under the LGPL: https://www.gnu.org/licenses/old-licenses/lgpl-2.1.en.html
+# For details: https://github.com/PyCQA/astroid/blob/master/COPYING.LESSER
+
+
+"""Different utilities for the numpy brains"""
+
+
+import astroid
+
+
+def infer_numpy_member(src, node, context=None):
+    node = astroid.extract_node(src)
+    return node.infer(context=context)
+
+
+def looks_like_numpy_member(member_name, node):
+    return (isinstance(node, astroid.Attribute)
+            and node.attrname == member_name
+            and node.expr.inferred()[-1].name == 'numpy')