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

add negative tests for bad precinct sizes, closes #280

Browse source
This commit is contained in:
John Evans 2014-10-22 15:34:13 -04:00
commit fc4f5cec80
1 changed files with 16 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

16
glymur/test/test_jp2k.py
View file

@ -843,6 +843,22 @@ class TestJp2k_write(unittest.TestCase):
def tearDown(self):
pass
def test_precinct_size_too_small(self):
"""first precinct size must be >= 2x that of the code block size"""
data = np.zeros((640, 480), dtype=np.uint8)
with tempfile.NamedTemporaryFile(suffix='.j2k') as tfile:
with self.assertRaises(IOError):
j = Jp2k(tfile.name, 'wb')
j.write(data, cbsize=(16, 16), psizes=[(16, 16)])
def test_precinct_size_not_power_of_two(self):
"""must be power of two"""
data = np.zeros((640, 480), dtype=np.uint8)
with tempfile.NamedTemporaryFile(suffix='.j2k') as tfile:
with self.assertRaises(IOError):
j = Jp2k(tfile.name, 'wb')
j.write(data, cbsize=(16, 16), psizes=[(48, 48)])
def test_unsupported_datatype(self):
"""Should raise a runtime error if trying to write uint32"""
data = np.zeros((128, 128), dtype=np.uint32)