def test_ldp_splitfilewrapper_discover(self):
"""Test the LDP interface for type discover with a SplitFileWrapper."""
w = SplitFileWrapper([
tbnFile,
])
f = ldp.LWASVDataFile(fh=w)
# File info
self.assertTrue(isinstance(f, ldp.TBNFile))
self.assertEqual(f.get_info("sample_rate"), 100e3)
self.assertEqual(f.get_info("data_bits"), 8)
self.assertEqual(f.get_info('nframe'), 29)
self.assertEqual(f.sample_rate, 100e3)
self.assertEqual(f.data_bits, 8)
self.assertEqual(f.nframe, 29)
# Read a frame
frame = f.read_frame()
f.close()
w.close()
def test_ldp_splitfilewrapper_mixed2(self):
"""Test the LDP interface for a SplitFileWrapper in a contorted way."""
w = SplitFileWrapper([drxFile, tbfFile], sort=False)
f = ldp.TBFFile(fh=w)
# Read some
frames = []
while True:
try:
frame = f.read_frame()
frames.append(frame)
except errors.SyncError:
continue
except errors.EOFError:
break
self.assertEqual(len(frames),
5 - 1) # We loose part of the first frame to DRX
def test_ldp_splitfilewrapper_mixed(self):
"""Test the LDP interface for a SplitFileWrapper in a contorted way."""
w = SplitFileWrapper([drxFile, tbfFile], sort=False)
f = ldp.LWASVDataFile(fh=w)
# File info
self.assertTrue(isinstance(f, ldp.DRXFile))
# Read some
frames = []
while True:
try:
frame = f.read_frame()
frames.append(frame)
except errors.SyncError:
continue
except errors.EOFError:
break
self.assertEqual(len(frames),
32 + 1) # There is data at the end of the "file" now
def test_ldp_splitfilewrapper_single(self):
"""Test the LDP interface for a SplitFileWrapper with a single file."""
w = SplitFileWrapper([
tbnFile,
])
f = ldp.TBNFile(fh=w)
# File info
self.assertEqual(f.get_info("sample_rate"), 100e3)
self.assertEqual(f.get_info("data_bits"), 8)
self.assertEqual(f.get_info('nframe'), 29)
self.assertEqual(f.sample_rate, 100e3)
self.assertEqual(f.data_bits, 8)
self.assertEqual(f.nframe, 29)
# Read a frame
frame = f.read_frame()
# Get the remaining frame count
self.assertEqual(f.get_remaining_frame_count(),
f.get_info('nframe') - 1)
self.assertEqual(f.nframe_remaining, f.get_info('nframe') - 1)
# Reset
f.reset()
# Read a chunk - short
tInt, tStart, data = f.read(0.005)
# Reset
f.reset()
# Read a chunk - long
tInt, tStart, data = f.read(1.00)
# Close it out
f.close()
w.close()
# 'with' statement support
with SplitFileWrapper([
tbnFile,
]) as w:
with ldp.TBNFile(fh=w) as f:
## File info
self.assertEqual(f.get_info("sample_rate"), 100e3)
self.assertEqual(f.get_info("data_bits"), 8)
self.assertEqual(f.get_info('nframe'), 29)
self.assertEqual(f.sample_rate, 100e3)
self.assertEqual(f.data_bits, 8)
self.assertEqual(f.nframe, 29)
## Read a frame
frame = f.read_frame()
## Get the remaining frame count
self.assertEqual(f.get_remaining_frame_count(),
f.get_info('nframe') - 1)
self.assertEqual(f.nframe_remaining, f.get_info('nframe') - 1)
# generator support
w = SplitFileWrapper([
tbnFile,
])
f = ldp.TBNFile(fh=w)
i = 0
for (tInt2, tStart2, data2) in f.read_sequence(1.0):
self.assertEqual(tInt, tInt2)
self.assertEqual(tStart, tStart2)
self.assertEqual(data.shape, data2.shape)
i += 1
self.assertEqual(i, 1)
f.close()
w.close()
# both at the same time
with SplitFileWrapper([
tbnFile,
]) as w:
with ldp.TBNFile(fh=w) as f:
i = 0
for (tInt2, tStart2, data2) in f.read_sequence(1.0):
self.assertEqual(tInt, tInt2)
self.assertEqual(tStart, tStart2)
self.assertEqual(data.shape, data2.shape)
i += 1
self.assertEqual(i, 1)
def testldp_splitfilewrapper(self):
"""Test the LDP interface for a SplitFileWrapper."""
# Split up the TBN file into many many parts
size = os.path.getsize(tbnFile)
block = size // 16
splitFiles = []
f = open(tbnFile, 'rb')
while size > 0:
splitFiles.append(
os.path.join(self.testPath, 'filepart%03i' % len(splitFiles)))
with open(splitFiles[-1], 'wb') as w:
w.write(f.read(block))
size -= block
f.close()
w = SplitFileWrapper(splitFiles)
f = ldp.TBNFile(fh=w)
# File info
self.assertEqual(f.get_info("sample_rate"), 100e3)
self.assertEqual(f.get_info("data_bits"), 8)
self.assertEqual(f.get_info('nframe'), 29)
self.assertEqual(f.sample_rate, 100e3)
self.assertEqual(f.data_bits, 8)
self.assertEqual(f.nframe, 29)
# Read a frame
frame = f.read_frame()
# Get the remaining frame count
self.assertEqual(f.get_remaining_frame_count(),
f.get_info('nframe') - 1)
self.assertEqual(f.nframe_remaining, f.get_info('nframe') - 1)
# Reset
f.reset()
# Read a chunk - short
tInt, tStart, data = f.read(0.005)
# Reset
f.reset()
# Read a chunk - long
tInt, tStart, data = f.read(1.00)
# Close it out
f.close()
w.close()
# 'with' statement support
with SplitFileWrapper(splitFiles) as w:
with ldp.TBNFile(fh=w) as f:
## File info
self.assertEqual(f.get_info("sample_rate"), 100e3)
self.assertEqual(f.get_info("data_bits"), 8)
self.assertEqual(f.get_info('nframe'), 29)
self.assertEqual(f.sample_rate, 100e3)
self.assertEqual(f.data_bits, 8)
self.assertEqual(f.nframe, 29)
## Read a frame
frame = f.read_frame()
## Get the remaining frame count
self.assertEqual(f.get_remaining_frame_count(),
f.get_info('nframe') - 1)
self.assertEqual(f.nframe_remaining, f.get_info('nframe') - 1)
# generator support
w = SplitFileWrapper(splitFiles)
f = ldp.TBNFile(fh=w)
i = 0
for (tInt2, tStart2, data2) in f.read_sequence(1.0):
self.assertEqual(tInt, tInt2)
self.assertEqual(tStart, tStart2)
self.assertEqual(data.shape, data2.shape)
i += 1
self.assertEqual(i, 1)
f.close()
w.close()
# both at the same time
with SplitFileWrapper(splitFiles) as w:
with ldp.TBNFile(fh=w) as f:
i = 0
for (tInt2, tStart2, data2) in f.read_sequence(1.0):
self.assertEqual(tInt, tInt2)
self.assertEqual(tStart, tStart2)
self.assertEqual(data.shape, data2.shape)
i += 1
self.assertEqual(i, 1)