Numpy check that all the element of each row of a 2D numpy array is the same