def test_zero_div():
mse = MeanSquaredError()
with pytest.raises(
NotComputableError,
match=
r"MeanSquaredError must have at least one example before it can be computed"
):
mse.compute()
def test_compute():
mse = MeanSquaredError()
y_pred = torch.Tensor([[2.0], [-2.0]])
y = torch.zeros(2)
mse.update((y_pred, y))
assert mse.compute() == 4.0
mse.reset()
y_pred = torch.Tensor([[3.0], [-3.0]])
y = torch.zeros(2)
mse.update((y_pred, y))
assert mse.compute() == 9.0
def test_zero_div():
mse = MeanSquaredError()
with pytest.raises(NotComputableError):
mse.compute()
