Bug 666 pure

ChangeLog

@@ -6,6 +6,12 @@ What's New in astroid 2.4.0?
 ============================
 Release Date: TBA
 
+
+* Added a call to ``register_transform`` for all functions of the ``brain_numpy_core_multiarray``
+  module in case the current node is an instance of ``astroid.Name``
+
+  Close #666
+
 * Added some functions to the ``brain_numpy_core_umath`` module
 
   Close PyCQA/pylint#3319

astroid/brain/brain_numpy_core_multiarray.py

@@ -85,3 +85,8 @@ def vdot(a, b):
         astroid.inference_tip(inference_function),
         functools.partial(looks_like_numpy_member, method_name),
     )
+    astroid.MANAGER.register_transform(
+        astroid.Name,
+        astroid.inference_tip(inference_function),
+        functools.partial(looks_like_numpy_member, method_name),
+    )

astroid/brain/brain_numpy_utils.py

@@ -48,9 +48,17 @@ def looks_like_numpy_member(
     :param node: node to test
     :return: True if the node is a member of numpy
     """
-    return (
+    if (
         isinstance(node, astroid.Attribute)
         and node.attrname == member_name
         and isinstance(node.expr, astroid.Name)
         and _is_a_numpy_module(node.expr)
-    )
+    ):
+        return True
+    if (
+        isinstance(node, astroid.Name)
+        and node.name == member_name
+        and node.root().name.startswith("numpy")
+    ):
+        return True
+    return False