From d19fd718b73f7a6b38e3bf23a0de81d2f76ac48a Mon Sep 17 00:00:00 2001
From: Takafumi Arakaki <aka.tkf@gmail.com>
Date: Mon, 23 Jul 2018 18:27:04 -0700
Subject: [PATCH] Test Bautin with NormalizingAS in test_examples.jl

---
 test/test_bautin.jl   |  2 --
 test/test_examples.jl | 21 ++++++++++++++++++++-
 2 files changed, 20 insertions(+), 3 deletions(-)

diff --git a/test/test_bautin.jl b/test/test_bautin.jl
index 72d2eb1..42700cb 100644
--- 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
index 5318f8a..17cc4a8 100644
--- 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(