Add tests for instantiate and general fallback for NodeType

ReactiveBayes · Sep 25, 2024 · 73d1304 · 73d1304
1 parent c8181d9
commit 73d1304
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/src/model/graphppl.jl b/src/model/graphppl.jl
@@ -169,7 +169,7 @@ function GraphPPL.NodeType(backend::ReactiveMPGraphPPLBackend{Static.True}, some
 end
 function GraphPPL.NodeType(backend::ReactiveMPGraphPPLBackend{Static.True}, ::ReactiveMP.UndefinedNodeFunctionalForm, something::F) where {F}
     # Fallback to the default behaviour if the node is not predefined
-    return GraphPPL.NodeType(ReactiveMPGraphPPLBackend(Static.False), something)
+    return GraphPPL.NodeType(ReactiveMPGraphPPLBackend(Static.False()), something)
 end
 function GraphPPL.NodeType(backend::ReactiveMPGraphPPLBackend{Static.True}, ::ReactiveMP.PredefinedNodeFunctionalForm, something::F) where {F}
     # Fallback to the default behaviour if the node is not predefined

diff --git a/test/model/graphppl_tests.jl b/test/model/graphppl_tests.jl
@@ -163,3 +163,15 @@ end
     @test_throws ErrorException GraphPPL.default_parametrization(backend, GraphPPL.Atomic(), f, (1,))
     @test_throws ErrorException GraphPPL.default_parametrization(backend, GraphPPL.Atomic(), f, (1, 2, 3))
 end
+
+@testitem "Backend Tests" begin
+    import GraphPPL
+    import Static
+    import RxInfer: ReactiveMPGraphPPLBackend
+
+    @test GraphPPL.instantiate(ReactiveMPGraphPPLBackend{Static.True}) == ReactiveMPGraphPPLBackend(Static.True())
+    @test GraphPPL.instantiate(ReactiveMPGraphPPLBackend{Static.False}) == ReactiveMPGraphPPLBackend(Static.False())
+    @test GraphPPL.instantiate(ReactiveMPGraphPPLBackend) == ReactiveMPGraphPPLBackend(Static.False())
+
+    @test GraphPPL.NodeType(ReactiveMPGraphPPLBackend(Static.True()), ReactiveMP.UndefinedNodeFunctionalForm(), sum) == GraphPPL.Atomic()
+end