def test_lapack_dgtsvx():
BB2 = np.array((B,))
BB2 = matrix_solvers.dgtsvx(dl,d,du,BB2)
#print "dgesv :", BB[0]
def test_lapack_tridiagonal():
BB2 = np.array((B,))
BB2 = matrix_solvers.tridiagonal_solve(dl,d,du,BB2)
#print "dgtsvx :", BB[0]
def test_lapack_dgtsvx():
BB2 = np.array((B,))
BB2 = matrix_solvers.dgtsvx(dl,d,du,BB2)
#print "dgtsvx :", BB[0]
if __name__ == '__main__':
n = 8
dl = np.ones(n-1, dtype=np.float64)-2.0
d = np.ones(n, dtype=np.float64)+2.0
du = np.ones(n-1, dtype=np.float64)-2.0
A = np.diag(d) + \
np.diag(du, k=1) + \
np.diag(dl, k=-1)
B = np.arange(n, dtype=np.float64)
BB = np.array((B,B)).transpose()
X = solve(A,BB)
print 'scipy :', X
L = matrix_solvers.dgtsvx(dl,d,du,BB.transpose())
print 'dgtsvx:', L
