diff --git a/test/test_stats.jl b/test/test_stats.jl index 8122f2b2..51aa2427 100644 --- a/test/test_stats.jl +++ b/test/test_stats.jl @@ -41,11 +41,15 @@ using DataFrames: DataFrames @test k ≈ result.pareto_shape # check against Python ArviZ - # NOTE: currently these implementations disagree - # see https://github.com/arviz-devs/arviz/issues/1941 logw_smoothed2, k2 = ArviZ.arviz.psislw(copy(logw), 0.9) + # NOTE: currently the smoothed weights disagree, while the shapes agree + # see https://github.com/arviz-devs/arviz/issues/1941 @test_broken logw_smoothed ≈ logw_smoothed2 - @test_broken k ≈ k2 + if length(sz) == 1 + @test k ≈ k2[] # k2 is a 0-dimensional array + else + @test k ≈ k2 + end end end