Improve dpctl.tensor.full error when fill_value is of an invalid type

pybind11 raises RuntimeError when a value cannot be cast to a type, i.e., a string to an integer. Now when scalar is not an array, a check is performed, and TypeError is raised

Author: ndgrigorian

dpctl/tensor/_ctors.py

@@ -15,6 +15,7 @@
 #  limitations under the License.
 
 import operator
+from numbers import Number
 
 import numpy as np
 
@@ -1113,6 +1114,11 @@ def full(
 
     sycl_queue = normalize_queue_device(sycl_queue=sycl_queue, device=device)
     usm_type = usm_type if usm_type is not None else "device"
+    if not isinstance(fill_value, Number):
+        raise TypeError(
+            "`full` array cannot be constructed with value of type "
+            f"{type(fill_value)}"
+        )
     dtype = _get_dtype(dtype, sycl_queue, ref_type=type(fill_value))
     res = dpt.usm_ndarray(
         shape,

dpctl/tests/test_usm_ndarray_ctor.py

@@ -2621,3 +2621,10 @@ def test_setitem_from_numpy_contig():
 
     expected = dpt.reshape(dpt.arange(-10, 10, dtype=fp_dt), (4, 5))
     assert dpt.all(dpt.flip(Xdpt, axis=-1) == expected)
+
+
+def test_full_raises_type_error():
+    get_queue_or_skip()
+
+    with pytest.raises(TypeError):
+        dpt.full(1, "0")