Skip to content

Commit e32b395

Browse files
Expanded testing of scalar copy in dtypes
1 parent 71fc06a commit e32b395

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

dpctl/tests/test_usm_ndarray_ctor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -116,7 +116,7 @@ def test_properties():
116116

117117
@pytest.mark.parametrize("func", [bool, float, int])
118118
@pytest.mark.parametrize("shape", [(1,), (1, 1), (1, 1, 1)])
119-
@pytest.mark.parametrize("dtype", ["|b1", "|f8", "|i8"])
119+
@pytest.mark.parametrize("dtype", ["|b1", "|u2", "|f4", "|i8"])
120120
def test_copy_scalar_with_func(func, shape, dtype):
121121
X = dpt.usm_ndarray(shape, dtype=dtype)
122122
Y = np.arange(1, X.size + 1, dtype=dtype).reshape(shape)
@@ -126,7 +126,7 @@ def test_copy_scalar_with_func(func, shape, dtype):
126126

127127
@pytest.mark.parametrize("method", ["__bool__", "__float__", "__int__"])
128128
@pytest.mark.parametrize("shape", [(1,), (1, 1), (1, 1, 1)])
129-
@pytest.mark.parametrize("dtype", ["|b1", "|f8", "|i8"])
129+
@pytest.mark.parametrize("dtype", ["|b1", "|u2", "|f4", "|i8"])
130130
def test_copy_scalar_with_method(method, shape, dtype):
131131
X = dpt.usm_ndarray(shape, dtype=dtype)
132132
Y = np.arange(1, X.size + 1, dtype=dtype).reshape(shape)

0 commit comments

Comments
 (0)