def test_anyframe_nofiles(self):
serie = FileSeries(filenames=[], fixed_frames=False)
self.assertEqual(serie.nframes, 0)
for _ in serie.frames():
self.fail()
self.assertRaises(IndexError, serie.get_frame, 0)
serie.close()
def test_singleframe_nofiles(self):
serie = FileSeries(filenames=[], single_frame=True)
self.assertEqual(serie.nframes, 0)
for _ in serie.frames():
self.fail()
self.assertRaises(IndexError, serie.get_frame, 0)
serie.close()
def test_anyframe_nofiles(self):
serie = FileSeries(filenames=[], fixed_frames=False)
self.assertEqual(serie.nframes, 0)
for _ in serie.frames():
self.fail()
self.assertRaises(IndexError, serie.get_frame, 0)
serie.close()
def test_singleframe_nofiles(self):
serie = FileSeries(filenames=[], single_frame=True)
self.assertEqual(serie.nframes, 0)
for _ in serie.frames():
self.fail()
self.assertRaises(IndexError, serie.get_frame, 0)
serie.close()
def test_singleframe_frames(self):
filenames = self.get_singleframe_files()
serie = FileSeries(filenames=filenames, single_frame=True)
for frame_id, frame in enumerate(serie.frames()):
self.assertEqual(frame.data[0, 0], frame_id)
self.assertEqual(frame.header["frame_id"], "0")
self.assertIn("%03d" % frame_id, frame.header["filename"])
self.assertEqual(frame_id, 2)
serie.close()
def test_singleframe_frames(self):
filenames = self.get_singleframe_files()
serie = FileSeries(filenames=filenames, single_frame=True)
for frame_id, frame in enumerate(serie.frames()):
self.assertEqual(frame.data[0, 0], frame_id)
self.assertEqual(frame.header["frame_id"], "0")
self.assertIn("%03d" % frame_id, frame.header["filename"])
self.assertEqual(frame_id, 2)
serie.close()
def test_multiframe_frames(self):
filenames = self.get_multiframe_files()
serie = FileSeries(filenames=filenames, fixed_frames=True)
for frame_id, frame in enumerate(serie.frames()):
if frame_id not in [0, 5, 6]:
continue
expected_frame_id = {0: 0, 5: 2, 6: 0}[frame_id]
expected_file_num = {0: 0, 5: 1, 6: 2}[frame_id]
expected_data = {0: 0, 5: 1, 6: 2}[frame_id]
self.assertEqual(frame.data[0, 0], expected_data)
self.assertEqual(frame.header["frame_id"], "%d" % expected_frame_id)
self.assertIn("%03d" % expected_file_num, frame.header["filename"])
self.assertEqual(frame_id, 6)
serie.close()
def test_anyframe_frames(self):
filenames = self.get_anyframe_files()
serie = FileSeries(filenames=filenames, fixed_frames=False)
for frame_id, frame in enumerate(serie.frames()):
if frame_id not in [0, 4, 9]:
continue
expected_file_frame_id = {0: 0, 4: 1, 9: 0}[frame_id]
expected_file_num = {0: 0, 4: 1, 9: 3}[frame_id]
expected_data = {0: 0, 4: 1, 9: 2}[frame_id]
expected_filename = filenames[expected_file_num]
self.assertEqual(frame.data[0, 0], expected_data)
self.assertEqual(frame.header["frame_id"], "%d" % expected_file_frame_id)
self.assertEqual(frame.index, frame_id)
self.assertEqual(frame.file_index, expected_file_frame_id)
self.assertIs(frame.container, serie)
self.assertIs(frame.file_container.filename, expected_filename)
self.assertIn("%03d" % expected_file_num, frame.header["filename"])
self.assertEqual(frame_id, 9)
serie.close()
