def test_num_digits(self): from jicbioimage.core.image import Image3D self.assertEqual(Image3D._num_digits(1), 1) self.assertEqual(Image3D._num_digits(9), 1) self.assertEqual(Image3D._num_digits(10), 2) self.assertEqual(Image3D._num_digits(99), 2) self.assertEqual(Image3D._num_digits(100), 3)
def to_directory(self, directory): if not os.path.isdir(directory): os.mkdir(directory) xdim, ydim, zdim = self.shape num_digits = Image3D._num_digits(zdim-1) for z in range(zdim): num = str(z).zfill(num_digits) fname = "z{}.png".format(num) fpath = os.path.join(directory, fname) with open(fpath, "wb") as fh: im = Image.from_array(unique_color_array(self[:, :, z])) fh.write(im.png())