Add PSO tests