make tests green

ReactiveBayes · Jun 10, 2024 · 2337aad · 2337aad
1 parent 5e74d19
commit 2337aad
Show file tree

Hide file tree

Showing 5 changed files with 27 additions and 2 deletions.
diff --git a/test/manopt_setuptests.jl b/test/manopt_setuptests.jl
@@ -0,0 +1,25 @@
+using Manopt
+
+# Non allocating version of the same `StopWhenGradientNormLess` from `Manopt.jl`
+struct StopWhenGradientNormLessNonAllocating{F} <: StoppingCriterion
+    threshold::F
+end
+
+function (sc::StopWhenGradientNormLessNonAllocating)(mp, s, i)
+    M = get_manifold(mp)
+    if (i > 0)
+        grad_norm = norm(M, get_iterate(s), get_gradient(s))
+        if grad_norm < sc.threshold
+            return true
+        end
+    end
+    return false
+end
+
+# Non allocating version of the same `ConstantStepsize` from `Manopt.jl`
+struct ConstantStepsizeNonAllocating{T} <: Stepsize
+    stepsize::T
+end
+function (cs::ConstantStepsizeNonAllocating)(args...; kwargs...)
+    return cs.stepsize
+end
diff --git a/test/manifolds/natural_manifolds_tests.jl → test/natural_manifolds_tests.jl b/test/manifolds/natural_manifolds_tests.jl → test/natural_manifolds_tests.jl
diff --git a/...folds/negative_definite_matrices_tests.jl → test/negative_definite_matrices_tests.jl b/...folds/negative_definite_matrices_tests.jl → test/negative_definite_matrices_tests.jl
diff --git a/...nifolds/shifted_negative_numbers_tests.jl → test/shifted_negative_numbers_tests.jl b/...nifolds/shifted_negative_numbers_tests.jl → test/shifted_negative_numbers_tests.jl
@@ -200,7 +200,7 @@ end
 
     import ExponentialFamilyManifolds: ShiftedNegativeNumbers
 
-    include("../manopt_setuptests.jl")
+    include("manopt_setuptests.jl")
 
     a = 2.0
     b = 10.0

diff --git a/...nifolds/shifted_positive_numbers_tests.jl → test/shifted_positive_numbers_tests.jl b/...nifolds/shifted_positive_numbers_tests.jl → test/shifted_positive_numbers_tests.jl
@@ -198,7 +198,7 @@ end
 
     import ExponentialFamilyManifolds: ShiftedPositiveNumbers
 
-    include("../manopt_setuptests.jl")
+    include("manopt_setuptests.jl")
 
     a = 2.0
     b = -10.0