diff options
| author | hippo91 <guillaume.peillex@gmail.com> | 2019-12-08 20:43:28 +0100 |
|---|---|---|
| committer | Claudiu Popa <pcmanticore@gmail.com> | 2019-12-09 08:42:00 +0100 |
| commit | f233e599f603ffb26eb8f39d547142bcd58b29df (patch) | |
| tree | c14c42ccccc72a5357ea0ffb7ca12dd94293d2fb | |
| parent | d26628884e840ec284812e5343181a638233d5b0 (diff) | |
| download | astroid-git-f233e599f603ffb26eb8f39d547142bcd58b29df.tar.gz | |
Adds a special Ufunc class for ufunc objects that return pair of ndarray
| -rw-r--r-- | astroid/brain/brain_numpy_core_umath.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/astroid/brain/brain_numpy_core_umath.py b/astroid/brain/brain_numpy_core_umath.py index b693ae22..029d9d81 100644 --- a/astroid/brain/brain_numpy_core_umath.py +++ b/astroid/brain/brain_numpy_core_umath.py @@ -52,6 +52,10 @@ def numpy_core_umath_transform(): def __call__(self, x, {opt_args:s}): return numpy.ndarray([0, 0]) + class FakeUfuncOneArgBis(FakeUfunc): + def __call__(self, x, {opt_args:s}): + return numpy.ndarray([0, 0]), numpy.ndarray([0, 0]) + class FakeUfuncTwoArgs(FakeUfunc): def __call__(self, x1, x2, {opt_args:s}): return numpy.ndarray([0, 0]) @@ -75,18 +79,14 @@ def numpy_core_umath_transform(): #exp2 = FakeUfuncOneArg() expm1 = FakeUfuncOneArg() fabs = FakeUfuncOneArg() - #frexp = FakeUfuncOneArg() - # frexp.__call__(self, x, {opt_args:s}): - # return (numpy.ndarray([0, 0]), numpy.ndarray([0, 0])) + frexp = FakeUfuncOneArgBis() #isfinite = FakeUfuncOneArg() #isinf = FakeUfuncOneArg() log = FakeUfuncOneArg() log1p = FakeUfuncOneArg() log2 = FakeUfuncOneArg() logical_not = FakeUfuncOneArg() - # modf = FakeUfuncOneArg() - # modf.__call__(self, x, {opt_args:s}): - # return (numpy.ndarray([0, 0]), numpy.ndarray([0, 0])) + modf = FakeUfuncOneArgBis() negative = FakeUfuncOneArg() rad2deg = FakeUfuncOneArg() reciprocal = FakeUfuncOneArg() |