Merge pull request #2830 from ven-k/vkb/vector-defaults

fix: allow vector defaults in `@mtkmodel`

Author: ChrisRackauckas

src/systems/model_parsing.jl

@@ -340,6 +340,9 @@ function parse_default(mod, a)
             (expr, nothing)
         end
         Expr(:if, condition, x, y) => (a, nothing)
+        Expr(:vect, x...) => begin
+            (a, nothing)
+        end
         _ => error("Cannot parse default $a $(typeof(a))")
     end
 end

test/model_parsing.jl

@@ -817,3 +817,38 @@ end
     @test defs[ordermodel.c] == 2
     @test defs[ordermodel.d] == 1
 end
+
+@testset "Vector defaults" begin
+    @mtkmodel VectorDefaultWithMetadata begin
+        @parameters begin
+            n[1:3] = [1, 2, 3], [description = "Vector defaults"]
+        end
+    end
+
+    @named vec = VectorDefaultWithMetadata()
+    for i in 1:3
+        @test getdefault(vec.n[i]) == i
+    end
+
+    @mtkmodel VectorConditionalDefault begin
+        @structural_parameters begin
+            flag = true
+        end
+        @parameters begin
+            n[1:3] = if flag
+                [2, 2, 2]
+            else
+                1
+            end
+        end
+    end
+
+    @named vec_true = VectorConditionalDefault()
+    for i in 1:3
+        @test getdefault(vec_true.n[i]) == 2
+    end
+    @named vec_false = VectorConditionalDefault(flag = false)
+    for i in 1:3
+        @test getdefault(vec_false.n[i]) == 1
+    end
+end