Skip to content

Commit 7bc67ba

Browse files
committed
feat: extend Base.parentmodule for Model
1 parent 5132be6 commit 7bc67ba

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
@@ -577,3 +577,15 @@ end
577577
@test getdefault(component.written_out_for_1.sc) == 2
578578
@test getdefault(component.written_out_for_2.sc) == 3
579579
end
580+
581+
module GetParentModule
582+
583+
using ModelingToolkit
584+
585+
@mtkmodel Component begin end
586+
587+
end
588+
589+
@testset "Parent module of Models" begin
590+
@test parentmodule(GetParentModule.Component) == GetParentModule
591+
end

0 commit comments

Comments
 (0)