Add JET tests

Author: devmotion

Project.toml

@@ -21,6 +21,7 @@ StatsAPI = "82ae8749-77ed-4fe6-ae5f-f523153014b0"
 AliasTables = "1"
 DataAPI = "1"
 DataStructures = "0.10, 0.11, 0.12, 0.13, 0.14, 0.17, 0.18"
+JET = "0.9.19"
 LinearAlgebra = "<0.0.1, 1"
 LogExpFunctions = "0.3"
 Missings = "0.3, 0.4, 1.0"
@@ -35,9 +36,10 @@ julia = "1.10"
 [extras]
 Dates = "ade2ca70-3891-5945-98fb-dc099432e06a"
 DelimitedFiles = "8bb1440f-4735-579b-a4ab-409b98df4dab"
+JET = "c3a54625-cd67-489e-a8e7-0a5a0ff4e31b"
 OffsetArrays = "6fe1bfb0-de20-5000-8ca7-80f57d26f881"
 StableRNGs = "860ef19b-820b-49d6-a774-d7a799459cd3"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 
 [targets]
-test = ["Dates", "DelimitedFiles", "OffsetArrays", "StableRNGs", "Test"]
+test = ["Dates", "DelimitedFiles", "JET", "OffsetArrays", "StableRNGs", "Test"]

src/cov.jl

@@ -23,11 +23,7 @@ end
 
 ## scatter matrix
 
-_unscaled_covzm(x::DenseMatrix, dims::Colon)   = unscaled_covzm(x)
 _unscaled_covzm(x::DenseMatrix, dims::Integer) = unscaled_covzm(x, dims)
-
-_unscaled_covzm(x::DenseMatrix, wv::AbstractWeights, dims::Colon)   =
-    _symmetrize!(unscaled_covzm(x, _scalevars(x, wv)))
 _unscaled_covzm(x::DenseMatrix, wv::AbstractWeights, dims::Integer) =
     _symmetrize!(unscaled_covzm(x, _scalevars(x, wv, dims), dims))
 

src/deprecates.jl

@@ -5,7 +5,7 @@ if !isdefined(Base, :stderr)
 else
     function (io::typeof(stderr))(obj::StatisticalModel)
         Base.depwarn("stderr(obj::StatisticalModel) is deprecated, use stderror(obj) instead", :stderr)
-        io === stderr ? stderror(obj) : throw(MethodErrror(io, (obj,)))
+        io === stderr ? stderror(obj) : throw(MethodError(io, (obj,)))
     end
 end
 
@@ -22,8 +22,6 @@ end
 @deprecate scattermatm(x::DenseMatrix, mean, wv::AbstractWeights, dims::Int) scattermat(x, wv, mean=mean, dims=dims)
 @deprecate scattermat(x::DenseMatrix, dims::Int) scattermat(x, dims=dims)
 @deprecate scattermat(x::DenseMatrix, wv::AbstractWeights, dims::Int) scattermat(x, wv, dims=dims)
-@deprecate scattermat_zm(x::DenseMatrix, dims::Int) scattermat_zm(x, dims=dims)
-@deprecate scattermat_zm(x::DenseMatrix, wv::AbstractWeights, dims::Int) scattermat_zm(x::DenseMatrix, wv::AbstractWeights, dims=dims)
 @deprecate mean!(R::AbstractArray, A::AbstractArray, w::AbstractWeights, dims::Int) mean!(R, A, w, dims=dims)
 @deprecate mean(A::AbstractArray{T}, w::AbstractWeights{W}, dims::Int) where {T<:Number,W<:Real} mean(A, w, dims=dims)
 

src/weights.jl

@@ -464,9 +464,9 @@ function _wsum1!(R::AbstractArray, A::AbstractVector, w::AbstractVector, init::B
 end
 
 function _wsum2_blas!(R::StridedVector{T}, A::StridedMatrix{T}, w::StridedVector{T}, dim::Int, init::Bool) where T<:BlasReal
-    beta = ifelse(init, zero(T), one(T))
+    beta = ifelse(init, false, true)
     trans = dim == 1 ? 'T' : 'N'
-    BLAS.gemv!(trans, one(T), A, w, beta, R)
+    BLAS.gemv!(trans, true, A, w, beta, R)
     return R
 end
 

test/jet.jl

@@ -0,0 +1,10 @@
+using StatsBase, JET, Test
+
+@testset "JET" begin
+    JET.test_package(StatsBase; target_modules = [StatsBase])
+    # JET errors on interface definitions: https://github.com/aviatesk/JET.jl/issues/495
+    # The (deprecated) `model_response(::StatisticalModel)` calls the interface
+    # function `response(::StatisticalModel)` for which no method exists yet
+    res = JET.report_package(StatsBase; target_defined_modules = true, toplevel_logger = nothing)
+    @test length(JET.get_reports(res)) == 1
+end

test/runtests.jl

@@ -4,6 +4,7 @@ using LinearAlgebra
 using Random
 
 tests = ["ambiguous",
+         "jet",
          "weights",
          "moments",
          "scalarstats",