Skip to content

Commit ea3477b

Browse files
committed
feat: extend Base.parentmodule for Model
1 parent 5fa720c commit ea3477b

File tree

2 files changed

+14
-0
lines changed

2 files changed

+14
-0
lines changed

src/systems/model_parsing.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,8 @@ struct Model{F, S}
2424
end
2525
(m::Model)(args...; kw...) = m.f(args...; kw...)
2626

27+
Base.parentmodule(m::Model) = parentmodule(m.f)
28+
2729
for f in (:connector, :mtkmodel)
2830
isconnector = f == :connector ? true : false
2931
@eval begin

test/model_parsing.jl

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -589,3 +589,15 @@ end
589589
@test getdefault(component.written_out_for_1.sc) == 2
590590
@test getdefault(component.written_out_for_2.sc) == 3
591591
end
592+
593+
module GetParentModule
594+
595+
using ModelingToolkit
596+
597+
@mtkmodel Component begin end
598+
599+
end
600+
601+
@testset "Parent module of Models" begin
602+
@test parentmodule(GetParentModule.Component) == GetParentModule
603+
end

0 commit comments

Comments
 (0)