Fix the tests

Author: ChrisRackauckas

test/downstream/ensemble_multi_prob.jl

@@ -17,10 +17,9 @@ prob3 = ODEProblem(sys3, [3.0, 3.0], (0.0, 1.0))
 ensemble_prob = EnsembleProblem([prob1, prob2, prob3])
 sol = solve(ensemble_prob, Tsit5(), EnsembleThreads())
 for i in 1:3
-    @test sol[x, :][i] == sol.u[i][x]
-    @test sol[y, :][i] == sol.u[i][y]
+    @test sol[1,:,i] == sol.u[i][x]
+    @test sol[2,:,i] == sol.u[i][y]
 end
 # Ensemble is a recursive array
-@test only.(sol(0.0, idxs = [x])) == sol[1, 1, :] == first.(sol[x, :])
-# TODO: fix the interpolation
-@test only.(sol(1.0, idxs = [x])) ≈ last.(sol[x, :])
+@test only.(sol(0.0, idxs = [x])) == sol[1, 1, :]
+@test only.(sol(1.0, idxs = [x])) ≈ [sol[i][1, end] for i in 1:3]