updating test to be explicit about return values

CDonnerer · Jan 15, 2023 · c72cfa8 · c72cfa8
1 parent 9ace51c
commit c72cfa8
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 32 deletions.
diff --git a/tests/distributions/test_exponential.py b/tests/distributions/test_exponential.py
@@ -56,11 +56,4 @@ def test_loss(exponential):
         params=np.array([np.log(1), ]),
     )
     assert loss_name == "Exponential-NLL"
-    np.testing.assert_array_equal(
-        loss_values,
-        np.array(
-            [
-                1.0,
-            ]
-        ),
-    )
+    np.testing.assert_array_equal(loss_values, np.array([1.0]))
diff --git a/tests/distributions/test_laplace.py b/tests/distributions/test_laplace.py
@@ -41,11 +41,4 @@ def test_loss(laplace):
         params=np.array([[1, np.log(1)], ]),
     )
     assert loss_name == "Laplace-NLL"
-    np.testing.assert_approx_equal(
-        loss_values,
-        np.array(
-            [
-                -np.log(0.5),
-            ]
-        ),
-    )
+    np.testing.assert_approx_equal(loss_values, np.array([-np.log(0.5)]))
diff --git a/tests/distributions/test_log_normal.py b/tests/distributions/test_log_normal.py
@@ -56,11 +56,4 @@ def test_loss(lognormal):
         params=np.array([[1, 0], ]),
     )
     assert loss_name == "LogNormal-NLL"
-    np.testing.assert_array_equal(
-        loss_values,
-        np.array(
-            [
-                np.inf,
-            ]
-        ),
-    )
+    np.testing.assert_array_equal(loss_values, np.array([np.inf]))
diff --git a/tests/distributions/test_poisson.py b/tests/distributions/test_poisson.py
@@ -54,11 +54,4 @@ def test_loss(poisson):
         params=np.array([np.log(1), ]),
     )
     assert loss_name == "Poisson-NLL"
-    np.testing.assert_array_equal(
-        loss_values,
-        np.array(
-            [
-                1.0,
-            ]
-        ),
-    )
+    np.testing.assert_array_equal(loss_values, np.array([1.0]))