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