you can use np._assert here