Skip to content

Commit f0d926a

Browse files
Add test_full_cmplx128
1 parent 2e73a6c commit f0d926a

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

dpctl/tests/test_usm_ndarray_ctor.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1682,6 +1682,27 @@ def test_full(dtype):
16821682
assert np.array_equal(dpt.asnumpy(X), np.full(10, 4, dtype=dtype))
16831683

16841684

1685+
def test_full_cmplx128():
1686+
q = get_queue_or_skip()
1687+
dtype = "c16"
1688+
skip_if_dtype_not_supported(dtype, q)
1689+
fill_v = 1 + 1j
1690+
X = dpt.full(tuple(), fill_value=fill_v, dtype=dtype, sycl_queue=q)
1691+
assert np.array_equal(
1692+
dpt.asnumpy(X), np.full(tuple(), fill_value=fill_v, dtype=dtype)
1693+
)
1694+
fill_v = 0 + 1j
1695+
X = dpt.full(tuple(), fill_value=fill_v, dtype=dtype, sycl_queue=q)
1696+
assert np.array_equal(
1697+
dpt.asnumpy(X), np.full(tuple(), fill_value=fill_v, dtype=dtype)
1698+
)
1699+
fill_v = 0 + 0j
1700+
X = dpt.full(tuple(), fill_value=fill_v, dtype=dtype, sycl_queue=q)
1701+
assert np.array_equal(
1702+
dpt.asnumpy(X), np.full(tuple(), fill_value=fill_v, dtype=dtype)
1703+
)
1704+
1705+
16851706
def test_full_dtype_inference():
16861707
try:
16871708
X = dpt.full(10, 4)

0 commit comments

Comments
 (0)