# test_solve.py     simeq is solve in Python, needs numpy
from numpy import array
from numpy.linalg import solve

print "test_solve.py  a x = y "

a=array([[1.0/2,1.0/3,1.0/4,1.0/5],[1.0/3,1.0/4,1.0/5,1.0/6],[1.0/4,1.0/5,1.0/6,1.0/7],[1.0/5,1.0/6,1.0/7,1.0/8]])
print "a="
print  a
y=array([1,2,3,4])
print "y="
print y

x=solve(a,y)
print "solve x="
print x
print " "

# check y1 = a * x
y1=array([0,0,0,0])
for i in range(4):
  for j in range(4):
    y1[i] = y1[i] + a[i][j]*x[j]

print "check y = y1 = a*x ="
print y1
print " "

y2=a*x
print "check y2 = a*x ="
print y2
print "check for i in range(4): sum(y2[i])"
for i in range(4):
  print sum(y2[i])