diff options
| -rw-r--r-- | astroid/brain/brain_numpy.py | 9 | ||||
| -rw-r--r-- | astroid/tests/unittest_brain.py | 1 |
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 |