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)