Skip to content

Commit

Permalink
simplify jet tests
Browse files Browse the repository at this point in the history
  • Loading branch information
Krastanov committed Sep 12, 2024
1 parent a0aa967 commit a053322
Showing 1 changed file with 2 additions and 19 deletions.
21 changes: 2 additions & 19 deletions test/test_jet.jl
Original file line number Diff line number Diff line change
Expand Up @@ -4,32 +4,15 @@ using Test

import LinearAlgebra, DelimitedFiles

using JET: ReportPass, BasicPass, InferenceErrorReport, UncaughtExceptionReport

# Custom report pass that ignores `UncaughtExceptionReport`
# Too coarse currently, but it serves to ignore the various
# "may throw" messages for runtime errors we raise on purpose
# (mostly on malformed user input)
struct MayThrowIsOk <: ReportPass end

# ignores `UncaughtExceptionReport` analyzed by `JETAnalyzer`
(::MayThrowIsOk)(::Type{UncaughtExceptionReport}, @nospecialize(_...)) = return

# forward to `BasicPass` for everything else
function (::MayThrowIsOk)(report_type::Type{<:InferenceErrorReport}, @nospecialize(args...))
BasicPass()(report_type, args...)
end

@testset "JET checks" begin
rep = report_package("LDPCDecoders";
report_pass=MayThrowIsOk(),
ignored_modules=(
AnyFrameModule(LinearAlgebra),
AnyFrameModule(DelimitedFiles),
AnyFrameModule(Base.Broadcast),
)
)
@show rep
@test_broken length(JET.get_reports(rep)) == 0
@test length(JET.get_reports(rep)) <= 2
#@test_broken length(JET.get_reports(rep)) == 0
@test length(JET.get_reports(rep)) == 0
end

0 comments on commit a053322

Please sign in to comment.