Skip to content

Commit 1dc6dec

Browse files
committed
Add proper casting of input arrays passed to assert_array_almost_equal
1 parent 1a000b9 commit 1dc6dec

File tree

3 files changed

+7
-0
lines changed

3 files changed

+7
-0
lines changed

dpnp/tests/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,5 +4,6 @@
44

55
numpy.testing.assert_allclose = testing.assert_allclose
66
numpy.testing.assert_almost_equal = testing.assert_almost_equal
7+
numpy.testing.assert_array_almost_equal = testing.assert_array_almost_equal
78
numpy.testing.assert_array_equal = testing.assert_array_equal
89
numpy.testing.assert_equal = testing.assert_equal

dpnp/tests/testing/__init__.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
from .array import (
22
assert_allclose,
33
assert_almost_equal,
4+
assert_array_almost_equal,
45
assert_array_equal,
56
assert_equal,
67
)

dpnp/tests/testing/array.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@
3030

3131
assert_allclose_orig = numpy.testing.assert_allclose
3232
assert_almost_equal_orig = numpy.testing.assert_almost_equal
33+
assert_array_almost_equal_orig = numpy.testing.assert_array_almost_equal
3334
assert_array_equal_orig = numpy.testing.assert_array_equal
3435
assert_equal_orig = numpy.testing.assert_equal
3536

@@ -49,6 +50,10 @@ def assert_almost_equal(result, expected, *args, **kwargs):
4950
_assert(assert_almost_equal_orig, result, expected, *args, **kwargs)
5051

5152

53+
def assert_array_almost_equal(result, expected, *args, **kwargs):
54+
_assert(assert_array_almost_equal_orig, result, expected, *args, **kwargs)
55+
56+
5257
def assert_array_equal(result, expected, *args, **kwargs):
5358
_assert(assert_array_equal_orig, result, expected, *args, **kwargs)
5459

0 commit comments

Comments
 (0)