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