11using MultiScaleArrays
22using OrdinaryDiffEq, DiffEqBase, Test, StochasticDiffEq, Statistics
3+ using ADTypes
34
45struct Cell{B} <: AbstractMultiScaleArrayLeaf{B}
56 values:: Vector{B}
@@ -64,9 +65,12 @@ prob = ODEProblem(f, embryo, (0.0, 1.0))
6465test_embryo = deepcopy (embryo)
6566
6667sol = solve (prob, Tsit5 (), callback = growing_cb, tstops = tstop)
67- sol = solve (prob, Rosenbrock23 (autodiff = false ), tstops = tstop)
68- sol = solve (prob, Rosenbrock23 (autodiff = false ), callback = growing_cb, tstops = tstop)
69- sol = solve (prob, Rosenbrock23 (chunk_size = 1 ), callback = growing_cb, tstops = tstop)
68+ sol = solve (prob, Rosenbrock23 (autodiff = AutoFiniteDiff ()), tstops = tstop)
69+ sol = solve (prob, Rosenbrock23 (autodiff = AutoFiniteDiff ()), callback = growing_cb, tstops = tstop)
70+ sol = solve (
71+ prob, Rosenbrock23 (autodiff = AutoForwardDiff (chunksize = 1 )), callback = growing_cb,
72+ tstops = tstop
73+ )
7074
7175affect_del! = function (integrator)
7276 return remove_node! (integrator, 1 , 1 , 1 )
@@ -76,9 +80,12 @@ shrinking_cb = DiscreteCallback(condition, affect_del!)
7680
7781sol = solve (prob, Tsit5 (), callback = shrinking_cb, tstops = tstop)
7882
79- sol = solve (prob, Rosenbrock23 (autodiff = false ), callback = shrinking_cb, tstops = tstop)
83+ sol = solve (prob, Rosenbrock23 (autodiff = AutoFiniteDiff () ), callback = shrinking_cb, tstops = tstop)
8084
81- sol = solve (prob, Rosenbrock23 (chunk_size = 1 ), callback = shrinking_cb, tstops = tstop)
85+ sol = solve (
86+ prob, Rosenbrock23 (autodiff = AutoForwardDiff (chunksize = 1 )), callback = shrinking_cb,
87+ tstops = tstop
88+ )
8289
8390@test length (sol[end ]) == 17
8491
0 commit comments