Test Bautin with NormalizingAS in test_examples.jl

tkf · Jul 24, 2018 · d19fd71 · d19fd71
1 parent 3ab1873
commit d19fd71
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 3 deletions.
diff --git a/test/test_bautin.jl b/test/test_bautin.jl
@@ -2,7 +2,6 @@ module TestBautin
 include("preamble_plots.jl")
 
 using Bifurcations: Codim1, Codim2, resolved_points
-using Bifurcations.Codim2: NormalizingAS
 using Bifurcations.Codim2LimitCycle: FoldLimitCycleProblem
 using Bifurcations.Examples: Bautin
 
@@ -19,7 +18,6 @@ hopf_prob = BifurcationProblem(
     solver1,
     (@lens _.β₂),
     (-2.0, 2.0),
-    augmented_system = NormalizingAS(),
 )
 hopf_solver = init(
     hopf_prob;

diff --git a/test/test_examples.jl b/test/test_examples.jl
@@ -4,6 +4,7 @@ include("preamble.jl")
 using DiffEqBase: remake
 
 using Bifurcations: Codim1, resolved_points
+using Bifurcations.Codim2: BackReferencingAS, NormalizingAS
 using Bifurcations.Continuations: find_errors, print_errors
 using Bifurcations: examples
 using Bifurcations.Examples: Calcium
@@ -20,7 +21,6 @@ end
 make_codim2(::Real, point, solver1) = []
 make_codim2(::Tuple, point, solver1) = []  # PredatorPrey
 make_codim2(::DuffingVanDerPolParam, point, solver1) = []
-make_codim2(::BautinParam, point, solver1) = []  # TODO: define something
 make_codim2(::CuspParam, point, solver1) = []  # TODO: define something
 
 make_codim2(::CalciumParam, point, solver1) = [
@@ -45,6 +45,25 @@ function make_codim2(::Bazykin85Param, point, solver1)
     )]
 end
 
+function make_codim2(::BautinParam, point, solver1)
+    args = (
+        point,
+        solver1,
+        (@lens _.β₂),
+        (-2.0, 2.0),
+    )
+    return [
+        BifurcationProblem(
+            args...;
+            augmented_system = BackReferencingAS(),
+        ),
+        BifurcationProblem(
+            args...;
+            augmented_system = NormalizingAS(),
+        ),
+    ]
+end
+
 EXAMPLES = [
     examples()...,
     :CalciumSTDArray => Example(Calcium.make_prob(