def base_reader(self, value):
if isinstance(value, str):
reader = None
try:
reader = open_received(value)
except SarpyIOError:
pass
if reader is None:
raise SarpyIOError('Could not open file {} as a CRSD reader'.format(value))
value = reader
if not isinstance(value, CRSDTypeReader):
raise TypeError('base_reader must be a CRSDReader, got type {}'.format(type(value)))
self._base_reader = value
# noinspection PyProtectedMember
self._chippers = value._get_chippers_as_tuple()
self._index = 0
self._data_size = value.get_data_size_as_tuple()[0]
def base_reader(self, value):
if isinstance(value, str):
reader = None
# try to open as sicd type
try:
reader = open_complex(value)
except SarpyIOError:
pass
# try to open as phase_history
if reader is None:
try:
reader = open_phase_history(value)
except SarpyIOError:
pass
if reader is None:
try:
reader = open_received(value)
except SarpyIOError:
pass
if reader is None:
raise SarpyIOError('Could not open file {} as a one of the complex type readers'.format(value))
value = reader
elif isinstance(value, (tuple, list)):
value = AggregateComplexReader(value)
if not isinstance(value, AbstractReader):
raise TypeError('base_reader must be of type AbstractReader, got type {}'.format(type(value)))
if value.reader_type not in ["SICD", "CPHD", "CRSD"]:
raise SarpyIOError(
'base_reader.reader_type must be "SICD", "CPHD", or "CRSD", got {}'.format(value.reader_type))
self._base_reader = value
# noinspection PyProtectedMember
self._chippers = value._get_chippers_as_tuple()
self._index = 0
self._data_size = value.get_data_size_as_tuple()[0]
def generic_io_test(instance, test_file, reader_type_string, reader_type):
assert isinstance(instance, unittest.TestCase)
reader = None
with instance.subTest(
msg='establish reader for type {} and file {}'.format(
reader_type_string, test_file)):
reader = open_received(test_file)
instance.assertTrue(reader is not None,
msg='Returned None, so opening failed.')
if reader is None:
return # remaining tests make no sense
assert isinstance(reader, CRSDReader)
with instance.subTest(
msg='Reader for type {} should be appropriate reader'):
instance.assertTrue(
isinstance(reader, reader_type),
msg='Returned reader should be of type {}'.format(reader_type))
if not isinstance(reader, reader_type):
return # remaining tests might be misleading
with instance.subTest(
msg='Verify reader_type for type {} and file {}'.format(
reader_type_string, test_file)):
instance.assertEqual(reader.reader_type,
"CRSD",
msg='reader.reader_type should be "CRSD"')
with instance.subTest(
msg='Validity of crsd in reader of '
'type {} for file {}'.format(reader_type_string, test_file)):
if not reader.crsd_meta.is_valid(recursive=True, stack=False):
logging.warning(
'crsd in reader of type {} for file {} not valid'.format(
reader_type_string, test_file))
with instance.subTest(
msg='Fetch data_sizes and sidds for type {} and file {}'.format(
reader_type_string, test_file)):
data_sizes = reader.get_data_size_as_tuple()
if isinstance(reader, CRSDReader1_0):
elements = reader.crsd_meta.Data.Channels
else:
raise TypeError('Got unhandled reader type {}'.format(
type(reader)))
for i, (data_size, element) in enumerate(zip(data_sizes, elements)):
with instance.subTest(
msg='Verify image size for sidd index {} in reader '
'of type {} for file {}'.format(i, reader_type_string,
test_file)):
instance.assertEqual(
data_size[0],
element.NumVectors,
msg='data_size[0] and NumVectors do not agree')
instance.assertEqual(
data_size[1],
element.NumSamples,
msg='data_size[1] and NumSamples do not agree')
with instance.subTest(
msg='Basic fetch test for crsd index {} in reader '
'of type {} for file {}'.format(i, reader_type_string,
test_file)):
instance.assertEqual(reader[:2, :2, i].shape[:2], (2, 2),
msg='upper left fetch')
instance.assertEqual(reader[-2:, :2, i].shape[:2], (2, 2),
msg='lower left fetch')
instance.assertEqual(reader[-2:, -2:, i].shape[:2], (2, 2),
msg='lower right fetch')
instance.assertEqual(reader[:2, -2:, i].shape[:2], (2, 2),
msg='upper right fetch')
with instance.subTest(
msg='Verify fetching complete row(s) have correct size '
'for crsd index {} in reader of type {} and file {}'.format(
i, reader_type_string, test_file)):
test_data = reader[:, :2, i]
instance.assertEqual(test_data.shape[:2], (data_size[0], 2),
msg='Complete row fetch size mismatch')
with instance.subTest(
msg='Verify fetching complete columns(s) have correct size '
'for crsd index {} in reader of type {} file {}'.format(
i, reader_type_string, test_file)):
test_data = reader[:2, :, i]
instance.assertEqual(test_data.shape[:2], (2, data_size[1]),
msg='Complete row fetch size mismatch')
with instance.subTest(
msg='Verify fetching entire pvp data has correct size for crsd '
'index {} in reader of type {} file {}'.format(
i, reader_type_string, test_file)):
test_pvp = reader.read_pvp_variable('TxTime', i, the_range=None)
instance.assertEqual(test_pvp.shape, (data_size[0], ),
msg='Unexpected pvp total fetch size')
with instance.subTest(
msg=
'Verify fetching pvp data for slice has correct size for crsd '
'index {} in reader of type {} file {}'.format(
i, reader_type_string, test_file)):
test_pvp = reader.read_pvp_variable('TxTime',
i,
the_range=(0, 10, 2))
instance.assertEqual(test_pvp.shape, (5, ),
msg='Unexpected pvp strided slice fetch size')
if isinstance(reader, CRSDReader1_0):
generic_writer_test(reader)
del reader