def test_movie_maker(mock_image, ImageSequenceClip):
from nglview.contrib.movie import MovieMaker
ImageSequenceClip.write_gif = MagicMock()
ImageSequenceClip.write_videofile = MagicMock()
traj = pytraj.datafiles.load_tz2()
view = nglview.show_pytraj(traj)
movie = MovieMaker(view,
in_memory=True,
download_folder='here',
render_params={'factor': 4},
moviepy_params={},
stop=2)
movie.make()
def _get_movie_maker(self, in_memory=True, **kwargs):
''' create MovieMaker object
Examples
--------
>>> movie = view._get_movie_maker(output='my.gif') # doctest: +SKIP
... movie.make()
Notes
-----
We only test with imageio 1.6 and moviepy 0.2.2.11
Good luck.
'''
from nglview.contrib.movie import MovieMaker
if 'in_memory' not in kwargs:
kwargs['in_memory'] = in_memory
movie_maker = MovieMaker(self, **kwargs)
return movie_maker
def test_movie_maker_base64_to_ndarray():
from nglview.contrib.movie import MovieMaker
s = 'iVBORw0KGgoAAAANSUhEUgAAAAUAAAAFCAYAAACNbyblAAAAHElEQVQI12P4//8/w38GIAXDIBKE0DHxgljNBAAO9TXL0Y4OHwAAAABJRU5ErkJggg=='
MovieMaker._base64_to_ndarray(s)
def test_movie_maker(ImageSequenceClip):
from nglview.contrib.movie import MovieMaker
ImageSequenceClip.write_gif = MagicMock()
ImageSequenceClip.write_videofile = MagicMock()
traj = pytraj.datafiles.load_tz2()
view = nglview.show_pytraj(traj)
movie = MovieMaker(view, in_memory=False)
movie.download_folder = os.path.join(os.path.dirname(__file__), 'data')
# fake _event
movie._event = FakeEvent()
movie._event._event_set = True
movie.make()
movie._event._event_set = False
movie.make()
movie = MovieMaker(view, in_memory=False)
movie.skip_render = True
movie.make()
movie.output = 'hello.mp4'
movie.make()
movie.interupt()
movie._event = None
movie.interupt()
movie.in_memory = True
movie.make()
movie._event._event_set = False
movie.make()
movie = MovieMaker(view,
download_folder='here',
render_params=dict(factor=4),
moviepy_params={},
stop=2)
movie.make()