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)