Merge pull request #1878 from IntelPython/improve-full-error-for-invalid-scalar-type

Improve `dpctl.tensor.full` error for invalid `fill_value`

Author: ndgrigorian

CHANGELOG.md

@@ -18,6 +18,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Fixed
 * Fix for `tensor.result_type` when all inputs are Python built-in scalars [gh-1877](https://github.com/IntelPython/dpctl/pull/1877)
 
+* Improved error in constructors `tensor.full` and `tensor.full_like` when provided a non-numeric fill value [gh-1878](https://github.com/IntelPython/dpctl/pull/1878)
+
 ### Maintenance
 
 * Update black version used in Python code style workflow [gh-1828](https://github.com/IntelPython/dpctl/pull/1828)

dpctl/tensor/_ctors.py

@@ -15,6 +15,7 @@
 #  limitations under the License.
 
 import operator
+from numbers import Number
 
 import numpy as np
 
@@ -1037,6 +1038,19 @@ def _cast_fill_val(fill_val, dt):
         return fill_val
 
 
+def _validate_fill_value(fill_val):
+    """
+    Validates that `fill_val` is a numeric or boolean scalar.
+    """
+    # TODO: verify if `np.True_` and `np.False_` should be instances of
+    # Number in NumPy, like other NumPy scalars and like Python bools
+    # check for `np.bool_` separately as NumPy<2 has no `np.bool`
+    if not isinstance(fill_val, Number) and not isinstance(fill_val, np.bool_):
+        raise TypeError(
+            f"array cannot be filled with scalar of type {type(fill_val)}"
+        )
+
+
 def full(
     shape,
     fill_value,
@@ -1110,6 +1124,8 @@ def full(
             sycl_queue=sycl_queue,
         )
         return dpt.copy(dpt.broadcast_to(X, shape), order=order)
+    else:
+        _validate_fill_value(fill_value)
 
     sycl_queue = normalize_queue_device(sycl_queue=sycl_queue, device=device)
     usm_type = usm_type if usm_type is not None else "device"
@@ -1480,6 +1496,8 @@ def full_like(
             )
             _manager.add_event_pair(hev, copy_ev)
             return res
+        else:
+            _validate_fill_value(fill_value)
 
         dtype = _get_dtype(dtype, sycl_queue, ref_type=type(fill_value))
         res = _empty_like_orderK(x, dtype, usm_type, sycl_queue)

dpctl/tests/test_usm_ndarray_ctor.py

@@ -2621,3 +2621,14 @@ 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_functions_raise_type_error():
+    get_queue_or_skip()
+
+    with pytest.raises(TypeError):
+        dpt.full(1, "0")
+
+    x = dpt.ones(1, dtype="i4")
+    with pytest.raises(TypeError):
+        dpt.full_like(x, "0")