test dgp reducing over O values

salbalkus · Jan 27, 2025 · 215c902 · 215c902
1 parent 2febd5a
commit 215c902
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/src/data_generating_process.jl b/src/data_generating_process.jl
@@ -37,7 +37,6 @@ nothing # hide
 
 """
 
-
 macro dgp(args...)
     names = [_parse_name(arg) for arg in args]
     # parse each line of the input into a vector of vectors

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -147,7 +147,16 @@ end
 
     @test typeof(ct) == CausalTables.CausalTable
     @test Tables.columnnames(ct) == Tuple(Symbol("X$(i)") for i in 1:10)
-    @test_throws ArgumentError Base.merge(DataGeneratingProcess([O -> Normal(0,1)]; varsymb = "Y"), DataGeneratingProcess([O -> Normal(0,1)]; varsymb = "Y"))
+    @test_throws ArgumentError merge(DataGeneratingProcess([O -> Normal(0,1)]; varsymb = "Y"), 
+                                          DataGeneratingProcess([O -> Normal(0,1)]; varsymb = "Y"))
+
+    dgp3 = @dgp(
+        Y ~ Normal.(reduce(+, values(O)), 1)
+    )        
+    scm = CausalTables.StructuralCausalModel(merge(dgp, dgp3), :X1, :Y)
+    rand(scm, 10)
+    @test typeof(ct) == CausalTables.CausalTable
+    @test Tables.columnnames(ct) == Tuple(Symbol("X$(i)") for i in 1:10)
 end
 
 @testset "DataGeneratingProcess using dgp macro, no graphs" begin
-Original file line number
+Diff line change
@@ Expand Up / @@ -37,7 +37,6 @@ nothing # hide @@
     """
     macro dgp(args...)
         names = [_parse_name(arg) for arg in args]
         # parse each line of the input into a vector of vectors
@@ Expand Down @@