From 5884e7cf74069b609720d867de77fb6a229dd6b1 Mon Sep 17 00:00:00 2001
From: kaandocal <26488673+kaandocal@users.noreply.github.com>
Date: Wed, 24 Apr 2024 10:42:06 +1000
Subject: [PATCH] Fixing unit tests

---
 test/birthdeath2D.jl | 4 ++--
 test/feedbackloop.jl | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/test/birthdeath2D.jl b/test/birthdeath2D.jl
index ce92d4c..07fa45c 100644
--- a/test/birthdeath2D.jl
+++ b/test/birthdeath2D.jl
@@ -41,10 +41,10 @@ sol = solve(prob, Vern7(), abstol=1e-6, saveat=tt)
 @test marg(sol.u[3], dims=1) ≈ pdf.(Poisson(ps[3] / ps[4] * (1 - exp(-ps[4] * tt[3]))), 0:Nmax) atol=1e-4
 
 A = convert(SparseMatrixCSC, sys, (Nmax+1, Nmax+1), ps, 0)
-f = (du,u,t) -> mul!(du, A, u)
+f = (du,u,p,t) -> mul!(vec(du), A, vec(u))
 
 probA = ODEProblem(f, u0, 10.0)
-solA = solve(prob, Vern7(), abstol=1e-6, saveat=tt)
+solA = solve(probA, Vern7(), abstol=1e-6, saveat=tt)
 
 @test sol.u[1] ≈ solA.u[1] atol=1e-4
 @test sol.u[2] ≈ solA.u[2] atol=1e-4
diff --git a/test/feedbackloop.jl b/test/feedbackloop.jl
index 973e7ac..e944fd2 100644
--- a/test/feedbackloop.jl
+++ b/test/feedbackloop.jl
@@ -49,10 +49,10 @@ u0[1] = 1.0
 prob = convert(ODEProblem, sys, u0, maximum(tt), ps)
 sol = solve(prob, Vern7(), abstol=1e-6, saveat=tt)
 
-f = (du,u,t) -> mul!(du, A, u)
+f = (du,u,p,t) -> mul!(vec(du), A, vec(u))
 
 probA = ODEProblem(f, u0, 10.0)
-solA = solve(prob, Vern7(), abstol=1e-6, saveat=tt)
+solA = solve(probA, Vern7(), abstol=1e-6, saveat=tt)
 
 @test sol.u[1] ≈ solA.u[1] atol=1e-4
 @test sol.u[2] ≈ solA.u[2] atol=1e-4