numpy if array contains inf code example

Example: check if a numpy array contains only 1's python

>>> import numpy as np
>>> array=np.array([[1,1,1],[1,1,1],[1,1,1]])
>>> array([[1, 1, 1],
       	   [1, 1, 1],
           [1, 1, 1]])
>>> np.all(array == 1)
>>> True