Fix tests

Author: utkarsh530

test/gpu.jl

@@ -21,17 +21,7 @@ sol = solve(prob, ParallelPSOKernel(n_particles; backend), maxiters = 500)
 @test sol.objective < 1e-4
 
 sol = solve(prob,
-    ParallelSyncPSO(n_particles, backend),
-    maxiters = 500)
-
-@test sol.objective < 1e-4
-
-prob = OptimizationProblem(rosenbrock, x0, p)
-
-n_particles = 2000
-
-sol = solve(prob,
-    ParallelPSOKernel(n_particles; threaded = true),
+    ParallelSyncPSOKernel(n_particles, backend),
     maxiters = 500)
 
 @test sol.objective < 1e-4

test/regression.jl

@@ -12,52 +12,61 @@ end
 x0 = @SArray zeros(Float32, N)
 p = @SArray Float32[1.0, 100.0]
 
+array_prob = OptimizationProblem(rosenbrock,
+    zeros(Float32, N),
+    Float32[1.0, 100.0];
+    lb = lb,
+    ub = ub)
+
 prob = OptimizationProblem(rosenbrock, x0, p; lb = lb, ub = ub)
 
 n_particles = 1000
 
-sol = solve(prob,
-    ParallelPSOKernel(n_particles; threaded = true),
+sol = solve(array_prob,
+    ParallelPSOArray(n_particles),
     maxiters = 500)
 
 @test sol.objective < 1e-4
 
 sol = solve(prob,
-    ParallelPSOKernel(n_particles; threaded = false),
+    SerialPSO(n_particles),
     maxiters = 500)
 
 @test sol.objective < 1e-4
 
 lb = @SVector fill(Float32(-Inf), N)
 ub = @SVector fill(Float32(Inf), N)
-prob = OptimizationProblem(rosenbrock, x0, p; lb = lb, ub = ub, N)
+
+array_prob = remake(array_prob; lb = lb, ub = ub)
+prob = remake(prob; lb = lb, ub = ub)
 
 n_particles = 2000
 
-sol = solve(prob,
-    ParallelPSOKernel(n_particles; threaded = true),
+sol = solve(array_prob,
+    ParallelPSOArray(n_particles),
     maxiters = 500)
 
 @test sol.objective < 1e-4
 
 sol = solve(prob,
-    ParallelPSOKernel(n_particles; threaded = false),
+    SerialPSO(n_particles),
     maxiters = 500)
 
 @test sol.objective < 1e-4
 
-prob = OptimizationProblem(rosenbrock, x0, p)
+array_prob = remake(array_prob; lb = nothing, ub = nothing)
+prob = remake(prob; lb = nothing, ub = nothing)
 
 n_particles = 2000
 
-sol = solve(prob,
-    ParallelPSOKernel(n_particles; threaded = true),
+sol = solve(array_prob,
+    ParallelPSOArray(n_particles),
     maxiters = 500)
 
 @test sol.objective < 1e-4
 
 sol = solve(prob,
-    ParallelPSOKernel(n_particles; threaded = false),
+    SerialPSO(n_particles),
     maxiters = 500)
 
 @test sol.objective < 1e-4