Python Test If Numpy Array Size