Skip to content

Commit 53fabfc

Browse files
committed
feat: extend Base.parentmodule for Model
1 parent 86d345f commit 53fabfc

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
@@ -688,3 +688,15 @@ end
688688
@test getdefault(component.written_out_for_1.sc) == 2
689689
@test getdefault(component.written_out_for_2.sc) == 3
690690
end
691+
692+
module GetParentModule
693+
694+
using ModelingToolkit
695+
696+
@mtkmodel Component begin end
697+
698+
end
699+
700+
@testset "Parent module of Models" begin
701+
@test parentmodule(GetParentModule.Component) == GetParentModule
702+
end

0 commit comments

Comments
 (0)