pass rhs! instead of mod

Author: JoshuaLampert

src/macros.jl

@@ -194,17 +194,17 @@ macro trixi_testset(name, expr)
 end
 
 """
-    @test_allocations(mod, semi, sol, allocs)
+    @test_allocations(rhs!, semi, sol, allocs)
 
-Test that the memory allocations of `mod.rhs!` are below `allocs`
-(e.g., from type instabilities), where `mod` is a module containing
+Test that the memory allocations of `rhs!` are below `allocs`
+(e.g., from type instabilities), where `rhs!` is a function with
 a method `rhs!(du, u, semi, t)`.
 """
-macro test_allocations(mod, semi, sol, allocs)
+macro test_allocations(rhs!, semi, sol, allocs)
     quote
         t = $(esc(sol)).t[end]
         u = $(esc(sol)).u[end]
         du = similar(u)
-        @test (@allocated $(esc(mod)).rhs!(du, u, $(esc(semi)), t)) < $(esc(allocs))
+        @test (@allocated $(esc(rhs!))(du, u, $(esc(semi)), t)) < $(esc(allocs))
     end
 end

test/test_test_allocations.jl

@@ -10,5 +10,5 @@ end
     semi = nothing
     sol = (t = [1.0], u = [[1.0, 2.0]])
     allocs = 1
-    @test_allocations(TestAllocations, semi, sol, allocs)
+    @test_allocations(TestAllocations.rhs!, semi, sol, allocs)
 end