def test_items(x):
""" verify that tensor.item() mirrors array.item()"""
tensor = Tensor(x)
try:
value = x.item()
assert_allclose(value, tensor.item())
except ValueError:
with raises(ValueError):
tensor.item()
def test_items(x):
""" verify that tensor.item() mirrors array.item()"""
tensor = Tensor(x)
try:
value = np.asarray(x).item()
assert_array_equal(value, tensor.item())
except ValueError:
with raises(ValueError):
tensor.item()
def test_to_scalar():
nd_tensor = Tensor([1, 2])
with raises(TypeError):
float(nd_tensor)
with raises(TypeError):
int(nd_tensor)
with raises(ValueError):
nd_tensor.item()
for size1_tensor in (Tensor(1), Tensor([[1]])):
assert float(size1_tensor) == 1.0
assert int(size1_tensor) == 1
assert size1_tensor.item() == 1.0
