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