def test_make_gif():
im = imcmc.load_image(EXAMPLE, "L")
trace = imcmc.sample_grayscale(im, samples=1000, nchains=4)
with tempfile.TemporaryDirectory() as tmpdirname:
filename = os.path.join(tmpdirname, "test_gif.gif")
assert not os.path.exists(filename)
imcmc.make_gif(trace, im, steps=2, filename=filename)
assert os.path.exists(filename)
def test_plot_multitrace():
im = imcmc.load_image(EXAMPLE, "L")
trace = imcmc.sample_grayscale(im, samples=1000, nchains=4)
imcmc.plot_multitrace(trace, im)
def test_sample_grayscale():
im = imcmc.load_image(EXAMPLE, "L")
trace = imcmc.sample_grayscale(im, samples=1000, nchains=4)
assert len(trace["image"]) == 1000 * 4