From 8eac9a4bb5b497ca29ebb852f21169ecfd0191e1 Mon Sep 17 00:00:00 2001 From: Matthew Barber Date: Mon, 24 Jan 2022 10:20:20 +0000 Subject: BUG: Fix `np.array_api.can_cast()` by not relying on `np.can_cast()` --- numpy/array_api/tests/test_validation.py | 27 +++++++++++++++++++++++++++ 1 file changed, 27 insertions(+) create mode 100644 numpy/array_api/tests/test_validation.py (limited to 'numpy/array_api/tests/test_validation.py') diff --git a/numpy/array_api/tests/test_validation.py b/numpy/array_api/tests/test_validation.py new file mode 100644 index 000000000..0dd100d15 --- /dev/null +++ b/numpy/array_api/tests/test_validation.py @@ -0,0 +1,27 @@ +from typing import Callable + +import pytest + +from numpy import array_api as xp + + +def p(func: Callable, *args, **kwargs): + f_sig = ", ".join( + [str(a) for a in args] + [f"{k}={v}" for k, v in kwargs.items()] + ) + id_ = f"{func.__name__}({f_sig})" + return pytest.param(func, args, kwargs, id=id_) + + +@pytest.mark.parametrize( + "func, args, kwargs", + [ + p(xp.can_cast, 42, xp.int8), + p(xp.can_cast, xp.int8, 42), + p(xp.result_type, 42), + ], +) +def test_raises_on_invalid_types(func, args, kwargs): + """Function raises TypeError when passed invalidly-typed inputs""" + with pytest.raises(TypeError): + func(*args, **kwargs) -- cgit v1.2.1