Instantaneous test API

Author: BenChung

src/ModelTesting.jl

@@ -8,5 +8,6 @@ include("test/continuous/delta_sol.jl")
 include("test/discrete/single_shooting.jl")
 include("test/discrete/merge.jl")
 include("test/metric/metric.jl")
+include("test/instantaneous/instant.jl")
 include("problem_config/problem.jl")
 end

src/test/instantaneous/instant.jl

@@ -0,0 +1,45 @@
+function test_instantaneous(
+    sys::ModelingToolkit.AbstractSystem,
+    ic,
+    checks::Num;
+    t = nothing) # todo: after porting to v9 use InitializationSystem to solve the checks system
+    if !SymbolicIndexingInterface.is_observed(checks)
+        @assert false "The given check values are not observed values of the given system"
+    end
+    sv = ModelingToolkit.varmap_to_vars(ic, states(sys); defaults=ModelingToolkit.defaults(sys))
+    pv = ModelingToolkit.varmap_to_vars(ic, parameters(sys); defaults=ModelingToolkit.defaults(sys))
+
+    obsfun = SymbolicIndexingInterface.observed(sys, checks)
+    if SymbolicIndexingInterface.istimedependent(sys)
+        @assert !isnothing(t) "The kwarg t must be given (and be a non-nothing value) if the system is time dependent"
+        return obsfun(sv, pv, t)
+    elseif !SymbolicIndexingInterface.constant_structure(sys)
+        @assert false "The system's structure must be constant to use test_instantaneous; to be fixed" # TODO
+    else
+        return obsfun(sv, pv)
+    end
+end
+
+function test_instantaneous(
+    sys::ModelingToolkit.AbstractSystem,
+    ic,
+    checks::Array;
+    t = nothing) # todo: after porting to v9 use InitializationSystem to solve the checks system
+    if !all(SymbolicIndexingInterface.is_observed.((sys, ), checks))
+        @assert false "The given check values are not observed values of the given system"
+    end
+    sv = ModelingToolkit.varmap_to_vars(ic, states(sys); defaults=ModelingToolkit.defaults(sys))
+    pv = ModelingToolkit.varmap_to_vars(ic, parameters(sys); defaults=ModelingToolkit.defaults(sys))
+
+    obsfuns = SymbolicIndexingInterface.observed.((ODEFunction(sys),), checks)
+    if SymbolicIndexingInterface.is_time_dependent(sys)
+        @assert !isnothing(t) "The kwarg t must be given (and be a non-nothing value) if the system is time dependent"
+        return map(fun -> fun(sv, pv, t), obsfuns)
+    elseif !SymbolicIndexingInterface.constant_structure(sys)
+        @assert false "The system's structure must be constant to use test_instantaneous; to be fixed" # TODO
+    else
+        return map(fun -> fun(sv, pv), obsfuns)
+    end
+end
+
+export test_instantaneous

test/block_modeling.jl

@@ -64,6 +64,12 @@ using ModelingToolkitStandardLibrary.Blocks: Constant
         sys = structural_simplify(rc_model)
         prob1 = ODEProblem(sys, Pair[], (0, 10.0))
         sol1 = solve(prob1, Tsit5())
+
+        ref = DataFrame(:timestamp => sol1.t, Symbol(resistor.v) => sol1[resistor.v])
+        measure, measured_sys = compare_data(resistor.v, ref)(sys)
+        prob_measured = ODEProblem(measured_sys, Pair[], (0, 10.0))
+        sol_measured = solve(prob_measured, Tsit5())
+
         prob2 = ODEProblem(sys, Pair[capacitor.C => 0.9], (0, 10.0))
         sol2 = solve(prob2, Tsit5())
         prob3 = ODEProblem(sys, Pair[capacitor.C => 5.0], (0, 10.0))

test/instantaneous.jl

@@ -0,0 +1,30 @@
+@testset "Instantaneous" begin 
+    @testset "RC Functional" begin 
+        R = 1.0
+        C = 1.0
+        V = 1.0
+        @variables t
+        @named resistor = Resistor(R = R)
+        @named capacitor = Capacitor(C = C)
+        @named source = Voltage()
+        @named constant = Constant(k = V)
+        @named ground = Ground()
+
+        @named sensor = PotentialSensor()
+        @named key_parameter = Measurement(sensor)
+
+        rc_eqs = [connect(constant.output, source.V)
+                connect(source.p, resistor.p)
+                connect(resistor.n, capacitor.p)
+                connect(capacitor.n, source.n, ground.g)
+                connect(sensor.p, capacitor.p)]
+
+        @named rc_model = ODESystem(rc_eqs, t,
+            systems = [resistor, capacitor, constant, source, ground, sensor, key_parameter])
+        sys = structural_simplify(rc_model)
+        @test test_instantaneous(sys, [], [resistor.v]; t = 0.0)[1] > 0
+        @test test_instantaneous(sys, [], [resistor.i]; t = 0.0)[1] > 0
+        @test test_instantaneous(sys, [constant.k => 0.0], [resistor.v]; t = 0.0)[1] == 0
+        @test test_instantaneous(sys, [constant.k => 5.0], [resistor.v]; t = 0.0)[1] == 5
+    end
+end

test/runtests.jl

@@ -7,4 +7,5 @@ using Test
 @testset "ModelTesting.jl" begin
     include("timeseries.jl")
     include("block_modeling.jl")
+    include("instantaneous.jl")
 end