Fix to the test_cross_validation.py script

I modified the lstsq Op:
  -The 3rd output is now an numpy array instead of an int.
  -We now check if the 1st output contains elements of type float instead of type int.