Throw error in trixi_include when assignment is not found

src/auxiliary/special_elixirs.jl

@@ -20,7 +20,7 @@ providing examples with sensible default values for users.
 
 Before replacing assignments in `elixir`, the keyword argument `maxiters` is inserted
 into calls to `solve` and `Trixi.solve` with it's default value used in the SciML ecosystem
-for ODEs, see the "Miscellaneous" section of the 
+for ODEs, see the "Miscellaneous" section of the
 [documentation](https://docs.sciml.ai/DiffEqDocs/stable/basics/common_solver_opts/).
 
 # Examples
@@ -36,6 +36,16 @@ julia> redirect_stdout(devnull) do
 ```
 """
 function trixi_include(mod::Module, elixir::AbstractString; kwargs...)
+    # Check that all kwargs exist as assignments
+    code = read(elixir, String)
+    expr = Meta.parse("begin \n$code \nend")
+    expr = insert_maxiters(expr)
+
+    for (key, val) in kwargs
+        # This will throw an error when `key` is not found
+        find_assignment(expr, key)
+    end
+
     # Print information on potential wait time only in non-parallel case
     if !mpi_isparallel()
         @info "You just called `trixi_include`. Julia may now compile the code, please be patient."
@@ -243,19 +253,25 @@ end
 function find_assignment(expr, destination)
     # declare result to be able to assign to it in the closure
     local result
+    found = false
 
     # find explicit and keyword assignments
     walkexpr(expr) do x
         if x isa Expr
             if (x.head === Symbol("=") || x.head === :kw) &&
                x.args[1] === Symbol(destination)
                 result = x.args[2]
+                found = true
                 # dump(x)
             end
         end
         return x
     end
 
+    if !found
+        throw(ArgumentError("assignment `$destination` not found in expression"))
+    end
+
     result
 end
 
@@ -274,7 +290,7 @@ function extract_initial_resolution(elixir, kwargs)
             return initial_refinement_level
         end
     catch e
-        if isa(e, UndefVarError)
+        if isa(e, ArgumentError)
             # get cells_per_dimension from the elixir
             cells_per_dimension = eval(find_assignment(expr, :cells_per_dimension))
 

test/test_trixi.jl

@@ -33,7 +33,7 @@ macro test_trixi_include(elixir, args...)
     local kwargs = Pair{Symbol, Any}[]
     for arg in args
         if (arg.head == :(=) &&
-            !(arg.args[1] in (:l2, :linf, :atol, :rtol, :coverage_override))
+            !(arg.args[1] in (:l2, :linf, :atol, :rtol, :coverage_override, :skip_coverage))
             && !(coverage && arg.args[1] in keys(coverage_override)))
             push!(kwargs, Pair(arg.args...))
         end

test/test_unit.jl

@@ -1325,6 +1325,103 @@ end
 
     @test mesh.boundary_faces[:entire_boundary] == [1, 2]
 end
+
+@testset "trixi_include" begin
+    @trixi_testset "Basic" begin
+        example = """
+            x = 4
+            """
+
+        filename = tempname()
+        try
+            open(filename, "w") do file
+                write(file, example)
+            end
+
+            # Use `@trixi_testset`, which wraps code in a temporary module, and call
+            # `trixi_include` with `@__MODULE__` in order to isolate this test.
+            @test_warn "You just called" trixi_include(@__MODULE__, filename)
+            @test @isdefined x
+            @test x == 4
+
+            @test_warn "You just called" trixi_include(@__MODULE__, filename, x = 7)
+            @test x == 7
+
+            @test_throws "assignment `y` not found in expression" trixi_include(@__MODULE__,
+                                                                                filename,
+                                                                                y = 3)
+        finally
+            rm(filename, force = true)
+        end
+    end
+
+    @trixi_testset "With `solve` Without `maxiters`" begin
+        # `trixi_include` assumes this to be the `solve` function of OrdinaryDiffEq,
+        # and therefore tries to insert the kwarg `maxiters`, which will fail here.
+        example = """
+            solve() = 0
+            x = solve()
+            """
+
+        filename = tempname()
+        try
+            open(filename, "w") do file
+                write(file, example)
+            end
+
+            # Use `@trixi_testset`, which wraps code in a temporary module, and call
+            # `trixi_include` with `@__MODULE__` in order to isolate this test.
+            @test_throws "no method matching solve(; maxiters::Int64)" trixi_include(@__MODULE__,
+                                                                                     filename)
+
+            @test_throws "no method matching solve(; maxiters::Int64)" trixi_include(@__MODULE__,
+                                                                                     filename,
+                                                                                     maxiters = 3)
+        finally
+            rm(filename, force = true)
+        end
+    end
+
+    @trixi_testset "With `solve` with `maxiters`" begin
+        # We need another example file that we include with `Base.include` first, in order to
+        # define the `solve` method without `trixi_include` trying to insert `maxiters` kwargs.
+        # Then, we can test that `trixi_include` inserts the kwarg in the `solve()` call.
+        example1 = """
+            solve(; maxiters=0) = maxiters
+            """
+
+        example2 = """
+            x = solve()
+            """
+
+        filename1 = tempname()
+        filename2 = tempname()
+        try
+            open(filename1, "w") do file
+                write(file, example1)
+            end
+            open(filename2, "w") do file
+                write(file, example2)
+            end
+
+            # Use `@trixi_testset`, which wraps code in a temporary module, and call
+            # `Base.include` and `trixi_include` with `@__MODULE__` in order to isolate this test.
+            Base.include(@__MODULE__, filename1)
+            @test_warn "You just called" trixi_include(@__MODULE__, filename2)
+            @test @isdefined x
+            # This is the default `maxiters` inserted by `trixi_include`
+            @test x == 10^5
+
+            @test_warn "You just called" trixi_include(@__MODULE__, filename2,
+                                                       maxiters = 7)
+            # Test that `maxiters` got overwritten
+            @test x == 7
+        finally
+            rm(filename1, force = true)
+            rm(filename2, force = true)
+        end
+    end
+end
 end
 
 end #module

test/test_visualization.jl

@@ -243,7 +243,6 @@ end
     @test_nowarn_mod trixi_include(@__MODULE__,
                                    joinpath(examples_dir(), "tree_2d_dgsem",
                                             "elixir_advection_basic.jl"),
-                                   tspan = (0, 0.1),
                                    analysis_callback = Trixi.TrivialCallback())
     @test adapt_to_mesh_level(sol, 5) isa Tuple
 
@@ -259,7 +258,6 @@ end
     @test_nowarn_mod trixi_include(@__MODULE__,
                                    joinpath(examples_dir(), "tree_3d_dgsem",
                                             "elixir_advection_basic.jl"),
-                                   tspan = (0, 0.1),
                                    analysis_callback = Trixi.TrivialCallback(),
                                    initial_refinement_level = 1)
     @test PlotData2D(sol) isa Trixi.PlotData2DCartesian