Fix psis test (#148)

* Fix test * Increment version number
arviz-devs · Jan 18, 2022 · 698fff9 · 698fff9 · sethaxen · Jan 18, 2022
1 parent ab1163b
commit 698fff9
Showing 1 changed file with 7 additions and 3 deletions.
diff --git 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