aboutsummaryrefslogtreecommitdiff
path: root/test/py_qr.py
diff options
context:
space:
mode:
Diffstat (limited to 'test/py_qr.py')
-rwxr-xr-xtest/py_qr.py32
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 )