diff options
Diffstat (limited to 'test/py_qr.py')
-rwxr-xr-x | test/py_qr.py | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/test/py_qr.py b/test/py_qr.py new file mode 100755 index 0000000..ef757a3 --- /dev/null +++ b/test/py_qr.py @@ -0,0 +1,32 @@ +#!/usr/bin/env python3 + +import sys +import numpy as np +import scipy.linalg as la + +n = 3 + +if sys.argv.__len__()!=n*n+1: + print( "bad # args" ) + sys.exit() + + +A = np.empty([n,n], dtype='f8') + +for j in range(n): + for i in range(n): + A[j][i] = sys.argv[1+j*n+i] + +print( "A:" ) +print( A ) + +Q,R,P = la.qr(A, pivoting=True) + +print( "Q:" ) +print( Q ) + +print( "R:" ) +print( R ) + +print( "P:" ) +print( P ) |