def test_show_video_gif(self): htmls = [] with mock.patch('IPython.display.display', htmls.append): media.show_video(media.moving_circle(), codec='gif') self.assertLen(htmls, 1) self.assertIsInstance(htmls[0], IPython.display.HTML) self.assertContainsInOrder(['<img', 'src="data:image/gif'], htmls[0].data)
def test_show_video(self): htmls = [] with mock.patch('IPython.display.display', htmls.append): media.show_video(media.moving_circle()) self.assertLen(htmls, 1) self.assertIsInstance(htmls[0], IPython.display.HTML) self.assertLen(re.findall('(?s)<video', htmls[0].data), 1) self.assertRegex(htmls[0].data, '(?s)<video .*>.*</video>')
def test_show_video_downsampled(self): np.random.seed(1) video = np.random.rand(5, 64, 128, 3) for downsample in (False, True): htmls = [] with mock.patch('IPython.display.display', htmls.append): media.show_video(video, height=32, downsample=downsample) size_min_max = (8_000, 15_000) if downsample else (40_000, 60_000) self.assertBetween(len(htmls[0].data), *size_min_max)