Skip to content

Commit 7138c73

Browse files
committed
Add testing of divmod operator
1 parent 93c52d9 commit 7138c73

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

dpnp/tests/test_binary_two_outputs_ufuncs.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -357,3 +357,17 @@ def test_float_diff_signs(self, dt):
357357
exp1, exp2 = numpy.divmod(a, b)
358358
assert_array_equal(res1, exp1)
359359
assert_array_equal(res2, exp2)
360+
361+
def test_divmod_op(self):
362+
a = numpy.arange(1, 6)
363+
ia = dpnp.array(a)
364+
365+
res1, res2 = divmod(ia, 3)
366+
exp1, exp2 = divmod(a, 3)
367+
assert_array_equal(res1, exp1)
368+
assert_array_equal(res2, exp2)
369+
370+
res1, res2 = divmod(2, ia)
371+
exp1, exp2 = divmod(2, a)
372+
assert_array_equal(res1, exp1)
373+
assert_array_equal(res2, exp2)

0 commit comments

Comments
 (0)