Skip to content

Commit 863ca9e

Browse files
committed
Fix tests
1 parent 8599849 commit 863ca9e

File tree

2 files changed

+8
-7
lines changed

2 files changed

+8
-7
lines changed

src/systems/abstractsystem.jl

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -513,6 +513,7 @@ function toexpr(sys::AbstractSystem)
513513
expr = Expr(:block)
514514
stmt = expr.args
515515

516+
name = Meta.quot(nameof(sys))
516517
iv = independent_variable(sys)
517518
ivname = gensym(:iv)
518519
if iv !== nothing
@@ -535,9 +536,9 @@ function toexpr(sys::AbstractSystem)
535536
defs_name = push_defaults!(stmt, defaults(sys), var2name)
536537

537538
if sys isa ODESystem
538-
push!(stmt, :($ODESystem($eqs_name, $ivname, $stsname, $psname; defaults=$defs_name)))
539+
push!(stmt, :($ODESystem($eqs_name, $ivname, $stsname, $psname; defaults=$defs_name, name=$name)))
539540
elseif sys isa NonlinearSystem
540-
push!(stmt, :($NonlinearSystem($eqs_name, $stsname, $psname; defaults=$defs_name)))
541+
push!(stmt, :($NonlinearSystem($eqs_name, $stsname, $psname; defaults=$defs_name, name=$name)))
541542
end
542543

543544
striplines(expr) # keeping the line numbers is never helpful

test/odesystem.jl

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,7 @@ eqs = [D(x) ~ σ*(y-x),
1717
D(z) ~ x*y - β*z]
1818

1919
ModelingToolkit.toexpr.(eqs)[1]
20-
de = ODESystem(eqs; defaults=Dict(x => 1))
20+
@named de = ODESystem(eqs; defaults=Dict(x => 1))
2121
@test eval(toexpr(de)) == de
2222

2323
generate_function(de)
@@ -389,17 +389,17 @@ let
389389
end
390390

391391
# Array vars
392-
using Symbolics: unwrap
392+
using Symbolics: unwrap, wrap
393393
using LinearAlgebra
394394
@variables t
395395
sts = @variables x[1:3](t) y(t)
396-
ps = @parameters p[1:3]
396+
ps = @parameters p[1:3] = [1, 2, 3]
397397
D = Differential(t)
398398
eqs = [
399-
collect(D(x) ~ x)
399+
collect(D.(x) ~ x)
400400
D(y) ~ norm(x)*y
401401
]
402-
sys = ODESystem(eqs, t, [sts...;], [ps...;])
402+
@named sys = ODESystem(eqs, t, [sts...;], [ps...;])
403403
@test isequal(@nonamespace(sys.x), unwrap(x))
404404
@test isequal(@nonamespace(sys.y), unwrap(y))
405405
@test isequal(@nonamespace(sys.p), unwrap(p))

0 commit comments

Comments
 (0)