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.18, 0.10"
 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,8 @@ 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))
     trans = dim == 1 ? 'T' : 'N'
-    BLAS.gemv!(trans, one(T), A, w, beta, R)
+    BLAS.gemv!(trans, true, A, w, !init, R)
     return R
 end
 

test/jet.jl

@@ -0,0 +1,22 @@
+using StatsBase, Test
+import JET
+
+# JET has only experimental support for Julia 1.12 currently
+# It throws an internal `AssertionError` in the tests below
+if VERSION < v"1.12-"
+    @testset "JET" begin
+        # Check that there are no undefined global references and undefined field accesses
+        JET.test_package("StatsBase"; target_defined_modules = true, mode = :typo)
+
+        # Default error analysis for common problem fails since JET errors on interface definitions
+        # The (deprecated) `model_response(::StatisticalModel)` calls the interface
+        # function `response(::StatisticalModel)` for which no method exists yet
+        # Note: This analysis is not enough strict to guarantee that there are no runtime errors!
+        # Ref https://github.com/aviatesk/JET.jl/issues/495
+        res = JET.report_package("StatsBase"; target_defined_modules = true, mode = :basic, toplevel_logger = nothing)
+        println(res)
+        reports = JET.get_reports(res)
+        @test_broken isempty(reports)
+        @test length(reports) <= 1
+    end
+end

test/runtests.jl

@@ -23,7 +23,9 @@ tests = ["ambiguous",
          "wsampling",
          "statmodels",
          "partialcor",
-         "transformations"]
+         "transformations",
+         # Test with JET after all other tests since it has side effects
+         "jet"] 
          #"statquiz"]
 
 println("Running tests:")