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