Skip to content

Commit 759e61e

Browse files
committed
Add test merging VarInfos with different dimensions for a variable
1 parent 938a69d commit 759e61e

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

test/varinfo.jl

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -869,6 +869,17 @@ end
869869
@test DynamicPPL.getgid(varinfo_merged, @varname(x)) == gidset_left
870870
@test DynamicPPL.getgid(varinfo_merged, @varname(y)) == gidset_right
871871
end
872+
873+
# The below used to error, testing to avoid regression.
874+
@testset "merge different dimensions" begin
875+
vn = @varname(x)
876+
vi_single = VarInfo()
877+
vi_single = push!!(vi_single, vn, 1.0, Normal())
878+
vi_double = VarInfo()
879+
vi_double = push!!(vi_double, vn, [2.0, 3.0], Normal())
880+
@test merge(vi_single, vi_double)[vn] == 1.0
881+
@test merge(vi_double, vi_single)[vn] == [2.0, 3.0]
882+
end
872883
end
873884

874885
@testset "VarInfo with selectors" begin

0 commit comments

Comments
 (0)