feat: extend Base.parentmodule for Model

SciML · Dec 5, 2023 · 3335751 · 3335751
1 parent d691373
commit 3335751
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/src/systems/model_parsing.jl b/src/systems/model_parsing.jl
@@ -24,6 +24,8 @@ struct Model{F, S}
 end
 (m::Model)(args...; kw...) = m.f(args...; kw...)
 
+Base.parentmodule(m::Model) = parentmodule(m.f)
+
 for f in (:connector, :mtkmodel)
     isconnector = f == :connector ? true : false
     @eval begin

diff --git a/test/model_parsing.jl b/test/model_parsing.jl
@@ -546,3 +546,15 @@ end
     @test getdefault(component.written_out_for_1.sc) == 2
     @test getdefault(component.written_out_for_2.sc) == 3
 end
+
+module GetParentModule
+
+using ModelingToolkit
+
+@mtkmodel Component begin end
+
+end
+
+@testset "Parent module of Models" begin
+    @test parentmodule(GetParentModule.Component) == GetParentModule
+end