def test_make_color_gif(): # just making sure it runs im = imcmc.load_image(EXAMPLE) trace = imcmc.sample_color(im, samples=1000, nchains=2) with tempfile.TemporaryDirectory() as tmpdirname: filename = os.path.join(tmpdirname, "test_gif.gif") assert not os.path.exists(filename) imcmc.make_color_gif(trace, im, steps=2, filename=filename) assert os.path.exists(filename)
def test_sample_color(): im = imcmc.load_image(EXAMPLE) trace = imcmc.sample_color(im, samples=1000, nchains=2) for color in ("red", "blue", "green"): assert len(trace[color]) == 1000 * 2
def test_plot_multitrace_color(): # just making sure it runs im = imcmc.load_image(EXAMPLE) trace = imcmc.sample_color(im, samples=1000, nchains=2) imcmc.plot_multitrace_color(trace, im)
def test_sample_color(): im = imcmc.load_image(EXAMPLE) trace = imcmc.sample_color(im, samples=1000, nchains=2) for color in ('red', 'blue', 'green'): assert len(trace[color]) == 1000 * 2