def testGetImageGrayscaleValuesInRange(self):
image_path = os.path.join(PREFIX_PATH, "jpeg", "testdata", "small.jpg")
image_data = convert_image_to_csv.get_image(27, 33, True, image_path)
self.assertLessEqual(0, np.min(image_data))
self.assertGreaterEqual(255, np.max(image_data))
def testGetImageConvertsGrayscaleToColor(self):
image_path = os.path.join(PREFIX_PATH, "png", "testdata",
"lena_gray.png")
image_data = convert_image_to_csv.get_image(23, 19, False, image_path)
self.assertEqual((19, 23, 3), image_data.shape)
def testGetImageCanLoadPng(self):
image_path = os.path.join(PREFIX_PATH, "png", "testdata",
"lena_rgba.png")
image_data = convert_image_to_csv.get_image(10, 10, False, image_path)
self.assertEqual((10, 10, 3), image_data.shape)
def testGetImageConvertsToGrayscale(self):
image_path = os.path.join(PREFIX_PATH, "jpeg", "testdata",
"medium.jpg")
image_data = convert_image_to_csv.get_image(40, 20, True, image_path)
self.assertEqual((20, 40, 1), image_data.shape)
def testGetImageSizeIsCorrect(self):
image_path = os.path.join(PREFIX_PATH, "jpeg", "testdata", "small.jpg")
image_data = convert_image_to_csv.get_image(64, 96, False, image_path)
self.assertEqual((96, 64, 3), image_data.shape)
def testGetImageRaisesMissingFile(self):
image_path = os.path.join(PREFIX_PATH, "jpeg", "testdata",
"no_such.jpg")
with self.assertRaises(NotFoundError):
_ = convert_image_to_csv.get_image(64, 96, False, image_path)