def test_load_images(self):
images, image_files = utils.load_images(self.images_path,
extensions=('.jpg',),
img_shape=(256, 256))
self.assertIsInstance(images, np.ndarray)
self.assertEqual(images.shape, (64, 256, 256))
self.assertNotEqual(np.count_nonzero(images), 0)
self.assertEqual(len(image_files), 64)
def test_read_input(self):
images, _ = utils.load_images(self.images_path,
extensions=('.jpg',),
img_shape=(256, 256))
inst = utils.read_input(images)
self.assertIsInstance(inst, types.GeneratorType)
for i, image, image_index in inst:
self.assertIsInstance(i, types.IntType)
self.assertIn(i, range(len(images)))
self.assertIsInstance(image_index, types.IntType)
self.assertIn(image_index, range(len(images)))
self.assertIsInstance(image, np.ndarray)
self.assertEqual(image.shape, (256, 256))
self.assertListEqual(image.tolist(), images[image_index].tolist())
