Add tests for deprecated interface

MurrellGroup · Nov 28, 2024 · 105b05e · 105b05e
1 parent 53da034
commit 105b05e
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -140,4 +140,17 @@ using Random
 
     end
 
+    @testset "deprecated.jl" begin
+        logits = log.([0.1, 0.2, 0.3, 0.4])
+        @test argmax_sampler(logits) isa Integer
+        @test_throws BoundsError top_pk_sampler(logits) # k defaults to 5
+        @test top_pk_sampler(logits; k = 3) isa Integer
+        @test min_p_sampler(logits) isa Integer
+        @test top_nσ_sampler(logits) isa Integer
+        @test argmax_sampler() isa Function
+        @test top_pk_sampler() isa Function
+        @test min_p_sampler() isa Function
+        @test top_nσ_sampler() isa Function
+    end
+
 end