Update test-multirate-output-export.jl

Add test with all defaults for models running at different rates (testing inference of multirate traits)

Author: VEZY

test/test-multirate-output-export.jl

@@ -15,6 +15,35 @@ function PlantSimEngine.run!(m::MRExportSourceModel, models, status, meteo, cons
     status.X = float(m.n[])
 end
 
+PlantSimEngine.@process "mrdefaultleafsource" verbose = false
+struct MRDefaultLeafSourceModel <: AbstractMrdefaultleafsourceModel
+    n::Base.RefValue{Int}
+end
+PlantSimEngine.inputs_(::MRDefaultLeafSourceModel) = NamedTuple()
+PlantSimEngine.outputs_(::MRDefaultLeafSourceModel) = (X=-Inf,)
+function PlantSimEngine.run!(m::MRDefaultLeafSourceModel, models, status, meteo, constants=nothing, extra=nothing)
+    m.n[] += 1
+    status.X = float(m.n[])
+end
+
+PlantSimEngine.@process "mrdefaultplantagg" verbose = false
+struct MRDefaultPlantAggModel <: AbstractMrdefaultplantaggModel end
+PlantSimEngine.inputs_(::MRDefaultPlantAggModel) = (X=-Inf,)
+PlantSimEngine.outputs_(::MRDefaultPlantAggModel) = (XP=-Inf,)
+function PlantSimEngine.run!(::MRDefaultPlantAggModel, models, status, meteo, constants=nothing, extra=nothing)
+    status.XP = sum(status.X) + 100.0
+end
+PlantSimEngine.timespec(::Type{<:MRDefaultPlantAggModel}) = ClockSpec(2.0, 1.0)
+
+PlantSimEngine.@process "mrdefaultsceneagg" verbose = false
+struct MRDefaultSceneAggModel <: AbstractMrdefaultsceneaggModel end
+PlantSimEngine.inputs_(::MRDefaultSceneAggModel) = (XP=-Inf,)
+PlantSimEngine.outputs_(::MRDefaultSceneAggModel) = (XS=-Inf,)
+function PlantSimEngine.run!(::MRDefaultSceneAggModel, models, status, meteo, constants=nothing, extra=nothing)
+    status.XS = sum(status.XP) + 1000.0
+end
+PlantSimEngine.timespec(::Type{<:MRDefaultSceneAggModel}) = ClockSpec(4.0, 1.0)
+
 @testset "Multi-rate output export API" begin
     mtg = Node(MultiScaleTreeGraph.NodeMTG("/", "Scene", 1, 0))
     plant = Node(mtg, MultiScaleTreeGraph.NodeMTG("+", "Plant", 1, 1))
@@ -120,3 +149,55 @@ end
     @test haskey(out_status_mtg, "Leaf")
     @test out_requested_mtg[:x_mtg][:, :value] == [1.0, 2.0, 3.0, 4.0]
 end
+
+@testset "Multi-rate output export defaults on multi-scale mapping with timespec traits" begin
+    mtg = Node(MultiScaleTreeGraph.NodeMTG("/", "Scene", 1, 0))
+    plant = Node(mtg, MultiScaleTreeGraph.NodeMTG("+", "Plant", 1, 1))
+    internode = Node(plant, MultiScaleTreeGraph.NodeMTG("/", "Internode", 1, 2))
+    Node(internode, MultiScaleTreeGraph.NodeMTG("+", "Leaf", 1, 2))
+
+    meteo8 = Weather(repeat([Atmosphere(T=20.0, Wind=1.0, Rh=0.65)], 8))
+
+    mapping_defaults = Dict(
+        "Leaf" => (
+            MultiScaleModel(
+                model=MRDefaultLeafSourceModel(Ref(0)),
+                mapped_variables=[:X => "Plant"],
+            ),
+        ),
+        "Plant" => (
+            MultiScaleModel(
+                model=MRDefaultPlantAggModel(),
+                mapped_variables=[:XP => "Scene"],
+            ),
+        ),
+        "Scene" => (
+            MRDefaultSceneAggModel(),
+        ),
+    )
+
+    sim_defaults = PlantSimEngine.GraphSimulation(
+        mtg,
+        mapping_defaults,
+        nsteps=8,
+        check=true,
+        outputs=Dict("Leaf" => (:X,), "Plant" => (:XP,), "Scene" => (:XS,)),
+    )
+
+    run!(
+        sim_defaults,
+        meteo8,
+        multirate=true,
+        executor=SequentialEx(),
+        tracked_outputs=[
+            OutputRequest("Plant", :XP),
+            OutputRequest("Scene", :XS),
+        ],
+    )
+
+    exported_defaults = collect_outputs(sim_defaults; sink=DataFrame)
+
+    @test sort(collect(keys(exported_defaults))) == [:XP, :XS]
+    @test exported_defaults[:XP][:, :timestep] == collect(1:8)
+    @test exported_defaults[:XS][:, :timestep] == collect(1:8)
+end