Add tests for get/set/acc/resetlogp

TuringLang · Dec 1, 2024 · e791193 · e791193
1 parent 408a6c6
commit e791193
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/test/varinfo.jl b/test/varinfo.jl
@@ -131,6 +131,25 @@ DynamicPPL.getspace(::DynamicPPL.Sampler{MySAlg}) = (:s,)
         test_base!!(SimpleVarInfo(DynamicPPL.VarNamedVector()))
     end
 
+    @testset "get/set/acc/resetlogp" begin
+        function test_varinfo_logp!(vi)
+            @test DynamicPPL.getlogp(vi) === 0.0
+            vi = DynamicPPL.setlogp!!(vi, 1)
+            @test DynamicPPL.getlogp(vi) === 1.0
+            vi = DynamicPPL.acclogp!!(vi, 1)
+            @test DynamicPPL.getlogp(vi) === 2.0
+            vi = DynamicPPL.resetlogp!!(vi)
+            @test DynamicPPL.getlogp(vi) === 0.0
+        end
+
+        vi = VarInfo()
+        test_varinfo_logp!!(vi)
+        test_varinfo_logp!!(TypedVarInfo(vi))
+        test_varinfo_logp!!(SimpleVarInfo())
+        test_varinfo_logp!!(SimpleVarInfo(Dict()))
+        test_varinfo_logp!!(SimpleVarInfo(DynamicPPL.VarNamedVector()))
+    end
+
     @testset "flags" begin
         # Test flag setting:
         #    is_flagged, set_flag!, unset_flag!