def test_get_extension_several_dots(self): test_img_path = SCRIPT_FOLDER + str(os.sep) + "test.hey.jpeg" self.assertEqual(images2grey.get_extension(test_img_path), "jpeg")
def test_get_extension(self): self.assertEqual(images2grey.get_extension(TEST_IMG_PATH_1), "png") self.assertEqual(images2grey.get_extension(TEST_IMG_PATH_2), "bmp") self.assertEqual(images2grey.get_extension(TEST_GREY_IMG_PATH), "jpg")