Skip to content

Commit 5de2f64

Browse files
committed
Add CFD tests
1 parent 2472ba0 commit 5de2f64

File tree

2 files changed

+37
-0
lines changed

2 files changed

+37
-0
lines changed

dpnp/tests/test_sycl_queue.py

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -356,6 +356,30 @@ def test_1in_1out(func, data, device):
356356
pass
357357

358358

359+
@pytest.mark.parametrize(
360+
"func, data",
361+
[
362+
pytest.param("frexp", numpy.arange(9)),
363+
],
364+
)
365+
@pytest.mark.parametrize("device", valid_dev, ids=dev_ids)
366+
def test_1in_2out(func, data, device):
367+
x = dpnp.array(data, device=device)
368+
res1, res2 = getattr(dpnp, func)(x)
369+
assert_sycl_queue_equal(res1.sycl_queue, x.sycl_queue)
370+
assert_sycl_queue_equal(res2.sycl_queue, x.sycl_queue)
371+
372+
out1 = dpnp.empty_like(res1)
373+
out2 = dpnp.empty_like(res2)
374+
try:
375+
# some functions do not support out kwarg
376+
getattr(dpnp, func)(x, out=(out1, out2))
377+
assert_sycl_queue_equal(out1.sycl_queue, x.sycl_queue)
378+
assert_sycl_queue_equal(out2.sycl_queue, x.sycl_queue)
379+
except TypeError:
380+
pass
381+
382+
359383
@pytest.mark.parametrize(
360384
"func,data1,data2",
361385
[

dpnp/tests/test_usm_type.py

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -649,6 +649,19 @@ def test_1in_1out(func, data, usm_type):
649649
assert x.usm_type == usm_type == res.usm_type
650650

651651

652+
@pytest.mark.parametrize(
653+
"func, data",
654+
[
655+
pytest.param("frexp", numpy.arange(9)),
656+
],
657+
)
658+
@pytest.mark.parametrize("usm_type", list_of_usm_types)
659+
def test_1in_2out(func, data, usm_type):
660+
x = dpnp.array(data, usm_type=usm_type)
661+
res1, res2 = getattr(dpnp, func)(x)
662+
assert x.usm_type == usm_type == res1.usm_type == res2.usm_type
663+
664+
652665
@pytest.mark.parametrize(
653666
"func,data1,data2",
654667
[

0 commit comments

Comments
 (0)