def testReadCalConstants(self):
# test read empty input
with self.assertRaisesRegex(ValueError, 'Please specify'):
read_image('')
# test wrong dimension
with patch('numpy.load', return_value=np.ones((2, 2, 2, 2))):
with self.assertRaisesRegex(ValueError, '2D or 3D array'):
read_cal_constants('abc')
with patch('numpy.load', return_value=np.ones(2)):
with self.assertRaisesRegex(ValueError, '2D or 3D array'):
read_cal_constants('abc')
# test valid data
for const_gt in [
np.ones([2, 2]),
np.ones([4, 2, 2], dtype=np.float32)
]:
fp = tempfile.NamedTemporaryFile(suffix='.npy')
np.save(fp.name, const_gt)
ret = read_cal_constants(fp)
np.testing.assert_array_equal(const_gt, ret)
# file does not have suffix '.npy'
with self.assertRaises(ValueError):
for const_gt in [
np.ones([2, 2]),
np.ones([4, 2, 2], dtype=np.float32)
]:
fp = tempfile.NamedTemporaryFile()
np.save(fp.name, const_gt)
read_cal_constants(fp)
def testReadCalConstants(self):
# test read empty input
with self.assertRaisesRegex(ValueError, 'Please specify'):
read_image('')
# test wrong dimension
with patch('numpy.load', return_value=np.ones((2, 2, 2, 2))):
with self.assertRaisesRegex(ValueError, 'dimensions'):
read_cal_constants('abc')
with patch('numpy.load', return_value=np.ones(2)):
with self.assertRaisesRegex(ValueError, 'dimensions'):
read_cal_constants('abc')
# test dtype
with patch('numpy.load', return_value=np.ones([3, 2], dtype=bool)):
img = read_cal_constants('abc')
self.assertEqual(img.dtype, config['SOURCE_PROC_IMAGE_DTYPE'])
self.assertEqual((3, 2), img.shape)
for const_gt in [np.ones([2, 2]), np.ones([4, 2, 2], dtype=np.float32)]:
fp = tempfile.NamedTemporaryFile(suffix='.npy')
np.save(fp.name, const_gt)
ret = read_cal_constants(fp)
np.testing.assert_array_equal(const_gt, ret)
# file does not have suffix '.npy'
with self.assertRaises(ValueError):
for const_gt in [np.ones([2, 2]), np.ones([4, 2, 2], dtype=np.float32)]:
fp = tempfile.NamedTemporaryFile()
np.save(fp.name, const_gt)
read_cal_constants(fp)
def testReadImage(self):
# test read empty input
with self.assertRaisesRegex(ValueError, 'Please specify'):
read_image('')
# test wrong shape
with patch('imageio.imread', return_value=np.ones((2, 2))):
with self.assertRaisesRegex(ValueError, 'Shape of'):
read_image('abc', expected_shape=(3, 2))
# test wrong dimension
with patch('imageio.imread', return_value=np.ones((2, 2, 2))):
with self.assertRaisesRegex(ValueError, '2 dimensions'):
read_image('abc')
# test dtype
with patch('imageio.imread', return_value=np.ones((3, 2), dtype=bool)):
img = read_image('abc')
self.assertEqual(img.dtype, config['SOURCE_PROC_IMAGE_DTYPE'])
self.assertEqual((3, 2), img.shape)
# test read invalid file format
with tempfile.NamedTemporaryFile(suffix='.txt') as fp:
with self.assertRaisesRegex(ValueError, 'Could not find a format'):
read_image(fp.name)
def testWriteImage(self):
# test write empty input
with self.assertRaisesRegex(ValueError, 'Please specify'):
read_image('')
# test write invalid file format
with tempfile.NamedTemporaryFile(suffix='.txt') as fp:
with self.assertRaisesRegex(ValueError, 'Could not find a format'):
read_image(fp.name)
# test read and write valid file formats
self._assert_write_read('.tif')
self._assert_write_read('.npy')
# test read and write .png file
self._assert_write_read('.png', scale=255)
def testReadImage(self):
# test read empty input
with self.assertRaisesRegex(ValueError, 'Please specify'):
read_image('')
# test wrong shape
with patch('imageio.imread', return_value=np.ones((2, 2))):
with self.assertRaisesRegex(ValueError, 'Shape of'):
read_image('abc', expected_shape=(3, 2))
# test wrong dimension
with patch('imageio.imread', return_value=np.ones((2, 2, 2))):
with self.assertRaisesRegex(ValueError, '2D array'):
read_image('abc')
# test read invalid file format
with tempfile.NamedTemporaryFile(suffix='.txt') as fp:
with self.assertRaisesRegex(ValueError, 'Could not find a format'):
read_image(fp.name)
def _assert_write_read(self, file_type, *, scale=1):
img = np.ones((2, 3), dtype=np.float32)
with tempfile.NamedTemporaryFile(suffix=file_type) as fp:
write_image(img, fp.name)
ref = read_image(fp.name)
np.testing.assert_array_equal(scale * img, ref)
def _assert_write_read(self, img, file_type, *, scale=1):
with tempfile.NamedTemporaryFile(suffix=file_type) as fp:
write_image(fp.name, img)
ref = read_image(fp.name)
np.testing.assert_array_equal(scale * img, ref)