Skip to content

Commit 784ffa6

Browse files
committed
Update remaining manipulation tests
1 parent 1f0447b commit 784ffa6

File tree

3 files changed

+8
-2
lines changed

3 files changed

+8
-2
lines changed

dpnp/tests/third_party/cupy/manipulation_tests/test_split.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
11
import unittest
22

3-
import dpnp as cupy
43
from dpnp.tests.third_party.cupy import testing
54

65

76
class TestSplit(unittest.TestCase):
7+
88
@testing.numpy_cupy_array_equal()
99
def test_array_split1(self, xp):
1010
a = testing.shaped_arange((3, 11), xp)

dpnp/tests/third_party/cupy/manipulation_tests/test_tiling.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,13 +17,15 @@
1717
{"repeats": [1, 2, 3], "axis": -2},
1818
)
1919
class TestRepeat(unittest.TestCase):
20+
2021
@testing.numpy_cupy_array_equal()
2122
def test_array_repeat(self, xp):
2223
x = testing.shaped_arange((2, 3, 4), xp)
2324
return xp.repeat(x, self.repeats, self.axis)
2425

2526

2627
class TestRepeatRepeatsNdarray(unittest.TestCase):
28+
2729
def test_func(self):
2830
a = testing.shaped_arange((2, 3, 4), cupy)
2931
repeats = cupy.array([2, 3], dtype=cupy.int32)
@@ -61,6 +63,7 @@ def test_array_repeat(self, xp):
6163
{"repeats": [1, 2, 3, 4], "axis": 0},
6264
)
6365
class TestRepeat1D(unittest.TestCase):
66+
6467
@testing.numpy_cupy_array_equal()
6568
def test_array_repeat(self, xp):
6669
x = testing.shaped_arange((4,), xp)
@@ -89,6 +92,7 @@ def test_array_repeat(self, xp):
8992
{"repeats": 2, "axis": 3},
9093
)
9194
class TestRepeatFailure(unittest.TestCase):
95+
9296
def test_repeat_failure(self):
9397
for xp in (numpy, cupy):
9498
x = testing.shaped_arange((2, 3, 4), xp)
@@ -105,6 +109,7 @@ def test_repeat_failure(self):
105109
{"reps": (2, 3, 4, 5)},
106110
)
107111
class TestTile(unittest.TestCase):
112+
108113
@testing.numpy_cupy_array_equal()
109114
def test_array_tile(self, xp):
110115
x = testing.shaped_arange((2, 3, 4), xp)
@@ -116,6 +121,7 @@ def test_array_tile(self, xp):
116121
{"reps": (-1, -2)},
117122
)
118123
class TestTileFailure(unittest.TestCase):
124+
119125
def test_tile_failure(self):
120126
for xp in (numpy, cupy):
121127
x = testing.shaped_arange((2, 3, 4), xp)

dpnp/tests/third_party/cupy/math_tests/test_arithmetic.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -266,7 +266,7 @@ def test_imag_complex(self, xp, dtype):
266266

267267
class ArithmeticBinaryBase:
268268

269-
@testing.numpy_cupy_allclose(atol=1e-4, type_check=has_support_aspect64())
269+
@testing.numpy_cupy_allclose(rtol=1e-4, type_check=has_support_aspect64())
270270
def check_binary(self, xp):
271271
arg1 = self.arg1
272272
arg2 = self.arg2

0 commit comments

Comments
 (0)