def test_full_pipeline(self): ''' Smoke test of entire pipeline, from stimulus loading to event file export. ''' stim = VideoStim(join(_get_test_data_path(), 'video', 'small.mp4')) extractors = [VibranceExtractor()] timeline = stim.extract(extractors, show=False) exp = FSLExporter() tmpdir = tempfile.mkdtemp() exp.export(timeline, tmpdir) from glob import glob files = glob(join(tmpdir, '*.txt')) self.assertEquals(len(files), 1) shutil.rmtree(tmpdir)
def test_full_pipeline(self): ''' Smoke test of entire pipeline, from stimulus loading to event file export. ''' stim = VideoStim(join(_get_test_data_path(), 'video', 'small.mp4')) extractors = [FaceDetectionExtractor()] timeline = stim.extract(extractors, show=False) exp = FSLExporter() tmpdir = tempfile.mkdtemp() exp.export(timeline, tmpdir) from glob import glob files = glob(join(tmpdir, '*.txt')) self.assertEquals(len(files), 1) shutil.rmtree(tmpdir)
def test_video_stim(): ''' Test VideoStim functionality. ''' filename = join(get_test_data_path(), 'video', 'small.mp4') video = VideoStim(filename) assert video.fps == 30 assert video.n_frames in (167, 168) assert video.width == 560 # Test frame iterator frames = [f for f in video] assert len(frames) == 168 f1 = frames[100] assert isinstance(f1, VideoFrameStim) assert isinstance(f1.onset, float) f1.data.shape == (320, 560, 3) # Test getting of specific frame f2 = video.get_frame(index=100) assert isinstance(f2, VideoFrameStim) assert isinstance(f2.onset, float) f2.data.shape == (320, 560, 3)
def test_video_stim(): ''' Test VideoStim functionality. ''' filename = join(_get_test_data_path(), 'video', 'small.mp4') video = VideoStim(filename) assert video.fps == 30 assert video.n_frames == 168 assert video.width == 560 # Test frame iterator frames = [f for f in video] assert len(frames) == 168 f = frames[100] assert isinstance(f, VideoFrameStim) assert isinstance(f.onset, float) f.data.shape == (320, 560, 3)
def test_video_stim(self): ''' Test VideoStim functionality. ''' filename = join(_get_test_data_path(), 'video', 'small.mp4') video = VideoStim(filename) self.assertEquals(video.fps, 30) self.assertEquals(video.n_frames, 168) self.assertEquals(video.width, 560) # Test frame iterator frames = [f for f in video] self.assertEquals(len(frames), 168) f = frames[100] self.assertIsInstance(f, VideoFrameStim) self.assertIsInstance(f.onset, float) self.assertEquals(f.data.shape, (320, 560, 3))