Add brain tip for numpy.sum. Close PyCQA/pylint#1558

2 files changed, 9 insertions, 1 deletions

diff --git a/astroid/brain/brain_numpy.py b/astroid/brain/brain_numpy.py
index 257e995f..9b09dacf 100644
--- a/astroid/brain/brain_numpy.py
+++ b/astroid/brain/brain_numpy.py
@@ -177,8 +177,17 @@ def numpy_core_numerictypes_transform():
     ''')
 
 
+def numpy_funcs():
+    return astroid.parse('''
+    import builtins
+    def sum(a, axis=None, dtype=None, out=None, keepdims=None):
+        return builtins.sum(a)
+    ''')
+
+
 astroid.register_module_extender(astroid.MANAGER, 'numpy.core.umath', numpy_core_umath_transform)
 astroid.register_module_extender(astroid.MANAGER, 'numpy.random.mtrand',
                                  numpy_random_mtrand_transform)
 astroid.register_module_extender(astroid.MANAGER, 'numpy.core.numerictypes',
                                  numpy_core_numerictypes_transform)
+astroid.register_module_extender(astroid.MANAGER, 'numpy', numpy_funcs)
diff --git a/astroid/tests/unittest_brain.py b/astroid/tests/unittest_brain.py
index 38d2e5e5..97c07e0d 100644
--- a/astroid/tests/unittest_brain.py
+++ b/astroid/tests/unittest_brain.py
@@ -1218,7 +1218,6 @@ class TestLenBuiltinInference:
         """)
         assert next(node.infer()).as_string() == '5'
 
-
     @pytest.mark.xfail(reason="Can't use list special astroid fields")
     def test_int_subclass_argument(self):
         """I am unable to access the length of a object which