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