Function _format_values

dali/test/python/test_tensor_formatting.py:108–113

Format array values as strings matching the output format.

(arr, count=2)

def _format_values(arr, count=2):
    """Format array values as strings matching the output format."""
    return [
        np.format_float_positional(v, trim="-") if arr.dtype.kind == "f" else str(int(v))
        for v in arr.flat[:count]
    ]


@params(*product(TENSOR_TEST_CASES, DEVICES))