GradientDescent tests updated