tests/functions: Move test_Stability_squeezes_variable to test_stability

Signed-off-by: Jan Vesely <[email protected]>
jvesely · Nov 7, 2023 · 8685f59 · 8685f59
1 parent 77557ad
commit 8685f59
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 9 deletions.
diff --git a/tests/functions/test_objective.py b/tests/functions/test_objective.py
diff --git a/tests/functions/test_stability.py b/tests/functions/test_stability.py
@@ -1,4 +1,3 @@
-
 import numpy as np
 
 import psyneulink.core.llvm as pnlvm
@@ -43,3 +42,10 @@ def test_basic(variable, metric, normalize, expected, benchmark, func_mode):
     res = benchmark(EX, variable)
     np.testing.assert_allclose(res, expected)
     assert np.isscalar(res) or len(res) == 1
+
+
+def test_Stability_squeezes_variable():
+    s1 = Functions.Stability(default_variable=[[0.0, 0.0]])
+    s2 = Functions.Stability(default_variable=[0.0, 0.0])
+
+    assert s1.execute([5, 5]) == s2.execute([5, 5])