Skip to content

Commit 1209e41

Browse files
committed
add tests for ODEs, SDEs and NLS
1 parent 3160770 commit 1209e41

File tree

1 file changed

+26
-2
lines changed

1 file changed

+26
-2
lines changed

test/dep_graphs.jl

Lines changed: 26 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ rs = ReactionSystem(rxs, t, [S,I,R], [k1,k2])
1414

1515

1616
#################################
17-
# testing for Jumps
17+
# testing for Jumps / all dgs
1818
#################################
1919
js = convert(JumpSystem, rs)
2020
S = convert(Variable,S); I = convert(Variable,I); R = convert(Variable,R)
@@ -71,4 +71,28 @@ end
7171
dg4 = varvar_dependencies(depsbg,deps2)
7272
@test dg == dg4
7373

74-
74+
#####################################
75+
# testing for ODE/SDEs
76+
#####################################
77+
os = convert(ODESystem, rs)
78+
deps = equation_dependencies(os)
79+
eq_sdeps = [[S,I], [S,I], [S,I,R]]
80+
@test all(i -> isequal(Set(eq_sdeps[i]),Set(deps[i])), 1:length(deps))
81+
82+
sdes = convert(ODESystem, rs)
83+
deps = equation_dependencies(sdes)
84+
@test all(i -> isequal(Set(eq_sdeps[i]),Set(deps[i])), 1:length(deps))
85+
86+
#####################################
87+
# testing for nonlin sys
88+
#####################################
89+
@variables x y z
90+
@parameters σ ρ β
91+
92+
eqs = [0 ~ σ*(y-x),
93+
0 ~ ρ-y,
94+
0 ~ y - β*z]
95+
ns = NonlinearSystem(eqs, [x,y,z],[σ,ρ,β])
96+
deps = equation_dependencies(ns)
97+
eq_sdeps = [[x,y],[y],[y,z]]
98+
@test all(i -> isequal(Set(deps[i]),Set(convert.(Variable,eq_sdeps[i]))), 1:length(deps))

0 commit comments

Comments
 (0)