fix #2215

SciML · Aug 24, 2023 · bed9504 · bed9504
1 parent 12c83d4
commit bed9504
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/src/systems/abstractsystem.jl b/src/systems/abstractsystem.jl
@@ -1424,7 +1424,12 @@ function markio!(state, orig_inputs, inputs, outputs; check = true)
  outputset = Dict{Any, Bool}(o => false for o in outputs)
  for (i, v) in enumerate(fullvars)
  if v in keys(inputset)
- v = setio(v, true, false)
+ if v in keys(outputset)
+ v = setio(v, true, true)
+ outputset[v] = true
+ else
+ v = setio(v, true, false)
+ end
  inputset[v] = true
  fullvars[i] = v
  elseif v in keys(outputset)

diff --git a/test/linearize.jl b/test/linearize.jl
@@ -19,6 +19,13 @@ lsys, ssys = linearize(sys, [r], [y])
 @test lsys.C[] == 1
 @test lsys.D[] == 0
 
+lsys, ssys = linearize(sys, [r], [r])
+
+@test lsys.A[] == -2
+@test lsys.B[] == 1
+@test lsys.C[] == 0
+@test lsys.D[] == 1
+
 ##
 ```