refactor: format

Author: AayushSabharwal

test/downstream/ensemble_diffeq.jl

@@ -2,18 +2,18 @@ using OrdinaryDiffEq, Test
 
 A = [1 2
      3 4]
-prob = ODEProblem((u, p, t) -> A*u, ones(2,2), (0.0, 1.0))
+prob = ODEProblem((u, p, t) -> A * u, ones(2, 2), (0.0, 1.0))
 function prob_func(prob, i, repeat)
     remake(prob, u0 = i * prob.u0)
 end
 ensemble_prob = EnsembleProblem(prob, prob_func = prob_func)
-sim = solve(ensemble_prob, Tsit5(), EnsembleThreads(), trajectories = 10, saveat=0.01)
+sim = solve(ensemble_prob, Tsit5(), EnsembleThreads(), trajectories = 10, saveat = 0.01)
 @test sim isa EnsembleSolution
-@test size(sim[1,:,:,:])  == (2,101,10)
-@test size(sim[:,1,:,:]) == (2,101,10)
-@test size(sim[:,:,1,:]) == (2,2,10)
-@test size(sim[:,:,:,1]) == (2,2,101)
-@test Array(sim)[1,:,:,:]  == sim[1,:,:,:]
-@test Array(sim)[:,1,:,:]  == sim[:,1,:,:]
-@test Array(sim)[:,:,1,:]  == sim[:,:,1,:]
-@test Array(sim)[:,:,:,1]  == sim[:,:,:,1]
+@test size(sim[1, :, :, :]) == (2, 101, 10)
+@test size(sim[:, 1, :, :]) == (2, 101, 10)
+@test size(sim[:, :, 1, :]) == (2, 2, 10)
+@test size(sim[:, :, :, 1]) == (2, 2, 101)
+@test Array(sim)[1, :, :, :] == sim[1, :, :, :]
+@test Array(sim)[:, 1, :, :] == sim[:, 1, :, :]
+@test Array(sim)[:, :, 1, :] == sim[:, :, 1, :]
+@test Array(sim)[:, :, :, 1] == sim[:, :, :, 1]

test/downstream/ensemble_multi_prob.jl

@@ -17,9 +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[1,:,i] == sol.u[i][x]
-    @test sol[2,:,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, :]
-@test only.(sol(1.0, idxs = [x])) ≈ [sol[i][1, end] for i in 1:3]
+@test only.(sol(1.0, idxs = [x])) ≈ [sol[i][1, end] for i in 1:3]