jyapayne/glymur
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Using 3.x warning infrastructure exclusively.

Browse source
This commit is contained in:
John Evans 2014-08-17 19:03:29 -04:00
commit 61126d6a8b
1 changed files with 8 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

29
glymur/test/test_codestream_warnings.py
View file

@ -20,7 +20,8 @@ import glymur
from .fixtures import opj_data_file, OPJ_DATA_ROOT from .fixtures import opj_data_file, OPJ_DATA_ROOT
@unittest.skipIf(sys.platform.startswith('linux'), 'warnings failing on linux') @unittest.skipIf(sys.hexversion < 0x03030000,
"assertWarn methods introduced in 3.x")
@unittest.skipIf(OPJ_DATA_ROOT is None, @unittest.skipIf(OPJ_DATA_ROOT is None,
"OPJ_DATA_ROOT environment variable not set") "OPJ_DATA_ROOT environment variable not set")
class TestCodestreamOpjDataWarnings(unittest.TestCase): class TestCodestreamOpjDataWarnings(unittest.TestCase):
@ -29,39 +30,26 @@ class TestCodestreamOpjDataWarnings(unittest.TestCase):
def test_bad_rsiz(self): def test_bad_rsiz(self):
"""Should warn if RSIZ is bad. Issue196""" """Should warn if RSIZ is bad. Issue196"""
filename = opj_data_file('input/nonregression/edf_c2_1002767.jp2') filename = opj_data_file('input/nonregression/edf_c2_1002767.jp2')
with warnings.catch_warnings(record=True) as w: with self.assertWarnsRegex(UserWarning, 'Invalid profile'):
warnings.simplefilter('always')
j = Jp2k(filename) j = Jp2k(filename)
self.assertEqual(len(w), 3)
self.assertTrue(issubclass(w[0].category, UserWarning))
self.assertTrue('Invalid profile' in str(w[0].message))
def test_bad_wavelet_transform(self): def test_bad_wavelet_transform(self):
"""Should warn if wavelet transform is bad. Issue195""" """Should warn if wavelet transform is bad. Issue195"""
filename = opj_data_file('input/nonregression/edf_c2_10025.jp2') filename = opj_data_file('input/nonregression/edf_c2_10025.jp2')
with warnings.catch_warnings(record=True) as w: with self.assertWarnsRegex(UserWarning, 'Invalid wavelet transform'):
warnings.simplefilter('always')
j = Jp2k(filename) j = Jp2k(filename)
self.assertTrue(issubclass(w[0].category, UserWarning))
self.assertTrue('Invalid wavelet transform' in str(w[0].message))
def test_invalid_progression_order(self): def test_invalid_progression_order(self):
"""Should still be able to parse even if prog order is invalid.""" """Should still be able to parse even if prog order is invalid."""
jfile = opj_data_file('input/nonregression/2977.pdf.asan.67.2198.jp2') jfile = opj_data_file('input/nonregression/2977.pdf.asan.67.2198.jp2')
with warnings.catch_warnings(record=True) as w: with self.assertWarnsRegex(UserWarning, 'Invalid progression order'):
warnings.simplefilter('always')
Jp2k(jfile) Jp2k(jfile)
self.assertTrue(issubclass(w[0].category, UserWarning))
self.assertTrue('Invalid progression order' in str(w[0].message))
def test_tile_height_is_zero(self): def test_tile_height_is_zero(self):
"""Zero tile height should not cause an exception.""" """Zero tile height should not cause an exception."""
filename = opj_data_file('input/nonregression/2539.pdf.SIGFPE.706.1712.jp2') filename = opj_data_file('input/nonregression/2539.pdf.SIGFPE.706.1712.jp2')
with warnings.catch_warnings(record=True) as w: with self.assertWarnsRegex(UserWarning, 'Invalid tile dimensions'):
warnings.simplefilter('always')
Jp2k(filename) Jp2k(filename)
self.assertTrue(issubclass(w[0].category, UserWarning))
self.assertTrue('Invalid tile dimensions' in str(w[0].message))
@unittest.skipIf(os.name == "nt", "Temporary file issue on window.") @unittest.skipIf(os.name == "nt", "Temporary file issue on window.")
def test_unknown_marker_segment(self): def test_unknown_marker_segment(self):
@ -85,11 +73,8 @@ class TestCodestreamOpjDataWarnings(unittest.TestCase):
tfile.write(read_buffer) tfile.write(read_buffer)
tfile.flush() tfile.flush()
with warnings.catch_warnings(record=True) as w: with self.assertWarnsRegex(UserWarning, 'Unrecognized marker'):
warnings.simplefilter('always')
codestream = Jp2k(tfile.name).get_codestream() codestream = Jp2k(tfile.name).get_codestream()
self.assertTrue(issubclass(w[0].category, UserWarning))
self.assertTrue('Unrecognized marker' in str(w[0].message))
self.assertEqual(codestream.segment[2].marker_id, '0xff79') self.assertEqual(codestream.segment[2].marker_id, '0xff79')
self.assertEqual(codestream.segment[2].length, 3) self.assertEqual(codestream.segment[2].length, 3)