updated tests

CDonnerer · Apr 2, 2022 · afb2fb1 · afb2fb1
1 parent f64a43a
commit afb2fb1
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/tests/distributions/test_exponential.py b/tests/distributions/test_exponential.py
@@ -55,5 +55,5 @@ def test_loss(exponential):
         y=np.array([1, ]),
         params=np.array([[np.log(1)], ]),
     )
-    assert loss_name == "ExponentialError"
+    assert loss_name == "Exponential-NLL"
     assert loss_value == 1.0
diff --git a/tests/distributions/test_laplace.py b/tests/distributions/test_laplace.py
@@ -40,5 +40,5 @@ def test_loss(laplace):
         y=np.array([1, ]),
         params=np.array([[1, np.log(1)], ]),
     )
-    assert loss_name == "LaplaceError"
+    assert loss_name == "Laplace-NLL"
     np.testing.assert_approx_equal(loss_value, -np.log(0.5))
diff --git a/tests/distributions/test_log_normal.py b/tests/distributions/test_log_normal.py
@@ -55,5 +55,5 @@ def test_loss(lognormal):
         y=np.array([0, ]),
         params=np.array([[1, 0], ]),
     )
-    assert loss_name == "LogNormalError"
+    assert loss_name == "LogNormal-NLL"
     assert loss_value == np.inf
diff --git a/tests/distributions/test_poisson.py b/tests/distributions/test_poisson.py
@@ -55,5 +55,5 @@ def test_loss(poisson):
         y=np.array([1, ]),
         params=np.array([[np.log(1)], ]),
     )
-    assert loss_name == "PoissonError"
+    assert loss_name == "Poisson-NLL"
     assert loss_value == 1.0