def test_is_image(self): image_files = [ 'image.gif', 'image.jpg', 'image.jpeg', 'image.png', 'image.GIF', 'image.JPG', 'image.JPEG', 'image.PNG', 'image.GiF', 'image.JpG', 'image.JpEg', 'image.PnG' ] for image_file in image_files: self.assertTrue(utils.is_image_file(image_file))
def test_is_not_image(self): image_files = [ 'image.docx', 'image.pdf', 'image.md', 'image.html', 'image.css', 'image.odt', 'image.sh', 'image.xlsx', 'image.txt', 'image.c', 'image.py', 'image' ] for image_file in image_files: self.assertFalse(utils.is_image_file(image_file))