def test_read_write_read():
"""Tests if a scalar and a vector image are written and read back correctly.
Can't test if they are compatible with other Inrimage tools here, use ZViewer!"""
# load a file, write it and reread it:
f = "SAM.inr.gz"
f_p = "SAM_prime.inr.gz"
read_im = read_inrimage(f)
write_inrimage(f_p, read_im)
reread_im = read_inrimage(f_p)
numpy.testing.assert_array_equal(read_im, reread_im)
def test_generate_write_read():
"""Tests if a scalar and a vector image are written and read back correctly.
Can't test if they are compatible with other Inrimage tools here, use ZViewer!"""
# create a checkerboard that fades to black in depth
cb = checkerboard(vs=(0.5,0.5,0.2))
coeffs = attenuation(numpy.linspace(0,1, cb.shape[2])).reshape(1,1,cb.shape[2])
cb = SpatialImage(cb*coeffs, voxelsize=cb.voxelsize, dtype=cb.dtype)
f = "test_inri_0.inr.gz"
write_inrimage(f, cb)
read_cb = read_inrimage(f)
numpy.testing.assert_array_equal(cb, read_cb)
os.remove(f)
# create a random vector field
ff = "test_inri_0_tr.inr.gz"
field = random_vector_field_like(cb, 4.0,20)
write_inrimage(ff, field)
read_field = read_inrimage(ff)
numpy.testing.assert_array_equal(field, read_field)
os.remove(ff)
