def test_output_stroke(): """ Test : output stroke """ img = random.random((3,4,5)) assert img.shape == (3,4,5) out = stroke(img,1,True) assert out.shape == (5,6,7) assert (out[0,:,:] == zeros((6,7))).all() assert (out[4,:,:] == zeros((6,7))).all() assert (out[:,0,:] == zeros((5,7))).all() assert (out[:,5,:] == zeros((5,7))).all() assert (out[:,:,0] == zeros((5,6))).all() assert (out[:,:,6] == zeros((5,6))).all() assert (out[1:4,1:5,1:6] == img).all()
def test_input_stroke(): """ Test : output stroke """ img = random.random((3,4,5)) assert img.shape == (3,4,5) out = stroke(img,1) assert out.shape == (3,4,5) assert (out[0,:,:] == zeros((4,5))).all() assert (out[2,:,:] == zeros((4,5))).all() assert (out[:,0,:] == zeros((3,5))).all() assert (out[:,3,:] == zeros((3,5))).all() assert (out[:,:,0] == zeros((3,4))).all() assert (out[:,:,4] == zeros((3,4))).all() assert (out[1:2,1:3,1:4] == img[1:2,1:3,1:4] ).all()