def test_iter_planes_grayscale(self):
img = imgs.create_gray()
planes = list(analyse.iter_planes(img))
self.assertEqual(8, len(planes)) # one band with 8 planes
band, bit, plane = planes[0]
self.assertEqual(0, bit)
self.assertEqual(np.prod(img.size), plane.size)
self.assertEqual(0, plane.sum())
def test_iter_planes_grayscale(self):
img = imgs.create_gray()
planes = list(analyse.iter_planes(img))
self.assertEqual(8, len(planes)) # one band with 8 planes
band, bit, plane = planes[0]
self.assertEqual(0, bit)
self.assertEqual(np.prod(img.size), plane.size)
self.assertEqual(0, plane.sum())
def test_iter_planes(self):
img = imgs.create_rgb()
planes = list(analyse.iter_planes(img))
# 3 bands (R, G, B) w/ 8 planes + 1 special band (RGB) with 8 planes
self.assertEqual(3 * 8 + 8, len(planes))
band, bit, plane = planes[0]
self.assertEqual('R', band)
self.assertEqual(0, bit)
self.assertEqual(np.prod(img.size), plane.size)
self.assertEqual(0, plane.sum())
def test_iter_planes(self):
img = imgs.create_rgb()
planes = list(analyse.iter_planes(img))
# 3 bands (R, G, B) w/ 8 planes + 1 special band (RGB) with 8 planes
self.assertEqual(3 * 8 + 8, len(planes))
band, bit, plane = planes[0]
self.assertEqual('R', band)
self.assertEqual(0, bit)
self.assertEqual(np.prod(img.size), plane.size)
self.assertEqual(0, plane.sum())
