def test_computeFactor():
im2=io.imread('data/design-2.png')
im3=io.imread('data/design-3.png')
w2=a5.computeWeight(im2)
w3=a5.computeWeight(im3)
out=a5.computeFactor(im2, w2, im3, w3)
if abs(out-50.8426955376)<1 :
print 'Correct'
def test_computeWeight():
im=io.imread('data/design-2.png')
out=a5.computeWeight(im)
io.imwrite(out, 'design-2_mask.png')