How to test if all rows are equal in a numpy,(arr == arr[0]).all()