test: test solving ODEProblem with eval_expression = true

Author: AayushSabharwal

test/precompile_test.jl

@@ -1,5 +1,6 @@
 using Test
 using ModelingToolkit
+using OrdinaryDiffEqDefault
 
 using Distributed
 
@@ -38,3 +39,5 @@ ODEPrecompileTest.f_eval_bad(u, p, 0.1)
 @test parentmodule(typeof(ODEPrecompileTest.f_eval_good.f.f_oop)) ==
       ODEPrecompileTest
 @test ODEPrecompileTest.f_eval_good(u, p, 0.1) == [4, 0, -16]
+
+@test_nowarn solve(ODEPrecompileTest.prob_eval)

test/precompile_test/ODEPrecompileTest.jl

@@ -36,4 +36,24 @@ const f_eval_bad = system(; eval_expression = true, eval_module = @__MODULE__)
 # Change the module the eval'd function is eval'd into to be the containing module,
 # which should make it be in the package image
 const f_eval_good = system(; eval_expression = true, eval_module = @__MODULE__)
+
+function problem(; kwargs...)
+    # Define some variables
+    @independent_variables t
+    @parameters σ ρ β
+    @variables x(t) y(t) z(t)
+    D = Differential(t)
+
+    # Define a differential equation
+    eqs = [D(x) ~ σ * (y - x),
+        D(y) ~ x * (ρ - z) - y,
+        D(z) ~ x * y - β * z]
+
+    @named de = System(eqs, t)
+    de = complete(de)
+    return ODEProblem(de, [x => 1, y => 0, z => 0, σ => 10, ρ => 28, β => 8/3], (0.0, 5.0); kwargs...)
+end
+
+const prob_eval = problem(; eval_expression = true, eval_module = @__MODULE__)
+
 end