JET @report_opt for modified test/Williamson2ThetaMethodFullNewtonTests.jl

[goerch]

To run @report_opt I had to modify the test from here to look like

module Williamsom2ThetaMethodFullNewtonTests

using Test
using Gridap
using GridapGeosciences
using SparseMatricesCSR

# Solves the steady state Williamson2 test case for the shallow water equations on a sphere
# of physical radius 6371220m. Involves a modified coriolis term that exactly balances
# the potential gradient term to achieve a steady state
# reference:
# D. L. Williamson, J. B. Drake, J. J.HackRüdiger Jakob, P. N.Swarztrauber, (1992)
# J Comp. Phys. 102 211-224

include("Williamson2InitialConditions.jl")

function test()
  l2_err_u = [0.011370921987771046 , 0.002991229234176333 ]
  l2_err_h = [0.005606685579166809, 0.001458107077200681 ]

  order=1
  degree=4
  θ=0.5
  for i in 1:2
    n      = 2*2^i
    nstep  = 5*n
    Uc     = sqrt(g*H₀)
    dx     = 2.0*π*rₑ/(4*n)
    dt     = 0.25*dx/Uc
    println("timestep: ", dt)   # gravity wave time step
    T      = dt*nstep
    τ      = dt/2
    model = CubedSphereDiscreteModel(n; radius=rₑ)
    hf, uf = shallow_water_theta_method_full_newton_time_stepper(model, order, degree,
                                                                 h₀, u₀, f₀, topography, g, θ, T, nstep, τ;
                                                                 write_solution=false,
                                                                 write_solution_freq=5,
                                                                 write_diagnostics=true,
                                                                 write_diagnostics_freq=1,
                                                                 dump_diagnostics_on_screen=true)

    Ω     = Triangulation(model)
    dΩ    = Measure(Ω, degree)
    hc    = CellField(h₀, Ω)
    e     = h₀-hf
    err_h = sqrt(sum(∫(e⋅e)*dΩ))/sqrt(sum(∫(hc⋅hc)*dΩ))
    uc    = CellField(u₀, Ω)
    e     = u₀-uf
    err_u = sqrt(sum(∫(e⋅e)*dΩ))/sqrt(sum(∫(uc⋅uc)*dΩ))
    println("n=", n, ",\terr_u: ", err_u, ",\terr_h: ", err_h)

    #@test abs(err_u - l2_err_u[i]) < 10.0^-12
    #@test abs(err_h - l2_err_h[i]) < 10.0^-12
  end
end

end # module

Executing @report_opt Main.Williamsom2ThetaMethodFullNewtonTests.test() then generated the attached file.

JET_report_opt_Williamson2ThetaMethodFullNewtonTests.txt