Pillow/Tests/test_file_mcidas.py

35 lines
998 B
Python
Raw Normal View History

2015-07-03 08:03:25 +03:00
from helper import unittest, PillowTestCase
2017-05-28 16:11:16 +03:00
from PIL import Image, McIdasImagePlugin
2015-07-03 08:03:25 +03:00
class TestFileMcIdas(PillowTestCase):
def test_invalid_file(self):
2015-07-03 09:22:56 +03:00
invalid_file = "Tests/images/flower.jpg"
self.assertRaises(SyntaxError,
2017-09-01 14:05:40 +03:00
McIdasImagePlugin.McIdasImageFile, invalid_file)
2015-07-03 08:03:25 +03:00
2017-05-28 16:11:16 +03:00
def test_valid_file(self):
# Arrange
# https://ghrc.nsstc.nasa.gov/hydro/details/cmx3g8
# https://ghrc.nsstc.nasa.gov/pub/fieldCampaigns/camex3/cmx3g8/browse/
test_file = "Tests/images/cmx3g8_wv_1998.260_0745_mcidas.ara"
saved_file = "Tests/images/cmx3g8_wv_1998.260_0745_mcidas.png"
# Act
im = Image.open(test_file)
im.load()
# Assert
self.assertEqual(im.format, "MCIDAS")
self.assertEqual(im.mode, "I")
self.assertEqual(im.size, (1800, 400))
im2 = Image.open(saved_file)
self.assert_image_equal(im, im2)
2015-07-03 08:03:25 +03:00
if __name__ == '__main__':
unittest.main()