Useful brackets indexing for trajectories + test.

core/observations.jl

@@ -35,10 +35,9 @@ function getindex(σ::AbstractTrajectory, var::String)
     end
 end
 # Get i-th state [i]
-getindex(σ::AbstractTrajectory, idx::Int) = get_state(σ, i)
+getindex(σ::AbstractTrajectory, idx::Int) = get_state(σ, idx)
 # Get i-th value of var ["I", idx]
-function getindex(σ::AbstractTrajectory, var_idx::Tuple{String,Int})
-    var, idx = var_idx[1], var_idx[2]
+function getindex(σ::AbstractTrajectory, var::String, idx::Int)
     if var  == "times"
         return σ.times[idx]
     elseif var == "transitions"

tests/run_unit.jl

@@ -10,5 +10,6 @@ using Test
     @test include("unit/simulate_er.jl")
     @test include("unit/change_obs_var_sir.jl")
     @test include("unit/change_obs_var_sir_2.jl")
+    @test include("unit/getindex_access_trajectory.jl")
 end
 

tests/unit/getindex_access_trajectory.jl

+
+using MarkovProcesses
+
+load_model("SIR")
+σ = simulate(SIR)
+σ["I"]
+σ["I",2]
+σ[3]
+
+return true
+