mirror of
https://github.com/huggingface/candle.git
synced 2025-06-16 18:48:51 +00:00
c05c0a8213
* add `equal` to tensor * add `__richcmp__` support for tensors and scalars * typo * more typos * Add `abs` + `candle.testing` * remove duplicated `broadcast_shape_binary_op` * `candle.i16` => `candle.i64` * `tensor.nelements` -> `tensor.nelement` * Cleanup `abs`
34 lines
1.1 KiB
Python
34 lines
1.1 KiB
Python
import candle
|
|
from candle import Tensor
|
|
from candle.testing import assert_equal, assert_almost_equal
|
|
import pytest
|
|
|
|
|
|
@pytest.mark.parametrize("dtype", [candle.f32, candle.f64, candle.f16, candle.u32, candle.u8, candle.i64])
|
|
def test_assert_equal_asserts_correctly(dtype: candle.DType):
|
|
a = Tensor([1, 2, 3]).to(dtype)
|
|
b = Tensor([1, 2, 3]).to(dtype)
|
|
assert_equal(a, b)
|
|
|
|
with pytest.raises(AssertionError):
|
|
assert_equal(a, b + 1)
|
|
|
|
|
|
@pytest.mark.parametrize("dtype", [candle.f32, candle.f64, candle.f16, candle.u32, candle.u8, candle.i64])
|
|
def test_assert_almost_equal_asserts_correctly(dtype: candle.DType):
|
|
a = Tensor([1, 2, 3]).to(dtype)
|
|
b = Tensor([1, 2, 3]).to(dtype)
|
|
assert_almost_equal(a, b)
|
|
|
|
with pytest.raises(AssertionError):
|
|
assert_almost_equal(a, b + 1)
|
|
|
|
assert_almost_equal(a, b + 1, atol=20)
|
|
assert_almost_equal(a, b + 1, rtol=20)
|
|
|
|
with pytest.raises(AssertionError):
|
|
assert_almost_equal(a, b + 1, atol=0.9)
|
|
|
|
with pytest.raises(AssertionError):
|
|
assert_almost_equal(a, b + 1, rtol=0.1)
|