Merge pull request #267 from ReactiveBayes/source-code-for-mf

Add source code for MeanField and NoConstraints structures

Author: wouterwln

Project.toml

@@ -1,7 +1,7 @@
 name = "GraphPPL"
 uuid = "b3f8163a-e979-4e85-b43e-1f63d8c8b42c"
 authors = ["Wouter Nuijten <[email protected]>", "Dmitry Bagaev <[email protected]>"]
-version = "4.6.0"
+version = "4.6.1"
 
 [deps]
 BitSetTuples = "0f2f92aa-23a3-4d05-b791-88071d064721"

src/plugins/variational_constraints/variational_constraints.jl

@@ -17,8 +17,12 @@ See also: [`BetheFactorization`](@ref)
 """
 struct MeanField end
 
+source_code(::MeanField) = "MeanField"
+
 struct NoConstraints end
 
+source_code(::NoConstraints) = "NoConstraints"
+
 include("variational_constraints_macro.jl")
 include("variational_constraints_engine.jl")
 

test/plugins/variational_constraints/variational_constraints_tests.jl

@@ -1160,4 +1160,12 @@ end
     @test occursin("q(x, y)", source)
     @test occursin("q(x)", source)
     @test occursin("q(y)", source)
+
+    mf_constraints = GraphPPL.MeanField()
+    source = GraphPPL.source_code(mf_constraints)
+    @test occursin("MeanField", source)
+
+    nc_constraints = GraphPPL.NoConstraints()
+    source = GraphPPL.source_code(nc_constraints)
+    @test occursin("NoConstraints", source)
 end