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')