Source code for snowdrop.src.numeric.solver.AIM.checkA

"""
Functions to check if the matrix a contains any NaN or Inf values

Created by: Jason Sockin
Date: September 11, 2013
"""

# import numpy package
import numpy as np

# Checks for any NaN in a
[docs] def existsNaN(a): aTemp = np.isnan(a) aRows, aColumns = aTemp.shape result = False for i in range(0,aRows): for j in range(0,aColumns): if aTemp[i,j] == True: result = True break if result == True: break return result
# Checks for any Inf in a
[docs] def existsInf(a): aTemp = np.isinf(a) aRows, aColumns = aTemp.shape result = False for i in range(0,aRows): for j in range(0,aColumns): if aTemp[i,j] == True: result = True break if result == True: break return result