Skip to content

Fix for issue 7402: implement missing numpy ufunc interface #7403

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
sklam merged 8 commits into from
Nov 24, 2021
Merged

Fix for issue 7402: implement missing numpy ufunc interface #7403

Changes from 1 commit
Commits
Show all changes
8 commits
Select commit Hold shift + click to select a range
4b3003b
fix for issue 7402
guilhermeleobas Sep 15, 2021
27decbd
flake8
guilhermeleobas Sep 15, 2021
e4ede0d
expose missing ufunc members
guilhermeleobas Sep 15, 2021
dc1ab6d
flake8
guilhermeleobas Sep 16, 2021
8168060
add tests
guilhermeleobas Sep 21, 2021
a343c6e
update test
guilhermeleobas Sep 21, 2021
ed2ea86
add tests for gufunc attributes
guilhermeleobas Sep 28, 2021
a2419a2
nit: fix grammar
guilhermeleobas Sep 28, 2021
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
update test
  • Loading branch information
@guilhermeleobas
guilhermeleobas committed Sep 21, 2021
commit a343c6e892e787d4a8c8adae820c8df8c23596b9
6 changes: 4 additions & 2 deletions numba/tests/npyufunc/test_gufunc.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -198,20 +198,22 @@ def my_cumsum(x, res):
my_cumsum(x, out=out_kw, axis=0)
np.testing.assert_equal(out_kw, expected)

def test_attrs(self):
def test_gufunc_attributes(self):
@guvectorize("(n)->(n)")
def gufunc(x, res):
acc = 0
for i in range(x.shape[0]):
acc += x[i]
res[i] = acc

# ensure gufunc exports attributes
attrs = ['signature', 'accumulate', 'at', 'outer', 'reduce', 'reduceat']
for attr in attrs:
# ensure gufunc exports the propertie above
contains = hasattr(gufunc, attr)
self.assertTrue(contains, 'dynamic gufunc not exporting "%s"' % (attr,))

self.assertEqual(gufunc.signature, "(n)->(n)")


class TestGUVectorizeScalar(TestCase):
"""
Expand Down