Accelerate gradient check test by using smaller arrays

3 jobs for 0.3.2 in 5 minutes and 58 seconds (queued for 2 seconds)